Wandb works, One output dir

2024-02-22 12:14:12 +00:00
parent ece89730e6
commit e3643d6146
11 changed files with 200 additions and 100 deletions
--- a/lerobot/configs/pusht.yaml
+++ b/lerobot/configs/pusht.yaml
@@ -5,12 +5,16 @@ hydra:
  job:
    name: pusht

-video_dir: tmp/2024_02_21_pusht
-
 # env
 env: pusht
+task: pusht
 image_size: 96
 frame_skip: 1
 state_dim: 2
 action_dim: 2
-fps: 10
+fps: 10
+eval_episodes: 50
+episode_length: 300
+eval_freq: 7500
+save_freq: 75000
+