A copy from https://github.com/huggingface/pytorch-pretrained-BERT - View it on GitHub