text: Title: Add mlflow tracking
  Body: 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][1]. 
  
  Have a working example on the `Image classification_PyTorch/` template, happy to submit a PR if you consider this of any interest.
  
  [1]: https://www.mlflow.org/docs/latest/index.html