SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix loading the best model on the last stage of training (#11718)

Browse Source
This commit is contained in:
Volodymyr Byno
2021-05-13 23:11:12 +03:00
committed by GitHub
parent 252082001d
commit 218d552f30
2 changed files with 15 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_modeling_common.py
View File

@@ -180,7 +180,8 @@ class ModelTesterMixin:
# Test we can load the state dict in the model, necessary for the checkpointing API in Trainer.
load_result = model.load_state_dict(state_dict_saved, strict=False)
self.assertTrue(
len(load_result.missing_keys) == 0 or load_result.missing_keys == model._keys_to_ignore_on_save
len(load_result.missing_keys) == 0
or set(load_result.missing_keys) == set(model._keys_to_ignore_on_save)
)
self.assertTrue(len(load_result.unexpected_keys) == 0)