Python DataParallelAMPEngine Examples

Example #1

def test_classification_on_amp_dp():
    train_experiment(None, dl.DataParallelAMPEngine())

Example #2

def test_finetune_on_amp_dp():
    train_experiment(None, dl.DataParallelAMPEngine())

Example #3

File: test_distillation.py Project: Podidiving/catalyst

def test_distillation_on_amp_dp():
    train_experiment(None, dl.DataParallelAMPEngine())