From 7dd45b9c755959f6f348e6c657035824b28bf1ba Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sebastian=20Wa=C5=82=C4=99sa?= <sebwal1@st.amu.edu.pl>
Date: Fri, 6 May 2022 20:17:41 +0200
Subject: [PATCH] add EPOCHS variable

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

diff --git a/ml_pytorch.py b/ml_pytorch.py
index 826b006..cd6b55c 100644
--- a/ml_pytorch.py
+++ b/ml_pytorch.py
@@ -22,7 +22,7 @@ import sys
 
 # In[2]:
 
-
+EPOCHS = int(sys.argv[1])
 #load data
 dataframe = pd.read_csv("understat.csv")
 
@@ -124,7 +124,7 @@ model=Model_xPosition()
 # In[11]:
 
 
-epochs = 1000
+epochs = EPOCHS
 lr = 1e-5
 learning_proccess = fit(epochs, lr, model, train_loader, val_loader)