fix

GreenWizard2015 · Jul 2, 2024 · 84b5493 · 84b5493
1 parent d2f1c89
commit 84b5493
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/Core/CDatasetLoader.py b/Core/CDatasetLoader.py
@@ -57,8 +57,7 @@ def __init__(self, folder, samplerArgs, sampling, stats):
       for k, size in validSamples.items():
         # all datasets should have the same number of samples represented in the indices
         # so that the sampling is uniform
-        chunk = np.full((maxSize, ), k, dtype=np.uint32) % size
-        chunk = chunk.astype(dtype)
+        chunk = np.full((maxSize, ), k, dtype=dtype)
         chunks.append(chunk)
         continue
       self._indices = np.concatenate(chunks)