FUNASR训练
parent: 08392f28 | 补丁 | 提交 | ignore whitespace
zhifu gao
2024-03-27 db82796f95e2f25e71d57df5f047c2354674c47a 
train (#1548)

1个文件已修改
3 ■■■■■ 已修改文件
funasr/train_utils/trainer.py 3 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 
 funasr/train_utils/trainer.py


@@ -377,7 +377,8 @@


                    model=model,


                    dataloader_val=dataloader_val,


                    epoch=epoch,


                    writer=writer


                    writer=writer,


                    step=batch_idx+1,


                )




            if (batch_idx+1) % self.save_checkpoint_interval == 0: