栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

BERT学习---01.tensorflow的bert转换为pytorch的bert

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

BERT学习---01.tensorflow的bert转换为pytorch的bert

参考:https://blog.csdn.net/yaohaishen/article/details/117467889
1.下载uncase-base的bert模型

2.点进去后下载一个你需要的模型

3.如图所示

4.创建一个.py文件

5.代码如下
本人参考的链接,直接点击就可以
https://blog.csdn.net/yaohaishen/article/details/117467889
6.将.py文件和下载的模型放在一个文件夹下

7.命令行(如:anaconda中)运行.py文件
python convert_tf_checkpoint_to_pytorch.py --tf_checkpoint_path=BERT_base_DIR/bert_model.ckpt --bert_config_file=BERT_base_DIR/bert_config.json --pytorch_dump_path=BERT_base_DIR/pytorch_model.bin
(复制到记事本中,要修改为一行)
8.运行成功可以得到这样一个文件

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/488737.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号