move self.data length check to the right place

carpedm20 · Jul 17, 2018 · 529950a · 529950a
1 parent 351a654
commit 529950a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/model.py b/model.py
@@ -86,11 +86,11 @@ def __init__(self, sess, input_height=108, input_width=108, crop=True,
       else:
         self.c_dim = 1
 
+      if len(self.data) < self.batch_size:
+        raise Exception("[!] Entire dataset size is less than the configured batch_size")
+
     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):