Fix check in load_from_full_state_dict for modified state dicts
#5006
| Job | Run time |
|---|---|
| 37s | |
| 37s |
load_from_full_state_dict for modified state dicts
#5006
| Job | Run time |
|---|---|
| 37s | |
| 37s |