Zaktualizuj 'train-mlflow.py'

2022-05-15 16:00:09 +02:00 · 2022-05-15 16:00:09 +02:00 · dbfb321c34
commit dbfb321c34
parent 38c237f321
1 changed files with 1 additions and 0 deletions
--- a/train-mlflow.py
+++ b/train-mlflow.py
@ -29,6 +29,7 @@ from urllib.parse import urlparse
 from mlflow.models import infer_signature

 mlflow.set_experiment("s478815")
+mlflow.set_tracking_uri("http://172.17.0.1:5000")

 # Model
 class Model(nn.Module):