Merge pull request carpedm20#294 from petcarerx/patch-1

Update model.py
liuwilbur · Jun 15, 2018 · 245cedc · 245cedc
2 parents f070232 + 5389e23
commit 245cedc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model.py b/model.py
@@ -204,7 +204,7 @@ def train(self, config):
         np.random.shuffle(self.data)
         batch_idxs = min(len(self.data), config.train_size) // config.batch_size
 
-      for idx in xrange(0, batch_idxs):
+      for idx in xrange(0, int(batch_idxs)):
         if config.dataset == 'mnist':
           batch_images = self.data_X[idx*config.batch_size:(idx+1)*config.batch_size]
           batch_labels = self.data_y[idx*config.batch_size:(idx+1)*config.batch_size]