本地加载hugging face模型:Bert

找了个hf的镜像站,把config.json和pytorch_model.bin两个文件进行下载下来,模型文件uncased_L-12_H-768_A-12.zip下载下来先。

解压模型文件压缩包,把前面下载的两个文件也放进去,总共6个文件。这个文件夹就是代码

tokenizer = BertTokenizer.from_pretrained(BERT_PATH)

BERT_PATH这里对应的文件路径。

Some weights of BertForSequenceClassification were not initialized from the model checkpoint at bert-base-uncased and are newly initialized: ['classifier.bias', 'classifier.weight']

pip install tensorflow

复制代码
from transformers import BertModel, BertTokenizer

model_name = 'bert-base-uncased'
tokenizer = BertTokenizer.from_pretrained(model_name)
model = BertModel.from_pretrained(model_name, from_tf=True)
相关推荐
一颗青果几秒前
传统机器学习(浅层模型)vs 深度学习(深度神经网络)
人工智能
深念Y1 分钟前
从张量到微分方程:AI计算架构的底层思考笔记
深度学习·架构·张量·npu·计算机架构·ai芯片·计算范式
wukangjupingbb6 分钟前
AI制药哲学:需区分“AI辅助研发“与“原生AI驱动研发“
人工智能
带娃的IT创业者7 分钟前
全局工作空间——大脑的“黑板“
人工智能·深度学习·神经网络·科普·技术分享·ai架构·全局工作空间
KC2708 分钟前
大模型提示词注入攻击与防御:当你的 AI 开始“不听话“
人工智能·安全·aigc
H Journey9 分钟前
OpenCV之Canny边缘检测
人工智能·opencv·计算机视觉
芯智工坊9 分钟前
第8章 Mosquitto消息高级特性
网络·人工智能·mqtt·开源·ssl
观远数据10 分钟前
未来3年企业数据分析的核心:基于自然语言的AI优先决策体系如何搭建
数据库·人工智能·数据分析
zhengyquan10 分钟前
微软砸1.6万亿日元布局日本AI!
人工智能·microsoft
Dev7z11 分钟前
基于MATLAB与SVM实现河道水面漂浮物的自动检测与识别
人工智能·支持向量机·matlab