Avoid unnecessary warnings when loading pretrained model (#5922)

* Avoid unnecessary warnings when loading pretrained model

* Fix test

* Add other keys to ignore

* keys_to_ignore_at_load -> authorized_missing_keys
This commit is contained in:
Sylvain Gugger
2020-07-23 18:13:36 -04:00
committed by GitHub
parent 29afb5764f
commit f5b5c5bd7e
5 changed files with 16 additions and 1 deletions

View File

@@ -311,6 +311,7 @@ class GPT2ModelTest(ModelTesterMixin, unittest.TestCase):
all_generative_model_classes = (
(GPT2LMHeadModel,) if is_torch_available() else ()
) # TODO (PVP): Add Double HeadsModel when generate() function is changed accordingly
test_missing_keys = False
def setUp(self):
self.model_tester = GPT2ModelTester(self)