From 144a5c9913fbda8ffdf6a5731b70edf50b09bf8e Mon Sep 17 00:00:00 2001 From: Justus Schock <12886177+justusschock@users.noreply.github.com> Date: Sat, 14 Nov 2020 05:03:30 +0100 Subject: [PATCH] Increase parity to match logging refactor (#4651) Co-authored-by: Jeff Yang Co-authored-by: chaton Co-authored-by: Jirka Borovec --- benchmarks/test_parity.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/test_parity.py b/benchmarks/test_parity.py index d54742489a62c..ce1c2eae106ad 100644 --- a/benchmarks/test_parity.py +++ b/benchmarks/test_parity.py @@ -11,7 +11,7 @@ @pytest.mark.parametrize('cls_model,max_diff', [ (ParityModuleRNN, 0.05), - (ParityModuleMNIST, 0.8) + (ParityModuleMNIST, 0.82) ]) @pytest.mark.skipif(not torch.cuda.is_available(), reason="test requires GPU machine") def test_pytorch_parity(tmpdir, cls_model, max_diff):