diff --git a/pytorch/pytorch.py b/pytorch/pytorch.py
index 8e44a1f..28b0975 100644
--- a/pytorch/pytorch.py
+++ b/pytorch/pytorch.py
@@ -200,7 +200,7 @@ with open("result.txt", "w+") as file:
 
 
 @ex.automain
-def main(epochs, _run):
+def main():
     lr = 1e-6
     #my_config()
     #print("number of epochs is: ", epochs)