hnluo
2023-03-29 7dc81334ad28e3d293e1cadb41ba4648b4998ef8
Merge pull request #308 from alibaba-damo-academy/dev_lhn

Dev lhn
1个文件已修改
2 ■■■ 已修改文件
funasr/train/trainer.py 2 ●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/train/trainer.py
@@ -580,7 +580,7 @@
        ):
            assert isinstance(batch, dict), type(batch)
            if rank == 0 and hasattr(model.module, "num_updates"):
            if rank == 0:
                if hasattr(model, "num_updates") or (hasattr(model, "module") and hasattr(model.module, "num_updates")):
                    num_batch_updates = model.get_num_updates() if hasattr(model,"num_updates") else model.module.get_num_updates()
                if (num_batch_updates%batch_interval == 0) and (options.oss_bucket is not None) and options.use_pai: