Merge pull request carpedm20#303 from pengwa/fix

move self.data length check to the right place
vraebfdsb · Aug 12, 2018 · 85edbcd · 85edbcd
2 parents 351a654 + 529950a
commit 85edbcd
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):