huggingface / huggingface_sb3

Additional code for Stable-baselines3 to load and upload models from the Hub.
74 stars 23 forks source link

Allow to pass TensorBoard logs files to package_to_hub #5

Closed osanseviero closed 2 years ago

osanseviero commented 2 years ago

It's very easy to add TensorBoard logging with SB3, but pushing the files right now needs to be done manually. As an alternative, we could add a param to package_to_hub to pass the logs.

Related: https://github.com/huggingface/deep-rl-class/pull/19