From 6173b3e37bd4cbbd95158dad9c1df55d359670ea Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <star_dust_chen@foxmail.com>
Date: Sat, 2 Sep 2023 14:44:17 +0800
Subject: [PATCH] fix

---
 data_utils.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/data_utils.py b/data_utils.py
index e847418..713b8ac 100644
--- a/data_utils.py
+++ b/data_utils.py
@@ -285,6 +285,8 @@ class DistributedBucketSampler(torch.utils.data.distributed.DistributedSampler):
         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]