diff --git a/run.py b/run.py
index ac9da7f..2c93545 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=10000)
+    model.fit(X_train_scaled, y_train_categorical, epochs=100)
     model.save('mymodel.keras')
     
     print("finished training")