热门标签
热门文章
- 1数据库总复习第十一章 并发控制_遵循两段锁协议的调度例题
- 2Tiny11作者开源：利用微软官方镜像制作独属于你的Tiny11镜像
- 3【灵境矩阵】零代码创建AI智能体之行业词句助手_智能体零代码和低代码
- 4手把手教你如何使用GitHub的服务器搭建自己的简单网站_github搭建网站 dns
- 5项目经理究竟一年能挣多少钱?一起来看看
- 6跨平台应用开发进阶(四十五)uni-app集成企微客服实战_uni-app进阶
- 7GPT-3不算机器学习算法创新？2020年AI顶会最佳论文榜单引发激烈争论
- 8【AI模型-机器学习工具部署】远程服务器配置Jupyter notebook或jupyter lab服务_jupyter 服务器
- 9Java、JavaScript以及python使用正则表达式校验用户的输入_用javascrip验证用户输入的个人信息:正则表达式经常被应用于对用户输入的信息进行
- 10【Android 构建新工具】Bazel 构建Android项目
当前位置: article > 正文
pytorch中load和load_state_dict区别_load和load dict
作者：神奇cpp | 2024-06-20 09:48:00
赞
踩
load和load dict
torch.load
- 1
加载的是训练好的模型
例如（但是这个有个缺点就是load 要自己找 net）
#这样保存
torch.save(net,"ad.pth")
#这样加载
torch.load("ad.pth")
- 1
- 2
- 3
- 4
如下则是上面代码的隐藏操作
上面代码很有可能找不到model 而报错
#这样保存
torch.save(net.state_dict- 1
推荐阅读
相关标签
