Merge branch 'main' into user/alexander-soare/multistep_policy_and_serial_env

2024-03-18 18:27:50 +00:00
parent bae7e7b41c 099a465367
commit 09ddd9bf92
2 changed files with 2 additions and 0 deletions
--- a/lerobot/scripts/eval.py
+++ b/lerobot/scripts/eval.py
@@ -33,6 +33,7 @@ def eval_policy(
    fps: int = 15,
    return_first_video: bool = False,
 ):
+    policy.eval()
    start = time.time()
    sum_rewards = []
    max_rewards = []