up (#10771)

f20d75a1 · Patrick von Platen · GitHub · c83fbc5f · f20d75a1
Unverified Commit f20d75a1 authored Mar 17, 2021 by Patrick von Platen Committed by GitHub Mar 17, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/test_modeling_prophetnet.py tests/test_modeling_prophetnet.py +1 -1

No files found.
--- a/tests/test_modeling_prophetnet.py
+++ b/tests/test_modeling_prophetnet.py
@@ -482,7 +482,7 @@ class ProphetNetModelTester:
            torch.allclose(
                outputs_no_mask.last_hidden_state_ngram[0, :5, 0],
                outputs_with_mask.last_hidden_state_ngram[0, :5, 0],
-                atol=1e-3,
+                atol=1e-2,
            )
        )