From bb212daf0b2823d780c19d167b9cebbfcbf89379 Mon Sep 17 00:00:00 2001
From: vahidrezanezhad <vahid631983@gmail.com>
Date: Tue, 10 Dec 2019 14:01:55 +0100
Subject: [PATCH] Update main.py

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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)