Imported from https://github.com/junhahyung/bert_transfer