Add resume training (#205)

Co-authored-by: Remi <re.cadene@gmail.com>
2026-07-25 18:56:09 +00:00 · 2024-05-28 12:04:23 +01:00
parent 7ec76ee235
commit e3b9f1c19b
15 changed files with 486 additions and 191 deletions
@@ -15,7 +15,7 @@ training:
  eval_freq: 10000
  save_freq: 100000
  log_freq: 250
-  save_model: true
+  save_checkpoint: true

  batch_size: 8
  lr: 1e-5