Examples

We provide a number of examples on how to use the suite, these can be found in the examples directory. Below we show how to run the files.

Training a new agent

In the example we show how to train a new agent using the suite:

# Run training script or evaluate pre-trained policy
cd examples/02_train/train.py
python train.py --task="AAURoverEnv-v0" --num_envs=128

Using pre-trained agent

# Run training script or evaluate pre-trained policy
cd examples/03_inference_pretrained
python eval.py --task="AAURoverEnv-v0" --num_envs=128

Recording data

# Run training script or evaluate pre-trained policy
cd examples/03_inference_pretrained
python record.py --task="AAURoverEnv-v0" --num_envs=16