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/configs/policy/diffusion.yaml
+++ b/lerobot/configs/policy/diffusion.yaml
@@ -27,7 +27,7 @@ training:
  eval_freq: 5000
  save_freq: 5000
  log_freq: 250
-  save_model: true
+  save_checkpoint: true

  batch_size: 64
  grad_clip_norm: 10