diff --git a/tests/nn/data_parallel/test_fsdp.py b/tests/nn/data_parallel/test_fsdp.py index e67c65fe0..d0df6dc60 100644 --- a/tests/nn/data_parallel/test_fsdp.py +++ b/tests/nn/data_parallel/test_fsdp.py @@ -481,7 +481,6 @@ def _get_model(self, group, config): return FullyShardedDataParallel(model, group, **config) -@pytest.mark.skip(reason="Recently flaky and not reproducible locally.") class TestSerialization(DistributedTest): @parameterized.expand([[False, False], [True, False], [True, True], [False, True]], name_func=rename_test) def test_pickle(self, mixed_precision, cpu_offload):