游雁
2024-06-09 1186cd96a5a8fa3466c5e3e41f86f0280deb1410
fix bug
1个文件已修改
5 ■■■■ 已修改文件
funasr/datasets/audio_datasets/samplers.py 5 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/datasets/audio_datasets/samplers.py
@@ -399,10 +399,7 @@
                    continue
                sample_length = 1 if self.batch_type == "example" else original_sample_length
                potential_batch_length = max(max_len_in_batch, sample_length) * (len(batch) + 1)
                if (
                    potential_batch_length <= self.batch_size
                    and count <= self.batch_size_sample_max
                ):
                if potential_batch_length <= self.batch_size and count < self.batch_size_sample_max:
                    batch.append(idx)
                    max_len_in_batch = max(max_len_in_batch, sample_length)
                    count += 1