diff --git a/pytorch/pytorch.py b/pytorch/pytorch.py
index e7690d5..2367513 100644
--- a/pytorch/pytorch.py
+++ b/pytorch/pytorch.py
@@ -37,6 +37,7 @@ ex.observers.append(FileStorageObserver('my_runs'))
 
 @ex.config
 def my_config():
+    global epochs
     epochs = 1500