Imported from https://github.com/codertimo/BERT-pytorch