Raise err if dataset size less than batch_size

Author: spacemunkay

model.py

@@ -85,6 +85,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):