fix bug

游雁

2024-06-09 56986acaa78b0ea105d6b136905ecf6e8379cd11

fix bug

 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)