Imported from https://github.com/qbxlvnf11/pretrained-language-model-BERT