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):