diff --git a/pytorch/pytorch.py b/pytorch/pytorch.py
index 650d030..faddc97 100644
--- a/pytorch/pytorch.py
+++ b/pytorch/pytorch.py
@@ -38,6 +38,7 @@ ex.observers.append(FileStorageObserver('my_runs'))
 @ex.config
 def my_config():
   epochs = 1500
+  
 
 
 
@@ -195,7 +196,7 @@ with open("result.txt", "w+") as file:
 
 
 @ex.main
-def main():
+def main(epochs, _run):
     lr = 1e-6
     my_config()
     print("number of epochs is: ", epochs)