train a simple Pytorch model to test the MLflow lib
Go to file
deng 1f86146b12 replace bash script to docker-compose to build server 2023-03-07 16:33:45 +08:00
env replace bash script to docker-compose to build server 2023-03-07 16:33:45 +08:00
service mod service file structure 2023-03-01 14:55:51 +08:00
.gitignore ignore saved model 2023-02-22 16:27:19 +08:00
README.md replace bash script to docker-compose to build server 2023-03-07 16:33:45 +08:00
docker-compose.yaml replace bash script to docker-compose to build server 2023-03-07 16:33:45 +08:00
get_registered_model_via_rest_api.py replace bash script to docker-compose to build server 2023-03-07 16:33:45 +08:00
predict.py replace bash script to docker-compose to build server 2023-03-07 16:33:45 +08:00
test_pytorch_m1.py update file description and reference 2023-03-01 17:13:17 +08:00
train.py replace bash script to docker-compose to build server 2023-03-07 16:33:45 +08:00

README.md

Abstract

Try to use MLflow platform to log PyTorch model training, and pull production model from model registry to run inference⛩

Requirements

  • MacOS 12.5
  • Docker 20.10

Dirs

  • service
    • House MLflow service data, including MLflow artifacts, backend store and model registry
  • env
    • mlflow.yaml
      • conda env yaml to run this repo

Files

  • docker-compose.yaml
    • a yaml to apply docker-compose to start MLflow service with basic configuration (run docker-compose -f docker-compose.yaml up)
  • test_pytorch_m1.py
    • a script to test PyTorch on Apple M1 platform with GPU acceleration
  • train.py
    • a sample code to apply PyTorch to train a small neural network to predict fortune with MLflow logging
  • predict.py
    • a sample code to call registered model to predict testing data and save model to local file system
  • get_registered_model_via_rest_api.py
    • a script to test MLflow REST api
tags: MLOps