游雁
2024-03-27 3d19549d2ee2429378e80b9bf338ba3b8381e625
bugfix seg_dict_file
1个文件已修改
2 ■■■ 已修改文件
funasr/train_utils/trainer.py 2 ●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/train_utils/trainer.py
@@ -134,7 +134,7 @@
                "val_acc_step_or_eoch": self.val_acc_step_or_eoch,
                "val_loss_step_or_eoch": self.val_loss_step_or_eoch,
                "best_step_or_epoch": self.best_step_or_epoch,
                "avg_keep_nbest_models_type": slef.avg_keep_nbest_models_type,
                "avg_keep_nbest_models_type": self.avg_keep_nbest_models_type,
            }
            if hasattr(model, "module"):
                state["state_dict"] = model.module.state_dict()