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