解决pytorch多GPU训练保存的模型,在单GPU环境下加载出错问题
背景
在公司用多卡训练模型，得到权值文件后保存，然后回到实验室，没有多卡的环境，用单卡训练，加载模型时出错，因为单卡机器上，没有使用DataParallel来加载模型，所以会出现加载错误。
原因
DataParallel包装的模型在保存时，权值参数前面会带有module字符，然而自己在单卡环境下，没有用DataParallel包装的模型权值参数不带module。本质上保存的权值文件是一个有序字典。
解决方法
1.在单卡环境下，用DataParallel包装模型。
2.自己重写Load函数，灵活。
from collections import OrderedDict
def myOwnLoa
文件下载
评论信息
其他资源
- 数字图像处理(MATLAB版)
- Visual Paradigm 11.2 激活文件
- PLECS免安装
- Arduino-Arduino-Library-for-Proteus.zip
- 仿音乐网站 - 静态页面
- 最新版本PowerDesigner 16.6汉化工具(含安装包下载地址)
- ATM流程图（visio画的）
- Python实现ISODATA算法
- 2-ltv_nl_mpc.pdf
- Android 安卓管理日志，所有输出日志保存本地
- 机器学习-MIT行人检测数据库
- 基于opencv的人脸表情识别的预处理
- ae cc shine插件
- Sorry.NET PHP在线生成王境泽表情
- 破解WPS密码的软件
- Principles of Communication Engineering
- MFC网络对战（TCP）五子棋
- NMEA规范语句对比
免责申明
【只为小站】的资源来自网友分享，仅供学习研究，请务必在下载后24小时内给予删除，不得用于其他任何用途，否则后果自负。基于互联网的特殊性，【只为小站】 无法对用户传输的作品、信息、内容的权属或合法性、合规性、真实性、科学性、完整权、有效性等进行实质审查；无论 【只为小站】 经营者是否已进行审查，用户均应自行承担因其传输的作品、信息、内容而可能或已经产生的侵权或权属纠纷等法律责任。
本站所有资源不代表本站的观点或立场，基于网友分享，根据中国法律《信息网络传播权保护条例》第二十二条之规定，若资源存在侵权或相关问题请联系本站客服人员，zhiweidada#qq.com，请把#换成@，本站将给予最大的支持与配合，做到及时反馈和处理。关于更多版权及免责申明参见 版权及免责申明
本站所有资源不代表本站的观点或立场，基于网友分享，根据中国法律《信息网络传播权保护条例》第二十二条之规定，若资源存在侵权或相关问题请联系本站客服人员，zhiweidada#qq.com，请把#换成@，本站将给予最大的支持与配合，做到及时反馈和处理。关于更多版权及免责申明参见 版权及免责申明