As mentioned in #17, adding training run and experiment tracking capabilities using Mlflow for the following templates:
templates/Image classification_PyTorch
templates/Image classification_scikit-learn
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
[x] all tests are passing (see README.md on how to run tests)
[ ] if you created a new template: it contains a file test-inputs.yml, which specifies a few input values to test the code template (the test is then automatically run by pytest)
Summary
As mentioned in #17, adding training run and experiment tracking capabilities using Mlflow for the following templates:
templates/Image classification_PyTorch
templates/Image classification_scikit-learn
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
test-inputs.yml
, which specifies a few input values to test the code template (the test is then automatically run by pytest)