funasr/train_utils/trainer_ds.py
@@ -313,12 +313,13 @@ state_dict = model.state_dict() if self.effective_save_name_excludes is not None: logging.info(f"effective_save_name_excludes: {self.effective_save_name_excludes}") dst_state_dict = {} for k in state_dict.keys(): for k_ex in self.effective_save_name_excludes: k_tmp = k.replace("module.", "") if k.startswith(k_ex): logging.info(f"key: {{k}} matching: {k_ex}, not save it") logging.info(f"key: {k} matching: {k_ex}, not save it") break else: dst_state_dict[k] = state_dict[k]
