Small fix and improve logging message

2024-02-27 11:44:26 +00:00
parent 21670dce90
commit 7df542445c
5 changed files with 37 additions and 16 deletions
--- a/lerobot/configs/env/pusht.yaml
+++ b/lerobot/configs/env/pusht.yaml
@@ -3,6 +3,7 @@
 eval_episodes: 50
 eval_freq: 7500
 save_freq: 75000
+log_freq: 250
 # TODO: same as simxarm, need to adjust
 offline_steps: 25000
 online_steps: 25000
--- a/lerobot/configs/policy/diffusion.yaml
+++ b/lerobot/configs/policy/diffusion.yaml
@@ -21,6 +21,9 @@ past_action_visible: False
 keypoint_visible_rate: 1.0
 obs_as_global_cond: True

+offline_steps: 50000
+online_steps: 0
+
 policy:
  name: diffusion