github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-24 14:33:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge 3596bd60e7 into 46ed5d856b

Browse Source
This commit is contained in:
Kimbing Ng 2025-04-29 11:26:23 +00:00 committed by GitHub
commit c7f9612630
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
colossalai/checkpoint_io/utils.py
View File

@ -729,6 +729,10 @@ def load_state_dict_into_model(
missing_keys = missing_keys.append(sub_missing_keys)
if strict:
if len(error_msgs) > 0:
raise RuntimeError(
"Error(s) in loading state_dict for {}:\n\t{}".format(model.__class__.__name__, "\n\t".join(error_msgs))
)
if len(unexpected_keys) > 0:
error_msgs = [
"Unexpected key(s) in state_dict: {}. ".format(", ".join('"{}"'.format(k) for k in unexpected_keys))