Ejemplos de DataParallelEngine en Python

Ejemplo n.º 1

Archivo: test_finetune2.py Proyecto: wwchung91/catalyst

def test_finetune2_on_torch_dp():
    train_experiment(None, dl.DataParallelEngine())

Ejemplo n.º 2

Archivo: test_mnist_multimodel.py Proyecto: catalyst-team/catalyst

def test_run_on_amp_dp():
    train_experiment(dl.DataParallelEngine(fp16=True))

Ejemplo n.º 3

Archivo: test_distillation.py Proyecto: zkid18/catalyst

def test_distillation_on_torch_dp():
    train_experiment(None, dl.DataParallelEngine())

Ejemplo n.º 4

Archivo: test_mnist_multimodel.py Proyecto: catalyst-team/catalyst

def test_run_on_torch_dp():
    train_experiment(dl.DataParallelEngine())