tests/test_modeling_gpt2.py · f5b5c5bd7e213dea1645f07902b681f88e3cf954 · chenpangpang / transformers

Avoid unnecessary warnings when loading pretrained model (#5922) · f5b5c5bd

Sylvain Gugger authored Jul 23, 2020

* Avoid unnecessary warnings when loading pretrained model

* Fix test

* Add other keys to ignore

* keys_to_ignore_at_load -> authorized_missing_keys

f5b5c5bd

test_modeling_gpt2.py 15 KB

Replace test_modeling_gpt2.py