Enable Mongo

2022-05-09 10:03:04 +02:00 · 2022-05-09 10:03:04 +02:00 · 1a44512224
commit 1a44512224
parent 076de7c773
1 changed files with 2 additions and 2 deletions
--- a/train_model.py
+++ b/train_model.py
@ -49,8 +49,8 @@ def main(batch_size, epochs, _run):
 def setup_experiment():
    ex = Experiment('Predict power output for a given time')
    ex.observers.append(FileStorageObserver('sacred_runs'))
-    # ex.observers.append(MongoObserver(url='mongodb://mongo_user:mongo_password_IUM_2022@localhost:27017',
-    #                                   db_name='sacred'))
+    ex.observers.append(MongoObserver(url='mongodb://admin:IUM_2021@172.17.0.1:27017',
+                                      db_name='sacred'))
    return ex