From f20d75a13f9545b0352ef93813d209707228a427 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Wed, 17 Mar 2021 16:15:14 +0300 Subject: [PATCH] up (#10771) --- tests/test_modeling_prophetnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_modeling_prophetnet.py b/tests/test_modeling_prophetnet.py index 91ea9f2c2b..7314f6f414 100644 --- 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, ) )