train a simple Pytorch model to test the MLflow lib

Go to file

deng 2dd734b87b update server data		2023-02-22 16:26:02 +08:00
artifacts	update server data	2023-02-22 16:26:02 +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	record model input&output, save model to file system	2023-02-22 16:25:26 +08:00
mlflow.db	update server data	2023-02-22 16:26:02 +08:00
predict.py	record model input&output, save model to file system	2023-02-22 16:25:26 +08:00
start_mlflow_server.sh	record model input&output, save model to file system	2023-02-22 16:25:26 +08:00
train.py	record model input&output, save model to file system	2023-02-22 16:25:26 +08:00

test_mlflow

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