search

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 #17

Open andodet opened 3 years ago

andodet commented 3 years ago

First of all thanks for the project, it's an interesting way to take a stab at reducing the amount of boilerplate needed even for fairly simple models. Secondly, it would be interesting to implement experiment/run tracking using MLflow.

Have a working example on the Image classification_PyTorch/ template, happy to submit a PR if you consider this of any interest.

jrieke commented 3 years ago

Yeah this would be great, feel free to open a PR, then I'll have a look!