Skip to content

GitLab

Unverified Commit f20d75a1 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

up (#10771)

parent c83fbc5f
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_modeling_prophetnet.py
View file @ f20d75a1
...@@ -482,7 +482,7 @@ class ProphetNetModelTester: ...@@ -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,
) )
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment