diff --git a/run.py b/run.py
index 83de8ef..4b6cf28 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=50)
+    model.fit(X_train_scaled, y_train_categorical, epochs=10)
     model.save('mymodel.keras')
     
     print("finished training")