Yabin Li
2024-01-17 dcc310e1a8ff75df5b8d091a9e26883b85b8afec
funasr/datasets/audio_datasets/samplers.py
@@ -20,7 +20,7 @@
        self.dataset = dataset
        self.total_samples = len(dataset)
        self.batch_type = batch_type
        self.batch_size = batch_size
        self.batch_size = int(batch_size)
        self.buffer_size = buffer_size
        self.max_token_length = kwargs.get("max_token_length", 5000)
        self.shuffle_idx = np.arange(self.total_samples)