diff --git a/pytorch/pytorch.py b/pytorch/pytorch.py
index 8d13800..e3a4c6a 100644
--- a/pytorch/pytorch.py
+++ b/pytorch/pytorch.py
@@ -138,8 +138,8 @@ output_size = len(output_cols)
 # In[10]:
 @ex.config
 def my_config():
-    global epochs
-    epochs = numberOfEpochParam
+    #epochs = numberOfEpochParam
+    epochs = 1500
     lr=lr
     model=model 
     train_loader=train_loader