diff --git a/pytorch/pytorch.py b/pytorch/pytorch.py
index 1cca6c4..6b27d95 100644
--- a/pytorch/pytorch.py
+++ b/pytorch/pytorch.py
@@ -18,7 +18,9 @@ from torch.utils.data import DataLoader, TensorDataset, random_split
 import random
 import os
 import sys
-
+from sacred import Experiment
+from sacred.observers import FileStorageObserver
+from sacred.observers import MongoObserver
 
 # In[2]:
 ex = Experiment(save_git_info=False)