Track training with mlflow

created branch 1-track-training-with-mlflow to address this issue

mentioned in merge request !1 (merged)

closed with merge request !1 (merged)