fix(sac): clarify torch.compile status

This commit is contained in:
Khalil Meftah
2026-04-18 15:19:35 +02:00
parent e4c1a8472d
commit 87d4c9879c
2 changed files with 3 additions and 1 deletions
@@ -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