首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >AutoModelForSequenceClassification和AutoModel的区别是什么?

AutoModelForSequenceClassification和AutoModel的区别是什么?
EN

Stack Overflow用户
提问于 2021-11-10 03:33:55
回答 1查看 4K关注 0票数 6

我们可以通过AutoModel(TFAutoModel)函数创建一个模型:

代码语言:javascript
复制
from transformers import AutoModel 
model = AutoModel.from_pretrained('distilbert-base-uncase')

另一方面,AutoModelForSequenceClassification(TFAutoModelForSequenceClassification):创建了一个模型

代码语言:javascript
复制
from transformers import AutoModelForSequenceClassification
model = AutoModelForSequenceClassification('distilbert-base-uncase')

据我所知,两种模型都使用蒸馏塔基-uncase库来创建模型。根据方法的名称,为序列分类创建了第二个类( AutoModelForSequenceClassification )。

但是,在两个类中,真正的区别是什么?以及如何正确地使用它们?

(我在拥抱的脸上搜寻,但还不清楚)

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2021-12-05 09:07:24

AutoModel模型和AutoModelForSequenceClassification模型的不同之处在于,AutoModelForSequenceClassification在模型输出上有一个分类头,可以很容易地用基本模型进行训练。

票数 5
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/69907682

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档