Imported from https://github.com/lonePatient/ERNIE-text-classification-pytorch