diff --git a/sota-implementations/crossq/batchrenorm.py b/sota-implementations/crossq/batchrenorm.py index 93ac9195e33..81000b4fa7b 100644 --- a/sota-implementations/crossq/batchrenorm.py +++ b/sota-implementations/crossq/batchrenorm.py @@ -31,7 +31,7 @@ class BatchRenorm(nn.Module): def __init__( self, num_features, - momentum=0.99, + momentum=0.01, eps=1e-5, max_r=3.0, max_d=5.0, diff --git a/sota-implementations/crossq/utils.py b/sota-implementations/crossq/utils.py index 1fabd994279..f6615689384 100644 --- a/sota-implementations/crossq/utils.py +++ b/sota-implementations/crossq/utils.py @@ -285,7 +285,7 @@ def make_crossQ_optimizer(cfg, loss_module): ) optimizer_alpha = optim.Adam( [loss_module.log_alpha], - lr=3.0e-4, + lr=cfg.optim.lr, ) return optimizer_actor, optimizer_critic, optimizer_alpha