使用拥抱面变换器加载TensorFlow .pb模型

332nm8kg 于 2023-10-23 发布在其他

关注(0)|答案(1)|浏览(112)

我想加载一个训练后保存的模型。原始模型是在TensorFlow上构建的，saved_model路径包含.pb文件：

assets/
fingerprint.pb
keras_metadata.pb
saved_model.pb
variables/

我的目标是使用Hugging Face transformers库加载这个模型。我读到AutoModel是可能的，但我还没有弄清楚如何实现这一点。

来源：https://stackoverflow.com/questions/76965928/load-a-tensorflow-pb-model-using-hugging-face-transformers

1条答案

按热度按时间

你可以尝试以下方法：

from transformers import TFAutoModel

model = TFAutoModel.from_pretrained("<path_to_saved_pretrained_model>")

赞(0）回复(0）举报 2023-10-23

相关问题

热门标签

Java query python Node 开发语言 request Util 数据库 Table 后端算法 Logger Message Element Parser

最新问答

xxl-job 安全组扫描到执行器端口服务存在信息泄露漏洞
回答(1) 发布于 5个月前
xxl-job 不能和nacos兼容？
回答(3) 发布于 5个月前
xxl-job 任务执行完后无法结束，日志一直转圈
回答(3) 发布于 5个月前
xxl-job-admin页面上查看调度日志样式问题
回答(1) 发布于 5个月前
xxl-job 参数512字符限制能否去掉
回答(1) 发布于 5个月前