funasr/datasets/audio_datasets/espnet_samplers.py
@@ -48,10 +48,10 @@ except: rank = 0 num_replicas = 1 if rank_split: logging.info(f"Warning, rank_split: {rank_split}, batch and shuffle data in local rank") rank = 0 num_replicas = 1 # if rank_split: # logging.info(f"Warning, rank_split: {rank_split}, batch and shuffle data in local rank") # rank = 0 # num_replicas = 1 self.rank = rank self.num_replicas = num_replicas self.dataset = dataset
