diff --git a/train.py b/train.py
index 07c7418..baeb847 100644
--- a/train.py
+++ b/train.py
@@ -169,7 +169,7 @@ def run(n_classes,n_epochs,input_height,
     
     model.fit_generator(
         train_gen,
-        steps_per_epoch=int(len(os.listdir(dir_flow_train_imgs))/n_batch),
+        steps_per_epoch=int(len(os.listdir(dir_flow_train_imgs))/n_batch)-1,
         validation_data=val_gen,
         validation_steps=1,
         epochs=n_epochs)