epochs run?

pytorch/pytorch.py

@@ -41,6 +41,7 @@ def my_config():
   
 
 
+
 dataframe_raw = pd.read_csv("winequality-red.csv")
 dataframe_raw.head()
 
@@ -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)