fix(tdmpc): Add missing save_freq to tdmpc policy config (#404)

Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
2024-09-02 14:04:41 -04:00 · 2024-09-02 14:04:41 -04:00 · 27ba2951d1
parent b2896d38f5
commit 27ba2951d1
1 changed files with 1 additions and 0 deletions
--- a/lerobot/configs/policy/tdmpc.yaml
+++ b/lerobot/configs/policy/tdmpc.yaml
@ -12,6 +12,7 @@ training:
  grad_clip_norm: 10.0
  lr: 3e-4

+  save_freq: 10000
  eval_freq: 5000
  log_freq: 100