update lr param

pytorch · Jul 8, 2024 · 02c94ff · 02c94ff
1 parent f94165e
commit 02c94ff
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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