riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 18:51:14 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Merge pull request #2540 from huggingface/torch14_fix

Browse Source 
[PyTorch 1.4] Fix failing torchscript test for xlnet
This commit is contained in:
Thomas Wolf 2020-01-16 13:16:59 +01:00 committed by GitHub
commit 880854846b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/test_modeling_common.py
View File

@ -236,11 +236,14 @@ class ModelTesterMixin:
loaded_model.to(torch_device) loaded_model.to(torch_device)
loaded_model.eval() loaded_model.eval()
model_params = model.parameters() model_state_dict = model.state_dict()
loaded_model_params = loaded_model.parameters() loaded_model_state_dict = loaded_model.state_dict()
self.assertEqual(set(model_state_dict.keys()), set(loaded_model_state_dict.keys()))
models_equal = True models_equal = True
for p1, p2 in zip(model_params, loaded_model_params): for layer_name, p1 in model_state_dict.items():
p2 = loaded_model_state_dict[layer_name]
if p1.data.ne(p2.data).sum() > 0: if p1.data.ne(p2.data).sum() > 0:
models_equal = False models_equal = False