Refactor train, eval_policy, logger, Add diffusion.yaml (WIP)

2024-02-26 01:10:09 +00:00
parent 5a219fed6e
commit 21670dce90
12 changed files with 306 additions and 443 deletions
--- a/lerobot/configs/env/pusht.yaml
+++ b/lerobot/configs/env/pusht.yaml
@@ -3,7 +3,9 @@
 eval_episodes: 50
 eval_freq: 7500
 save_freq: 75000
-train_steps: 50000  # TODO: same as simxarm, need to adjust
+# TODO: same as simxarm, need to adjust
+offline_steps: 25000
+online_steps: 25000

 fps: 10

--- a/lerobot/configs/env/simxarm.yaml
+++ b/lerobot/configs/env/simxarm.yaml
@@ -3,7 +3,9 @@
 eval_episodes: 20
 eval_freq: 1000
 save_freq: 10000
-train_steps: 50000
+log_freq: 50
+offline_steps: 25000
+online_steps: 25000

 fps: 15