def test_run_on_amp_ddp():
    train_experiment(dl.DistributedDataParallelEngine(fp16=True))
Пример #2
0
def test_classification_on_torch_ddp():
    train_experiment(None, dl.DistributedDataParallelEngine())
def test_run_on_torch_ddp():
    train_experiment(dl.DistributedDataParallelEngine())