Change SAC policy implementation with configuration and modeling classes

This commit is contained in:
Adil Zouitine
2025-01-17 09:39:04 +01:00
committed by AdilZouitine
parent bd8c768f62
commit 4b78ab2789
4 changed files with 51 additions and 714 deletions

View File

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