diff --git a/benchmarks/test_parity.py b/benchmarks/test_parity.py index d54742489a..ce1c2eae10 100644 --- a/benchmarks/test_parity.py +++ b/benchmarks/test_parity.py @@ -11,7 +11,7 @@ from tests.base.models import ParityModuleMNIST, ParityModuleRNN @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):