fix

Akito-UzukiP · Sep 2, 2023 · 6173b3e · 6173b3e
1 parent 4d5c845
commit 6173b3e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/data_utils.py b/data_utils.py
@@ -285,6 +285,8 @@ def __iter__(self):
         for i in range(len(self.buckets)):
             bucket = self.buckets[i]
             len_bucket = len(bucket)
+            if (len_bucket == 0):
+                continue
             ids_bucket = indices[i]
             num_samples_bucket = self.num_samples_per_bucket[i]