A tryout of MLflow on a custom PyTorch cnn - View it on GitHub