mlflow-fun PyTorch Examples Overview From https://mlflow.org/docs/latest/python_api/mlflow.pytorch.html Initialization pip3 install numpy pip3 install torch pip3 install torchvision pip3 install mlflow Run python3 pytorch_test.py