saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-05-23 22:20:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
stable-baselines3/README.md
Antonin RAFFIN b157b4465a Add logo
2019-09-22 13:57:18 +02:00

904 B
Raw Blame History

Build Status Documentation Status

Torchy Baselines

PyTorch version of Stable Baselines, a set of improved implementations of reinforcement learning algorithms.

TODO:

  • SAC
  • save/load
  • automatic choice for action distribution
  • predict
  • better rescale (min + action * range)
  • documentation
  • flexible mlp
  • logger
  • better monitor wrapper?

Later:

  • get_parameters / set_parameters
  • CNN policies + normalization
  • tensorboard support
  • DQN
  • TRPO
  • A2C
  • ACER
  • HER -> use stable-baselines because does not depends on tf?