diff --git a/python/pokedex_ResNet50.py b/python/pokedex_ResNet50.py
index 971ac693cc771b4cc8bb20a47c41ed54c4979304..2c8d413dc489e8e183abcdeeed63d755c2dffb69 100644
--- a/python/pokedex_ResNet50.py
+++ b/python/pokedex_ResNet50.py
@@ -110,7 +110,7 @@ callbacks = [
 model.fit(
     train_ds,
     validation_data=val_ds,
-    epochs=20,
+    epochs=5,
     callbacks=callbacks,
     class_weight=class_weight_dict
 )