Imported from https://github.com/circlePi/BERT_Chinese_Text_Class_By_pytorch