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