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/common/envs/factory.py
+++ b/lerobot/common/envs/factory.py
@@ -11,6 +11,7 @@ def make_env(cfg):
        "from_pixels": cfg.from_pixels,
        "pixels_only": cfg.pixels_only,
        "image_size": cfg.image_size,
+        "max_episode_length": cfg.episode_length,
    }

    if cfg.env == "simxarm":