Fix convergence of sac, multiple torch compile on the same model caused divergence

2026-06-02 20:01:25 +00:00 · 2025-03-31 13:54:21 +00:00
parent 8494634d48
commit 026ad463a9
3 changed files with 1 additions and 5 deletions
--- a/lerobot/scripts/server/actor_server.py
+++ b/lerobot/scripts/server/actor_server.py
@@ -231,7 +231,6 @@ def act_with_policy(
        cfg=cfg.policy,
        env_cfg=cfg.env,
    )
-    policy = torch.compile(policy)
    assert isinstance(policy, nn.Module)

    obs, info = online_env.reset()