train a simple Pytorch model to test the MLflow lib

Go to file

deng 6e80927c40 add .gitignore		2023-02-21 16:09:05 +08:00
artifacts/0	init	2023-02-21 14:15:39 +08:00
.gitignore	add .gitignore	2023-02-21 16:09:05 +08:00
README.md	add README	2023-02-21 14:20:12 +08:00
env.yaml	init	2023-02-21 14:15:39 +08:00
mlflow.db	init	2023-02-21 14:15:39 +08:00
predict.py	init	2023-02-21 14:15:39 +08:00
start_mlflow_server.sh	init	2023-02-21 14:15:39 +08:00
train.py	init	2023-02-21 14:15:39 +08:00

README.md

test_mlflow

測試使用MLflow紀錄Pytorch模型訓練，以及從Model registry中拉取Production model進行推論。