update dllib-mlflow.py

2022-05-09 22:31:58 +02:00 · 2022-05-09 22:31:58 +02:00 · 820a8663ec
commit 820a8663ec
parent c2a7fbbf25
1 changed files with 1 additions and 0 deletions
--- a/Biblioteka_DL/dllib-mlflow.py
+++ b/Biblioteka_DL/dllib-mlflow.py
@ -17,6 +17,7 @@ from sacred.observers import MongoObserver

 # EPOCHS = int(sys.argv[1])

+mlflow.set_tracking_uri("http://localhost:5000")
 mlflow.set_experiment("s444356")

 class Model(nn.Module):