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
