This commit is contained in:
游雁 2024-06-12 20:03:55 +08:00
parent f6cae2b48b
commit 508ad6573a

View File

@ -96,7 +96,7 @@ def load_pretrained_model(
dst_state[k] = src_state[k_src]
else:
logging.info(f"Warning, miss key in ckpt: {k}, mapped: {k_src}")
print(f"Warning, miss key in ckpt: {k}")
flag = obj.load_state_dict(dst_state, strict=True)
logging.info(f"Loading ckpt: {path}, status: {flag}")