switch between train and eval

2024-03-18 09:45:17 +00:00
parent 75cc10198f
commit 8e346b379d
2 changed files with 2 additions and 0 deletions
--- a/lerobot/scripts/eval.py
+++ b/lerobot/scripts/eval.py
@@ -32,6 +32,7 @@ def eval_policy(
    fps: int = 15,
    return_first_video: bool = False,
 ):
+    policy.eval()
    start = time.time()
    sum_rewards = []
    max_rewards = []