解决pytorch多GPU训练保存的模型,在单GPU环境下加载出错问题
背景
在公司用多卡训练模型，得到权值文件后保存，然后回到实验室，没有多卡的环境，用单卡训练，加载模型时出错，因为单卡机器上，没有使用DataParallel来加载模型，所以会出现加载错误。
原因
DataParallel包装的模型在保存时，权值参数前面会带有module字符，然而自己在单卡环境下，没有用DataParallel包装的模型权值参数不带module。本质上保存的权值文件是一个有序字典。
解决方法
1.在单卡环境下，用DataParallel包装模型。
2.自己重写Load函数，灵活。
from collections import OrderedDict
def myOwnLoa
文件下载
评论信息
其他资源
- 最新版swan近岸海浪模式的程序用户手册及相关实例
- 内网渗透测试技术
- 基于倒谱图判断浊音的基音周期MATLAB仿真 语音信号处理
- 人工智能导论.zip
- Matlab CRC校验（GUI）
- STC15W系列单片机的定时器0和定时器2以16位自动重装模式分别在两个端口输出脉冲
- USB串口通信JAR包及DLL文件完美支持WIN32WIN64
- Ethernet.zip
- web课程设计 个人网站制作
- 2018年全国大学生计算机技能应用大赛决赛 大题
- three.js从入门当放弃（四）图片资源
- 解决QT无法切换fcitx中文输入的时make编译出现的问题-附件资源
- Resplash.zip
- 安全攻防技术文档Redis
- MVC架构设计与经典的三层模型
- 基于android时间管理安排项目TimeManager-master.zip
- vs 2012 百度网盘 失效私聊
- 《嵌入式Linux C语言应用程序设计》读书笔记
- 停车场管理系统
- 抽奖平台前台开发（前后端最终源码）实现将抽奖结果提交的表单上传至数据库
- REDHAT6.9 kmod-oracleasm rpm包
- 毕业论文模板
- jxl完美实现Excel加水印(内附源码及jar包，基于jxl2.6.12修改编译版)
- bec 高级 词汇资料
- WIN7 64位下，西门子PPI电缆驱动更新（插件）
免责申明
【只为小站】的资源来自网友分享，仅供学习研究，请务必在下载后24小时内给予删除，不得用于其他任何用途，否则后果自负。基于互联网的特殊性，【只为小站】 无法对用户传输的作品、信息、内容的权属或合法性、合规性、真实性、科学性、完整权、有效性等进行实质审查；无论 【只为小站】 经营者是否已进行审查，用户均应自行承担因其传输的作品、信息、内容而可能或已经产生的侵权或权属纠纷等法律责任。
本站所有资源不代表本站的观点或立场，基于网友分享，根据中国法律《信息网络传播权保护条例》第二十二条之规定，若资源存在侵权或相关问题请联系本站客服人员，zhiweidada#qq.com，请把#换成@，本站将给予最大的支持与配合，做到及时反馈和处理。关于更多版权及免责申明参见 版权及免责申明
本站所有资源不代表本站的观点或立场，基于网友分享，根据中国法律《信息网络传播权保护条例》第二十二条之规定，若资源存在侵权或相关问题请联系本站客服人员，zhiweidada#qq.com，请把#换成@，本站将给予最大的支持与配合，做到及时反馈和处理。关于更多版权及免责申明参见 版权及免责申明