热门标签
热门文章
- 1C语言经典编程282例14 (阶乘、最大公约数与最小公倍数等)_任意数的n次幂求解:分别从键盘输入底数与次幂,求出该底数的本次幂结果与后续5个+1
- 2ArcGIS 分幅分割栅格_arcgis栅格分割
- 3C++ 消息中间件（MQ4CPP）_c++ 消息中间件 mq4cpp)
- 4chatGLM3+chatchat实现本地知识库
- 5单片机中的ROM,RAM和FLASH的作用_内部程序处理器rom作用
- 6如何在Go中使用泛型
- 7华为鲲鹏云认证考试内容有哪些？华为鲲鹏云认证考试报名条件
- 8前端调用 Stable diffusion 秋叶启动器 Web-Ui API_subseed
- 9PCA+SVM进行人脸识别（ORL人脸数据集分类）_下载orl人脸库人脸图像数据集,将数据集分为训练集和测试集。 (2)计算所有训练集样
- 10 C++的原码放在C语言里为什么不能用呢 _c语言程序无法调用c++编写的代码
当前位置: article > 正文
PyTorch加载预训练模型的问题_pytorch resnet18 加载本地预训练模型
作者：Cpp五条 | 2024-04-02 06:04:21
赞
踩
pytorch resnet18 加载本地预训练模型
解决方法：
!wget http://download.pytorch.org/models/resnet18-5c106cde.pth
然后：
import torch
import torchvision.models as models
resnet18 = models.resnet18()
resnet18.load_state_dict(torch.load('resnet18-5c106cde.pth'))
print resnet18
声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/Cpp五条/article/detail/351061
推荐阅读
相关标签
