def test_run_on_amp_ddp(): train_experiment(dl.DistributedDataParallelEngine(fp16=True))
def test_classification_on_torch_ddp(): train_experiment(None, dl.DistributedDataParallelEngine())
def test_run_on_torch_ddp(): train_experiment(dl.DistributedDataParallelEngine())