From 74c705d529ed1253917c5dd31427088ecb17a889 Mon Sep 17 00:00:00 2001 From: Dominik Strzako Date: Sun, 23 May 2021 23:53:37 +0200 Subject: [PATCH] tracking? --- Zadanie_08_and_09_MLflow.py | 1 + 1 file changed, 1 insertion(+) diff --git a/Zadanie_08_and_09_MLflow.py b/Zadanie_08_and_09_MLflow.py index d57db95..4637617 100644 --- 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)