FUNASR训练
parent: 8ba12df6 | 补丁 | 提交 | ignore whitespace
游雁
2024-06-07 870fa4a37a194fb7919b53930064129b6d616cb7 
auto frontend
1个文件已修改
2 ■■■■■ 已修改文件
funasr/datasets/audio_datasets/samplers.py 2 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 
 funasr/datasets/audio_datasets/samplers.py


@@ -417,7 +417,7 @@


            rank_batches[i % self.num_replicas].append(batch)




        # Assign all batches for the current rank directly


        final_batches = rank_batches[self.rank][self.start_step :]


        final_batches = rank_batches[self.rank]  # [self.start_step :]


        self.batch_num = len(final_batches)




        logging.info(