Wandb works, One output dir

lerobot/configs/default.yaml

@@ -1,7 +1,10 @@
+hydra:
+  run:
+    dir: outputs/${now:%Y-%m-%d}/${now:%H-%M-%S}_${hydra.job.name}
+  job:
+    name: default
+
 seed: 1337
-log_dir: logs/2024_01_26_train
-video_dir: tmp/2024_01_26_xarm_lift_medium
-exp_name: default
 device: cuda
 buffer_device: cuda
 eval_freq: 1000
@@ -21,6 +24,7 @@ fps: 15
 
 reward_scale: 1.0
 
+
 # xarm_lift
 episode_length: 25
 modality: 'all'
@@ -97,6 +101,7 @@ mlp_dim: 512
 latent_dim: 50
 
 # wandb
-use_wandb: false
-wandb_project: FOWM
+use_wandb: true
+wandb_project: lerobot
 wandb_entity: rcadene  # insert your own
+notes: ""

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
+