Exemple #1
0
def test_classification_on_amp_dp():
    train_experiment(None, dl.DataParallelAMPEngine())
Exemple #2
0
def test_finetune_on_amp_dp():
    train_experiment(None, dl.DataParallelAMPEngine())
def test_distillation_on_amp_dp():
    train_experiment(None, dl.DataParallelAMPEngine())