We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent fa88905 commit 144a5c9Copy full SHA for 144a5c9
benchmarks/test_parity.py
@@ -11,7 +11,7 @@
11
12
@pytest.mark.parametrize('cls_model,max_diff', [
13
(ParityModuleRNN, 0.05),
14
- (ParityModuleMNIST, 0.8)
+ (ParityModuleMNIST, 0.82)
15
])
16
@pytest.mark.skipif(not torch.cuda.is_available(), reason="test requires GPU machine")
17
def test_pytorch_parity(tmpdir, cls_model, max_diff):
0 commit comments