tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

added optimizer and sac to factory.py

Browse Source
This commit is contained in:
Michel Aractingi
2024-12-23 14:12:03 +01:00
parent 22a1899ff4
commit ad7eea132d
2 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lerobot/common/policies/factory.py
View File

@@ -59,6 +59,12 @@ def get_policy_class(name: str) -> PreTrainedPolicy:
from lerobot.common.policies.pi0fast.modeling_pi0fast import PI0FASTPolicy from lerobot.common.policies.pi0fast.modeling_pi0fast import PI0FASTPolicy
return PI0FASTPolicy return PI0FASTPolicy
elif name == "sac":
from lerobot.common.policies.sac.configuration_sac import SACConfig
from lerobot.common.policies.sac.modeling_sac import SACPolicy
return SACPolicy, SACConfig
else: else:
raise NotImplementedError(f"Policy with name {name} is not implemented.") raise NotImplementedError(f"Policy with name {name} is not implemented.")

1
lerobot/common/policies/sac/configuration_sac.py
View File

@@ -26,6 +26,7 @@ class SACConfig:
num_subsample_critics = None num_subsample_critics = None
critic_lr = 3e-4 critic_lr = 3e-4
actor_lr = 3e-4 actor_lr = 3e-4
temperature_lr = 3e-4
critic_target_update_weight = 0.005 critic_target_update_weight = 0.005
utd_ratio = 2 utd_ratio = 2
critic_network_kwargs = { critic_network_kwargs = {