jrieke / traingenerator

🧙 A web app to generate template code for machine learning
https://traingenerator.jrieke.com
MIT License
1.36k stars 181 forks source link

Add Mlflow tracking #18

Closed andodet closed 3 days ago

andodet commented 3 years ago

Summary

As mentioned in #17, adding training run and experiment tracking capabilities using Mlflow for the following templates:

Details

This PR implements basic metric/model info logging features. As a second logical next step, model artifact logging (checkpoints and final model) will be added.

Checklist