From 18b020594beb10acafd0658bf6f97b12d853ca46 Mon Sep 17 00:00:00 2001
From: Elvis Sun <elvissunlihe@gmail.com>
Date: Sun, 14 Jan 2018 00:00:29 -0800
Subject: [PATCH] mini-batch with generator

---
 main.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/main.py b/main.py
index d1fdb05..ac81fb6 100644
--- 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,