Merge pull request #295 from spacemunkay/error-data-batch-size

Raise error if dataset size less than batch_size

Author: carpedm20

model.py

@@ -88,6 +88,9 @@ def __init__(self, sess, input_height=108, input_width=108, crop=True,
 
     self.grayscale = (self.c_dim == 1)
 
+    if len(self.data) < self.batch_size:
+      raise Exception("[!] Entire dataset size is less than the configured batch_size")
+
     self.build_model()
 
   def build_model(self):