Mark AMP test as flaky (#15055)

2022-10-10 23:26:13 +02:00 · 2022-10-10 23:26:13 +02:00 · 713a2f79ee
parent c334b7766c
commit 713a2f79ee
1 changed files with 1 additions and 0 deletions
--- a/tests/tests_pytorch/models/test_amp.py
+++ b/tests/tests_pytorch/models/test_amp.py
@ -68,6 +68,7 @@ class AMPTestModel(BoringModel):


@RunIf(min_torch="1.10")
+@pytest.mark.flaky(reruns=3)
@pytest.mark.parametrize(
    ("strategy", "precision", "devices"),
    (