diff --git a/sknn/backend/lasagne/mlp.py b/sknn/backend/lasagne/mlp.py index 6cbabf4..08a02fb 100644 --- a/sknn/backend/lasagne/mlp.py +++ b/sknn/backend/lasagne/mlp.py @@ -252,7 +252,7 @@ def cast(array): 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]), None if w is not None: