Skip to content

SB3 v1.5.0: Support for Weight and Biases experiment tracking

Compare
Choose a tag to compare
@araffin araffin released this 25 Mar 14:21
· 94 commits to master since this release
2fe4418

Release 1.5.0 (2022-03-25)

Support for Weight and Biases experiment tracking

Breaking Changes

  • Upgrade to Stable-Baselines3 (SB3) >= 1.5.0
  • Upgrade to sb3-contrib >= 1.5.0
  • Upgraded to gym 0.21

New Features

  • Verbose mode for each trial (when doing hyperparam optimization) can now be activated using the debug mode (verbose == 2)
  • Support experiment tracking via Weights and Biases via the --track flag (@vwxyzjn)
  • Support tracking raw episodic stats via RawStatisticsCallback (@vwxyzjn, see #216)

Bug fixes

  • Policies saved during during optimization with distributed Optuna load on new systems (@JKTerry)
  • Fixed script for recording video that was not up to date with the enjoy script