no epochs in main

2022-05-08 14:46:39 +02:00 · 2022-05-08 14:46:39 +02:00 · 4893c15bb5
commit 4893c15bb5
parent 5b652011c5
1 changed files with 1 additions and 1 deletions
--- a/pytorch/pytorch.py
+++ b/pytorch/pytorch.py
@ -204,6 +204,6 @@ def main():
    lr = 1e-6
    #my_config()
    #print("number of epochs is: ", epochs)
-    history5 = fit(epochs, lr, model, train_loader, val_loader)
+    history5 = fit(lr, model, train_loader, val_loader)

 #ex.run()