epochs no longer missing

2022-05-08 15:36:31 +02:00 · 2022-05-08 15:36:31 +02:00 · e379116876
commit e379116876
parent 044740662e
1 changed files with 2 additions and 2 deletions
--- a/pytorch/pytorch.py
+++ b/pytorch/pytorch.py
@ -138,8 +138,8 @@ output_size = len(output_cols)
 # In[10]:
@ex.config
 def my_config():
-    global epochs
-    epochs = numberOfEpochParam
+    #epochs = numberOfEpochParam
+    epochs = 1500
    lr=lr
    model=model 
    train_loader=train_loader