Remove EMA model from Diffusion Policy (#134)

2024-05-05 11:26:12 +01:00
parent d747195c57
commit f3bba0270d
11 changed files with 21 additions and 117 deletions
--- a/lerobot/scripts/eval.py
+++ b/lerobot/scripts/eval.py
@@ -121,7 +121,7 @@ def rollout(
    max_steps = env.call("_max_episode_steps")[0]
    progbar = trange(
        max_steps,
-        desc=f"Running rollout with {max_steps} steps (maximum) per rollout",
+        desc=f"Running rollout with at most {max_steps} steps",
        disable=not enable_progbar,
        leave=False,
    )