From 78dd1d38fd2e9d9fc0760ecf0a153426c0404c2e Mon Sep 17 00:00:00 2001
From: Agata <agakul1@st.amu.edu.pl>
Date: Sun, 8 May 2022 15:53:03 +0200
Subject: [PATCH] Add MongoObserver (sacred)

---
 Dockerfile         |  1 +
 training_sacred.py | 14 +++++++++++---
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 3ddd895..26bd3e5 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -8,6 +8,7 @@ RUN pip3 install torch
 RUN pip3 install numpy
 RUN pip3 install sacred
 RUN pip3 install GitPython
+RUN pip3 install pymongo
 
 WORKDIR /app
 
diff --git a/training_sacred.py b/training_sacred.py
index a10dbd9..3d7b184 100755
--- a/training_sacred.py
+++ b/training_sacred.py
@@ -12,14 +12,15 @@ from torch import nn, optim
 import torch.nn.functional as F
 import sys
 from sacred import Experiment
-from sacred.observers import FileStorageObserver
+from sacred.observers import FileStorageObserver, MongoObserver
 
 
 # In[ ]:
 
 
-ex = Experiment("file_observer", save_git_info=False)
-ex.observers.append(FileStorageObserver('my_runs'))
+ex = Experiment(save_git_info=False)
+# ex.observers.append(FileStorageObserver('my_runs'))
+ex.observers.append(MongoObserver(url='mongodb://mongo_user:mongo_password@localhost:27017', db_name='sacred'))
 
 @ex.config
 def my_config():
@@ -111,3 +112,10 @@ def my_main(epochs, _run):
     _run.info["epochs"] = epochs
     _run.info["loss"] = loss
 
+
+    
+# In[ ]:
+
+
+ex.run()
+