Add resume training (#205)

Co-authored-by: Remi <re.cadene@gmail.com>
2024-05-28 12:04:23 +01:00
parent 7ec76ee235
commit e3b9f1c19b
15 changed files with 486 additions and 191 deletions
--- a/lerobot/scripts/eval.py
+++ b/lerobot/scripts/eval.py
@@ -28,7 +28,7 @@ OR, you want to evaluate a model checkpoint from the LeRobot training script for

 ```
 python lerobot/scripts/eval.py \
-    -p outputs/train/diffusion_pusht/checkpoints/005000 \
+    -p outputs/train/diffusion_pusht/checkpoints/005000/pretrained_model \
    eval.n_episodes=10
 ```