diff --git a/run.py b/run.py
index 2e75642..69747f8 100644
--- a/run.py
+++ b/run.py
@@ -71,7 +71,7 @@ with tf.device('/GPU:0'):
     model.compile(optimizer='SGD',
                   loss='categorical_crossentropy',
                   metrics=['accuracy'])
-    model.fit(X_train_scaled, y_train_categorical, epochs=15)
+    model.fit(X_train_scaled, y_train_categorical, epochs=20)
     model.save('mymodel.keras')
     
     print("finished training")