upload model config to mlflow

SiLiKhon · May 2, 2022 · 7de367f · 7de367f
1 parent 6d1189b
commit 7de367f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/export_model.py b/export_model.py
@@ -96,6 +96,7 @@ def to_save(x):
             mlflow.set_tracking_uri(args.mlflow_url)
             mlflow.set_experiment('model_export')
 
+            mlflow.log_artifact(str(model_path / 'config.yaml'), artifact_path='model_onnx')
             mlflow.onnx.log_model(onnx_model, artifact_path='model_onnx', registered_model_name=args.mlflow_model_name)