diff --git a/experiment_plan.md b/experiment_plan.md index 67184cd..0391ba8 100644 --- a/experiment_plan.md +++ b/experiment_plan.md @@ -11,20 +11,29 @@ - ant, cheetah, humanoid, walker, hopper - Results: https://wandb.ai/dominik_roth/reppo_brax_production -**DMC Suite (mujoco_playground)**: 23 tasks +🔄 **DMC Suite (mujoco_playground)**: 24 tasks - **IN PROGRESS** - AcrobotSwingup, CartpoleBalance, CheetahRun, FingerSpin, HumanoidRun, WalkerRun, etc. +- Dev test completed successfully: https://wandb.ai/dominik_roth/reppo_dmc_dev +- Production jobs: 72 jobs (24 tasks × 3 seeds) submitted, IDs 3382257-3382330 **ManiSkill Suite**: 8 tasks (need wrapper first) - PickSingleYCB-v1, PegInsertionSide-v1, UnitreeG1TransportBox-v1, etc. -**Settings**: 50M steps, 1024 envs, 5 seeds each, paper hyperparameters +**Settings**: 50M steps, 1024 envs, 3 seeds each, paper hyperparameters -## Scripts Needed +## Scripts Available -### `submit_experiments.py` -Uses existing working SLURM script: +### DMC Experiments ```bash -python submit_experiments.py --experiment brax --seeds 5 -python submit_experiments.py --experiment mjx --seeds 5 -python submit_experiments.py --experiment maniskill --seeds 5 +# Submit all 24 DMC tasks with 3 seeds each +python submit_dmc_experiments.py --seeds 3 + +# Submit specific tasks +python submit_dmc_experiments.py --tasks CartpoleBalance CheetahRun --seeds 3 +``` + +### Brax Experiments (Already completed) +```bash +./submit_job.sh brax ant mjx_dmc_medium_data +./submit_job.sh brax cheetah mjx_dmc_medium_data ``` \ No newline at end of file