PyTorch加载预训练模型的问题
2018-06-02 13:06
621 查看
解决方法：
!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
内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理 
相关文章推荐
- windows下用caffe加载二进制模型（linux下训练）的问题
- deep learning coursera yolo.h5已训练模型的的导入问题
- pytorch加载和保存模型
- TensorFlow 深度学习框架（7）-- 变量管理及训练模型的保存与加载
- tensorflow: 保存和加载模型, 参数；以及使用预训练参数方法
- 机器学习中的数据不平衡问题----通过随机采样比例大的类别使得训练集中大类的个数与小类相当，或者模型中加入惩罚项
- 训练深度模型的优化问题（十）
- 训练深度模型的优化问题（十六）
- TensorFlow保存和加载训练模型
- tensorflow模型参数保存和加载问题
- TensorFlow 深度学习框架（7）-- 变量管理及训练模型的保存与加载
- 训练深度模型的优化问题（二）
- tensorflow加载预训练模型的时候报错：ValueError:No OP Named DecodeBmp in difined operations的解决
- 180209 Keras加载自定义模型weight shape not compatible问题的解决方法
- Pytorch 加载训练好的模型 遇到的error
- TensorFlow 深度学习框架（7）-- 变量管理及训练模型的保存与加载
- 训练深度模型的优化问题（三）
- 训练深度模型的优化问题（十七）
- PyTorch(7)——模型的训练和测试、保存和加载
- CMU sphinx训练命令控制声学模型问题