mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-06 17:41:47 +00:00
fix(sac): clarify torch.compile status
This commit is contained in:
@@ -194,6 +194,8 @@ class SACConfig(PreTrainedConfig):
|
||||
concurrency: ConcurrencyConfig = field(default_factory=ConcurrencyConfig)
|
||||
|
||||
# Optimizations
|
||||
# torch.compile is currently disabled by default due to known issues with the SAC
|
||||
# critic ensemble and shared encoder.
|
||||
use_torch_compile: bool = False
|
||||
|
||||
def __post_init__(self):
|
||||
|
||||
@@ -114,7 +114,7 @@ class SACAlgorithm(RLAlgorithm):
|
||||
**asdict(self.config.discrete_critic_network_kwargs),
|
||||
)
|
||||
|
||||
# TODO: (maractingi, azouitine) Compile the discrete critic
|
||||
# TODO(Khalil): Compile the discrete critic
|
||||
discrete_critic_target.load_state_dict(discrete_critic.state_dict())
|
||||
return discrete_critic, discrete_critic_target
|
||||
|
||||
|
||||
Reference in New Issue
Block a user