added optimizer and sac to factory.py

2024-12-23 14:12:03 +01:00
parent 22a1899ff4
commit ad7eea132d
2 changed files with 7 additions and 0 deletions
--- a/lerobot/common/policies/sac/configuration_sac.py
+++ b/lerobot/common/policies/sac/configuration_sac.py
@@ -26,6 +26,7 @@ class SACConfig:
    num_subsample_critics = None
    critic_lr = 3e-4
    actor_lr = 3e-4
+    temperature_lr = 3e-4
    critic_target_update_weight = 0.005
    utd_ratio = 2
    critic_network_kwargs = {