reduce parity test

2020-11-27 18:50:01 +00:00 · 2020-11-27 18:50:01 +00:00 · b36b9a0145
parent ba41733802
commit b36b9a0145
1 changed files with 1 additions and 1 deletions
--- a/benchmarks/test_parity.py
+++ b/benchmarks/test_parity.py
@ -12,7 +12,7 @@ from tests.base.models import ParityModuleMNIST, ParityModuleRNN
 # ParityModuleMNIST runs with num_workers=1
@pytest.mark.parametrize('cls_model,max_diff', [
    (ParityModuleRNN, 0.05),
-    (ParityModuleMNIST, 0.22)
+    (ParityModuleMNIST, 0.18)
 ])
@pytest.mark.skipif(not torch.cuda.is_available(), reason="test requires GPU machine")
 def test_pytorch_parity(tmpdir, cls_model, max_diff):