Skip to content

Commit

Permalink
Merge branch 'master' into automatic_optimization
Browse files Browse the repository at this point in the history
  • Loading branch information
Jeff Yang authored Nov 14, 2020
2 parents 1ce9115 + 144a5c9 commit 27bdd7a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion benchmarks/test_parity.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit 27bdd7a

Please sign in to comment.