diff --git a/pytorch/pytorch.py b/pytorch/pytorch.py
index 6ef38f1..650d030 100644
--- a/pytorch/pytorch.py
+++ b/pytorch/pytorch.py
@@ -166,25 +166,6 @@ def fit(epochs, lr, model, train_loader, val_loader, _run, opt_func=torch.optim.
     return history
 
 
-# In[12]:
-
-try: 
-  numberOfEpochParam = int(sys.argv[1])
-except:
-  # dafault val
-  numberOfEpochParam = 1500
-
-
-@ex.config
-def my_config():
-  epochs = numberOfEpochParam
-
-
-
-
-
-# In[27]:
-
 
 def predict_single(input, target, model):
     inputs = input.unsqueeze(0)