Imported from https://github.com/mkavim/finetune_bert