创建model的方法
automodel创建预训练模型
使用automodel直接创建,使用的是hug face官网上的预训练好的模型参数,完成创建后再使用自己本地的数据集进行迁移训练
automodel api 去官网上下载用户指定类型模型的config file 和model file,config说明了如何创建模型,model包括该模型的参数。
automodel api首先根据config去创建model类,再实例化model并随机初始化其参数
最后采用model file里面的参数来修改model 实例

文章介绍了如何使用transformers库的automodel接口从HuggingFace官网下载预训练模型的configfile和modelfile,创建并初始化模型,然后进行迁移训练。另外,也提到了autoconfig方法,它会生成随机初始化的模型,需要从头训练。此外,还展示了如何保存和重新加载训练好的模型。
最低0.47元/天 解锁文章
2万+

被折叠的 条评论
为什么被折叠?



