tracking?

2021-05-23 23:53:37 +02:00 · 2021-05-23 23:53:37 +02:00 · 74c705d529
commit 74c705d529
parent 817b1a7cbb
1 changed files with 1 additions and 0 deletions
--- a/Zadanie_08_and_09_MLflow.py
+++ b/Zadanie_08_and_09_MLflow.py
@ -17,6 +17,7 @@ with mlflow.start_run():
    batch_param = int(sys.argv[1]) if len(sys.argv) > 1 else 16
    epoch_param = int(sys.argv[2]) if len(sys.argv) > 2 else 15

+    mlflow.set_tracking_uri("http://172.17.0.1:5000")
    mlflow.set_experiment("s434788")
    mlflow.log_param("batch_size", batch_param)
    mlflow.log_param("epochs", epoch_param)