test_mlflow

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

Description
train a simple Pytorch model to test the MLflow lib
Readme 522 KiB
Languages
Python 100%