conventional_training train.py The common training pipeline. train_amp.py The mixed precision training pipeline. You should install the apex firstly.