DLR-RM / rl-baselines3-zoo

A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.
https://rl-baselines3-zoo.readthedocs.io
MIT License
2k stars 510 forks source link

[Feature Request] Store git hash of key repos/packages #426

Open araffin opened 10 months ago

araffin commented 10 months ago

🚀 Feature

Allow to store git hash (à la W&B) and maybe do a pip freeze to known which version of what is used. Could be opt-in (log only SB3, RL Zoo at first and then log more if the user ask) or opt out (log every package unless specified).

I see mainly RL Zoo dependencies + env used as candidates.

Motivation

The idea is to make experiment even more reproducible, we store the SB3 version and related libs (like pytorch or gym) in the model zip but not the environment version/hash.

Pitch

No response

Alternatives

No response

Additional context

No response

Checklist