Add tracking to MLFlow Models

2022-05-09 18:30:08 +02:00 · 2022-05-09 18:30:08 +02:00 · 4f5f887456
commit 4f5f887456
parent 858e9ec215
1 changed files with 1 additions and 1 deletions
--- 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")