Stablebaselines agent compatibility with rlberry

rlberry-py / rlberry

An easy-to-use reinforcement learning library for research and education.

https://rlberry-py.github.io/rlberry

MIT License

160 stars 30 forks source link

Stablebaselines agent compatibility with rlberry #457

Open TimotheeMathieu opened 2 months ago

TimotheeMathieu commented 2 months ago

Hello,

There are some minor compatibility issues in StableBaselinesAgent. In particular: WriterWrapper does not work with StableBaselinesAgent, it would be nice to be able to access easily evaluations of an agent during training. It would also be nice to save the episode reward if possible.