Merge branch 'pr/157' into batch-size

aigamedev · Jan 1, 2016 · 65b6542 · 65b6542
2 parents 7831214 + ac26499
commit 65b6542
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sknn/backend/lasagne/mlp.py b/sknn/backend/lasagne/mlp.py
@@ -261,7 +261,7 @@ def cast(array, indices):
         if shuffle:
             numpy.random.shuffle(indices)
 
-        for start_idx in range(0, total_size - batch_size + 1, batch_size):
+        for start_idx in range(0, total_size, batch_size):
             excerpt = indices[start_idx:start_idx + batch_size]
             Xb, yb, wb = cast(X, excerpt), cast(y, excerpt), cast(w, excerpt)
             yield Xb, yb, wb