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,
