main without parameters

2022-05-07 23:57:11 +02:00 · 2022-05-07 23:57:11 +02:00 · 5b652011c5
commit 5b652011c5
parent bea443eb78
1 changed files with 1 additions and 1 deletions
--- 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)