FluxML / FluxTraining.jl

A flexible neural net training library inspired by fast.ai
https://fluxml.ai/FluxTraining.jl
MIT License
117 stars 25 forks source link

added MLFlowBackend #163

Open itan1 opened 4 months ago

itan1 commented 4 months ago

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