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.
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.