Rl Baselines Zoo Versions Save

A collection of 100+ pre-trained RL agents using Stable Baselines, training and hyperparameter optimization included.

v1.6

4 years ago
  • Add EvalCallback for proper evaluation using a separate env (--eval-freq, --eval-episodes)
  • Add --save-freq argument to save periodic checkpoints of the model
  • Add --load-best argument to enjoy script to load the best model according to evaluation
  • Better seeding and fix possible race condition in folder creation
  • Use new callback for hyperparameter tuning

v1.5

4 years ago
  • Upgrade to Stable-Baselines v2.10.0

Many improvements:

  • save on ctrl+c
  • save training logs with the model
  • specify a wrapper or list of wrapper in the config
  • change hyperparams via the command line

v1.4

4 years ago
  • Upgrade to Stable-Baselines v2.8.0

v1.3

4 years ago

v1.2

4 years ago
  • Add support for Hindsight Experience Replay
  • Upgrade Stable-Baselines version (v2.6.0) required
  • Add wrappers for HER and for time constrained envs
  • Changed default for hyperparameter tuning

v1.1

4 years ago
  • added hyperparameter tuning using optuna
  • a2c for continuous actions
  • upgrade stable-baselines (v2.5.1)
  • add support for trpo + mpi training
  • fixed frame stack loading

now more than 100 trained agents.

v1.0

5 years ago
  • Add Soft Actor-Critic (SAC)
  • Tune bullet env for PPO2
  • Improve logging

v0.5

5 years ago

Add video recorder