Change SAC policy implementation with configuration and modeling classes
This commit is contained in:
committed by
AdilZouitine
parent
bd8c768f62
commit
4b78ab2789
@@ -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.")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user