From 087df12aa76633ce312fe20dcb87a4bc73042c69 Mon Sep 17 00:00:00 2001
From: Kornelia Girejko <korgir@st.amu.edu.pl>
Date: Fri, 6 May 2022 20:09:09 +0200
Subject: [PATCH] Zaktualizuj 'DL.py'

---
 DL.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/DL.py b/DL.py
index ff3ab2f..6854313 100644
--- a/DL.py
+++ b/DL.py
@@ -8,6 +8,8 @@ from sklearn import preprocessing
 from sklearn.preprocessing import StandardScaler
 scaler = StandardScaler()
 
+EPOCHS = int(sys.argv[1])
+
 # Model
 class Model(nn.Module):
     def __init__(self):