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

Run tracking with MLflow #448

Closed macmacal closed 5 months ago

macmacal commented 5 months ago

Description

This PR refactores current tracking with Wieghts and Biases to be more abstract:

In addition:

Motivation and Context

Types of changes

Checklist:

Note: we are using a maximum length of 127 characters per line

macmacal commented 5 months ago

Sorry for opening PR to main repository, this should have ended in https://github.com/AGH-CEAI/rl-baselines3-zoo