I've added a MLFlowBackend type for logging to MLFlow, similar to the TensorBoardBackend. It uses the MLFLogger from MLFlowLogger.jl (which now uses the REST API).
Currently, the log_to() method is implemented for Loggables.Value. I can see if I can add log methods for the other types Loggables.Image, Loggables.Text, Loggables.Histogram later, just wanted to get some first feedback.
I also still have to figure out how to start the mlflow server in the CI on Windows to make all tests pass.
Any first feedback?
PR Checklist
[x] Tests are added: added a test for LogMetrics
[x] Documentation, if applicable: documentation in the code and added a reference in Features.md
I've added a
MLFlowBackend
type for logging to MLFlow, similar to theTensorBoardBackend
. It uses theMLFLogger
from MLFlowLogger.jl (which now uses the REST API).Currently, the
log_to()
method is implemented forLoggables.Value
. I can see if I can add log methods for the other typesLoggables.Image
,Loggables.Text
,Loggables.Histogram
later, just wanted to get some first feedback.I also still have to figure out how to start the mlflow server in the CI on Windows to make all tests pass.
Any first feedback?
PR Checklist
LogMetrics