Skip to content

Releases: araffin/rl-baselines-zoo

Checkpoints and new callback collection

22 Mar 12:54
79a6fd1
Compare
Choose a tag to compare
  • 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

Stable-Baselines v2.10.0 support and improvements

14 Mar 15:24
1c2a3cf
Compare
Choose a tag to compare
  • 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

New save format, ACKTR with continuous actions

29 Sep 18:35
a41e611
Compare
Choose a tag to compare
  • Upgrade to Stable-Baselines v2.8.0

Add Twin Delayed DDPG (TD3) trained agents

29 Sep 16:48
Compare
Choose a tag to compare

HER support + Time wrappers

16 Jun 21:44
b0f43e1
Compare
Choose a tag to compare
  • 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

Hyperparameter tuning + TRPO

28 May 09:19
64267d3
Compare
Choose a tag to compare
  • 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.

Soft Actor-Critic + Improvements

17 Jan 19:56
Compare
Choose a tag to compare
  • Add Soft Actor-Critic (SAC)
  • Tune bullet env for PPO2
  • Improve logging

Video Recorder

23 Nov 13:01
Compare
Choose a tag to compare

Add video recorder