Bert预训练模型的使用

pytorch_pretrained_bert的配置使用

pytorch_pretrained_bert

https://github.com/huggingface/transformers

  • 安装加载预训练模型和权重的包：pip install pytorch-pretrained-bert

  • 修改源码

    从亚马逊站点下载非常慢，可以将源码中的地址更换为本地文件。

    1. pytorch_pretrained_bert/modeling.py的 40-51行

    image-20201207210830430

    1. pytorch_pretrained_bert/tokenization.py的30-41行

image-20201207210928928

版权声明：本文为qq_39827677原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。