This commit is contained in:
SeanNaren 2020-11-26 23:01:04 +00:00
parent fc9b2bf015
commit 508eaff541
1 changed files with 1 additions and 1 deletions

View File

@ -62,7 +62,7 @@ def test_ddp_sharded_plugin_correctness_amp_multi_gpu():
@pytest.mark.skipif(not os.getenv("PL_RUNNING_SPECIAL_TESTS", '0') == '1',
reason="test should be run outside of pytest")
@DDPLauncher.run("--distributed_backend ddp --gpus 2 --precision 32")
def test_ddp_sharded_plugin_correctness_amp_multi_gpu_ddp(tmpdir, args=None):
def test_ddp_sharded_plugin_correctness_multi_gpu_ddp(tmpdir, args=None):
run_sharded_correctness(gpus=args.gpus, precision=args.precision, accelerator=args.distributed_backend)