Simple MLflow example

Simple example that demonstrates use of MLflow by training a Pytorch model using MNIST-dataset.

Structure

.
├── mlflow_example.ipynb
├── README.md
└── requirements.txt

Usage

Create virtual environment and install dependencies:

python -m venv .venv
pip install -r requirements.txt

Follow the instructions in mlflow_example.ipynb.