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

up (#10771)

Browse Source
This commit is contained in:
Patrick von Platen
2021-03-17 16:15:14 +03:00
committed by GitHub
parent c83fbc5f2d
commit f20d75a13f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_modeling_prophetnet.py
View File

@@ -482,7 +482,7 @@ class ProphetNetModelTester:
torch.allclose( torch.allclose(
outputs_no_mask.last_hidden_state_ngram[0, :5, 0], outputs_no_mask.last_hidden_state_ngram[0, :5, 0],
outputs_with_mask.last_hidden_state_ngram[0, :5, 0], outputs_with_mask.last_hidden_state_ngram[0, :5, 0],
atol=1e-3, atol=1e-2,
) )
) )