游雁
2024-03-25 1df10311c4e6df252db748293fd47792abae6a61
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,