funasr/datasets/audio_datasets/samplers.py
@@ -392,7 +392,7 @@ ) batch = [] max_len_in_batch = 0 count = 0 count = 1 for idx in buffer: original_sample_length = self.dataset.get_source_len(idx) if original_sample_length > self.max_token_length: @@ -410,7 +410,7 @@ buffer_batches.append(batch) batch = [idx] max_len_in_batch = sample_length count = 0 count = 1 if batch: buffer_batches.append(batch)
