雾聪
2023-12-13 de87e1d180d214e1f49682d2b5fb7c9d2c89ae7e
funasr/datasets/data_sampler.py
@@ -4,7 +4,7 @@
class BatchSampler(torch.utils.data.BatchSampler):
   
   def __init__(self, dataset, batch_size_type: str="example", batch_size: int=14, sort_size: int=30, drop_last: bool=False, shuffle: bool=True, **kwargs):
   def __init__(self, dataset, batch_size_type: str="example", batch_size: int=100, sort_size: int=30, drop_last: bool=False, shuffle: bool=True, **kwargs):
      
      self.drop_last = drop_last
      self.pre_idx = -1