游雁
2024-06-09 8bb997175304f7d11ab403a3705d3242c0f2d28e
fix bug
1个文件已修改
2 ■■■ 已修改文件
funasr/datasets/audio_datasets/samplers.py 2 ●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/datasets/audio_datasets/samplers.py
@@ -431,7 +431,7 @@
        self.batch_num = len(final_batches)
        logging.info(
            f"rank: {self.rank}, dataloader start from step: {self.start_step}, batch_num: {rank_batches[self.rank]}, after: {self.batch_num}"
            f"rank: {self.rank}, dataloader start from step: {self.start_step}, batch_num: {len(rank_batches[self.rank])}, after: {self.batch_num}"
        )
        return iter(final_batches)