allow equal stride and dimsize

Cindy Chiao · web-flow · commit 5ab2284c3e05 · 2021-06-07T11:56:33.000-07:00
The current assertion breaks if we want to get 1 sample per batch (where the input dims and output dims are identical). Changing &lt; to &lt;= to allow such cases.
diff --git a/xbatcher/generators.py b/xbatcher/generators.py
@@ -19,7 +19,7 @@ def _slices(dimsize, size, overlap=0):
     slices = []
     stride = size - overlap
     assert stride > 0
-    assert stride < dimsize
+    assert stride <= dimsize
     for start in range(0, dimsize, stride):
         end = start + size
         if end <= dimsize:
