Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Mark AMP test as flaky (#15055)

This commit is contained in:
Carlos Mocholí 2022-10-10 23:26:13 +02:00 committed by GitHub
parent c334b7766c
commit 713a2f79ee
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/tests_pytorch/models/test_amp.py
View File

@ -68,6 +68,7 @@ class AMPTestModel(BoringModel):
@RunIf(min_torch="1.10") @RunIf(min_torch="1.10")
@pytest.mark.flaky(reruns=3)
@pytest.mark.parametrize( @pytest.mark.parametrize(
("strategy", "precision", "devices"), ("strategy", "precision", "devices"),
( (