This folder contains a submitit-release-check.sh file that executes all the training scripts using sbatch with the default configuration and long them into a common WandB project.

This script is to be executed before every release to assess the performance of the various algorithms available in torchrl. The name of the project will include the specific commit of torchrl used to run the scripts (e.g. torchrl-examples-check-<commit>).

To display the script usage, you can use the --help option:

./submitit-release-check.sh --help

The following setup should allow you to run the scripts:

export MUJOCO_GL=egl

conda create -n rl-sota-bench python=3.10 -y 
conda install anaconda::libglu -y
pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu121
pip3 install "gymnasium[accept-rom-license,atari,mujoco]" vmas tqdm wandb pygame moviepy imageio submitit hydra-core transformers

cd /path/to/tensordict
python setup.py develop
cd /path/to/torchrl
python setup.py develop