From 27ba2951d128a3db2497d1337031e01fb995ccfe Mon Sep 17 00:00:00 2001 From: Jack Vial Date: Mon, 2 Sep 2024 14:04:41 -0400 Subject: [PATCH] fix(tdmpc): Add missing save_freq to tdmpc policy config (#404) Co-authored-by: Alexander Soare --- lerobot/configs/policy/tdmpc.yaml | 1 + 1 file changed, 1 insertion(+) diff --git a/lerobot/configs/policy/tdmpc.yaml b/lerobot/configs/policy/tdmpc.yaml index 40eab35f..7c85fcf8 100644 --- 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