(StackOverflow)使用Transformers从磁盘加载预训练模型

大表哥汽车人 2023-10-07 21:59:18

这是在Stack Overflow上的一个问答,链接如下:

https://automanbro.blog.csdn.net/article/details/133623232?ydreferer=aHR0cHM6Ly9tcC5jc2RuLm5ldC9tcF9ibG9nL21hbmFnZS9hcnRpY2xlP3NwbT0xMDM1LjIwMjIuMzAwMS41NDQ4

从这个问答中,我们可以了解到

使用from_pretrained方法加载预训练模型,无需每次下载权重数据。
在加载模型时,需要确保提供的路径是正确的模型标识符或包含config.json文件的目录路径。
使用相对路径或绝对路径。
使用save_pretrained方法保存文件。
示例代码展示了如何加载和保存预训练模型。
 

...全文
150 回复 打赏 收藏 转发到动态 举报
写回复
用AI写文章
回复
切换为时间正序
请发表友善的回复…
发表回复

682

社区成员

发帖
与我相关
我的任务
社区描述
自然语言处理实战,讨论算法、模型、服务、应用。请具体描述问题,提出问题。
算法推荐算法人工智能 个人社区
社区管理员
  • community_282
  • SoftwareTeacher
  • Alexxinlu
加入社区
  • 近7日
  • 近30日
  • 至今
社区公告

自然语言处理实战。

试试用AI创作助手写篇文章吧