A collection of 100+ pre-trained RL agents using Stable Baselines, training and hyperparameter optimization included.
EvalCallback
for proper evaluation using a separate env (--eval-freq
, --eval-episodes
)--save-freq
argument to save periodic checkpoints of the model--load-best
argument to enjoy script to load the best model according to evaluationnow more than 100 trained agents.
Add video recorder