From 4f5f887456f1aa7b9ca7372bbffa29ce3bd2c097 Mon Sep 17 00:00:00 2001 From: Marcin Kostrzewski Date: Mon, 9 May 2022 18:30:08 +0200 Subject: [PATCH] Add tracking to MLFlow Models --- train_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train_model.py b/train_model.py index d3af166..d648d55 100644 --- a/train_model.py +++ b/train_model.py @@ -13,7 +13,7 @@ default_epochs = 5 device = "cuda" if torch.cuda.is_available() else "cpu" -#mlflow.set_tracking_uri("http://172.17.0.1:5000") +mlflow.set_tracking_uri("http://172.17.0.1:5000") mlflow.set_experiment("s444409")