mini-batch with generator

elvisun · Jan 14, 2018 · 18b0205 · 18b0205
1 parent 205dd95
commit 18b0205
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/main.py b/main.py
@@ -103,6 +103,7 @@ def train(self):
 
         self.build_model()
 
+        print("training with epochs of: " number_of_epoch)
         self.model.fit_generator(self.generate_batch(),
           steps_per_epoch=MINI_BATCH_SIZE,
           epochs=number_of_epoch,