From e379116876a1b39282fd6e9eac6ec7529fa0f72e Mon Sep 17 00:00:00 2001
From: Adrian Charkiewicz <riraasaa@gmail.com>
Date: Sun, 8 May 2022 15:36:31 +0200
Subject: [PATCH] epochs no longer missing

---
 pytorch/pytorch.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pytorch/pytorch.py b/pytorch/pytorch.py
index 8d13800..e3a4c6a 100644
--- 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