Hugging H5加载模型错误:在配置文件中找不到模型
Hugging H5加载模型错误:在配置文件中找不到模型
提问于 2021-09-27 08:25:54
我正在尝试从Hugging加载模型，我从这里下载了h5模型：https://huggingface.co/distilbert-base-uncased-finetuned-sst-2-english/tree/main
from flask import Flask, jsonify, request # import objects from the Flask model
from keras.models import load_model
from transformers import AutoTokenizer, AutoModelForSequenceClassification,TextClassificationPipeline
model = load_model('./tf_model.h5') # trying to load model here然后错误就出现了：
File "C:\D\Learning\Flask\flask-pp-rest\main.py", line 11, in <module>
model = load_model('./tf_model.h5') File "C:\Users\ndrez\AppData\Local\Programs\Python\Python39\lib\site-packages\keras\saving\save.py",
line 200, in load_model
return hdf5_format.load_model_from_hdf5(filepath, custom_objects, File
"C:\Users\ndrez\AppData\Local\Programs\Python\Python39\lib\site-packages\keras\saving\hdf5_format.py",
line 176, in load_model_from_hdf5
raise ValueError('No model found in config file.') ValueError: **No model found in config file.**有人知道怎么解决这个问题吗？如果你知道，请帮帮我。我将关注这个问题，并尝试实现您的解决方案的答案。
回答 2
Stack Overflow用户
回答已采纳
发布于 2021-09-27 19:04:42
您可以使用TFAutoModelForSequenceClassification类加载distilbert-base-uncased-finetuned-sst-2-english的tensorflow版本：
from transformers import AutoTokenizer, TFAutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
model = TFAutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")Stack Overflow用户
发布于 2021-09-27 08:33:06
要加载您指定的模型，代码如下：
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
model = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/69343395
复制相关文章
相似问题
腾讯云开发者
Copyright © 2013 - 2026 Tencent Cloud. All Rights Reserved. 腾讯云 版权所有
深圳市腾讯计算机系统有限公司 ICP备案/许可证号：粤B2-20090059 
粤公网安备44030502008569号
腾讯云计算（北京）有限责任公司 京ICP证150476号 | 京ICP备11018762号