diff --git a/pytorch/pytorch.py b/pytorch/pytorch.py
index 5fae4d1..1cca6c4 100644
--- a/pytorch/pytorch.py
+++ b/pytorch/pytorch.py
@@ -21,6 +21,22 @@ import sys
 
 
 # In[2]:
+ex = Experiment(save_git_info=False)
+
+ex.observers.append(FileStorageObserver('my_runs'))
+
+# ex.observers.append(MongoObserver(url='mongodb://mongo_user:mongo_password_IUM_2021@localhost:27017', db_name='sacred'))
+
+try: 
+  numberOfEpochParam = int(sys.argv[1])
+except:
+  # dafault val
+  numberOfEpochParam = 1500
+
+@ex.config
+def my_config():
+  epochs = numberOfEpochParam
+
 
 
 dataframe_raw = pd.read_csv("winequality-red.csv")