backup wip

2024-04-11 17:51:35 +01:00
parent 91ff69d64c
commit 976a197f98
26 changed files with 661 additions and 2733 deletions
--- a/lerobot/scripts/eval.py
+++ b/lerobot/scripts/eval.py
@@ -121,7 +121,7 @@ def eval_policy(

        # get the next action for the environment
        with torch.inference_mode():
-            action = policy.select_action(observation, step)
+            action = policy.select_action(observation, step=step)

        # apply inverse transform to unnormalize the action
        action = postprocess_action(action, transform)