游雁
2024-03-25 64d6ef50403903c1c97c827420a84e9e42756a6a
funasr/datasets/audio_datasets/samplers.py
@@ -212,7 +212,7 @@
    def set_epoch(self, epoch):
        self.epoch = epoch
class CustomDistributedDynamicBatchSampler(Sampler):
class CustomDistributedDynamicBatchSampler(DistributedSampler):
    def __init__(self, dataset,
                 batch_size,
                 num_replicas=None,