mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-24 18:26:11 +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)
|
concurrency: ConcurrencyConfig = field(default_factory=ConcurrencyConfig)
|
||||||
|
|
||||||
# Optimizations
|
# 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
|
use_torch_compile: bool = False
|
||||||
|
|
||||||
def __post_init__(self):
|
def __post_init__(self):
|
||||||
|
|||||||
@@ -114,7 +114,7 @@ class SACAlgorithm(RLAlgorithm):
|
|||||||
**asdict(self.config.discrete_critic_network_kwargs),
|
**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())
|
discrete_critic_target.load_state_dict(discrete_critic.state_dict())
|
||||||
return discrete_critic, discrete_critic_target
|
return discrete_critic, discrete_critic_target
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user