diff --git a/train.py b/train.py
index acf1871..ca3ab8e 100644
--- a/train.py
+++ b/train.py
@@ -6,6 +6,7 @@ from keras.applications.densenet import layers
 from sklearn.model_selection import train_test_split
 import tensorflow
 
+EPOCHS = int(os.environ['EPOCHS'])
 
 train_data_x = pandas.read_csv('./X_train.csv')
 
@@ -25,6 +26,6 @@ norm_games_model.compile(
     loss=tensorflow.keras.losses.MeanSquaredError(),
     optimizer=tensorflow.keras.optimizers.Adam())
 
-norm_games_model.fit(games_all, games_predict, epochs=500)
+norm_games_model.fit(games_all, games_predict, epochs=EPOCHS)
 
 norm_games_model.save('test')
\ No newline at end of file