This commit is contained in:
游雁 2024-06-12 19:27:35 +08:00
parent e4a69d4768
commit 2ac79cd3f3

View File

@ -319,7 +319,7 @@ class Trainer:
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]