diff --git a/dlgssdpytorch.py b/dlgssdpytorch.py
index d6ee4f5..5edb142 100644
--- a/dlgssdpytorch.py
+++ b/dlgssdpytorch.py
@@ -11,7 +11,7 @@ from sacred.observers import MongoObserver, FileStorageObserver
 
 ex = Experiment("426206", interactive=False, save_git_info=False)
 ex.observers.append(FileStorageObserver('my_runs'))
-#ex.observers.append(MongoObserver(url='mongodb://mongo_user:mongo_password_IUM_2021@172.17.0.1:27017', db_name='sacred'))
+ex.observers.append(MongoObserver(url='mongodb://mongo_user:mongo_password_IUM_2021@172.17.0.1:27017', db_name='sacred'))
 class LayerLinearRegression(nn.Module):
     def __init__(self):
         super().__init__()
diff --git a/generate_MLmodel.py b/generate_MLmodel.py
index a93941f..cddf618 100644
--- a/generate_MLmodel.py
+++ b/generate_MLmodel.py
@@ -29,8 +29,8 @@ if __name__ == "__main__":
 
     lr = args.lr
     n_epochs = args.epochs
-    mlflow.set_tracking_uri("http://127.0.0.1:5000")
-    #mlflow.set_tracking_uri("http://172.17.0.1:5000")
+    #mlflow.set_tracking_uri("http://127.0.0.1:5000")
+    mlflow.set_tracking_uri("http://172.17.0.1:5000")
     mlflow.set_experiment("s426206")
     with mlflow.start_run():
         mlflow.log_param("lr", lr)