pytorch 使用加载训练好的模型做inference


Posted in Python onFebruary 20, 2020

前提: 模型参数和结构是分别保存的

1、 构建模型(# load model graph)

model = MODEL()

2、加载模型参数(# load model state_dict)

model.load_state_dict
 (
 {

 k.replace('module.',''):v for k,v in

 torch.load(config.model_path, map_location=config.device).items()

 }
 )
 
model = self.model.to(config.device)

* config.device 指定使用哪块GPU或者CPU

*k.replace('module.',''):v 防止torch.DataParallel训练的模型出现加载错误

(解决RuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found one of them on device: cuda:1问题)

3、设置当前阶段为inference(# predict)

model.eval()

以上这篇pytorch 使用加载训练好的模型做inference就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python魔术方法详解
Feb 14 Python
在Python的Flask框架下使用sqlalchemy库的简单教程
Apr 09 Python
Django中几种重定向方法
Apr 28 Python
探究Python中isalnum()方法的使用
May 18 Python
Python中字典的浅拷贝与深拷贝用法实例分析
Jan 02 Python
python编程培训 python培训靠谱吗
Jan 17 Python
Python利用递归实现文件的复制方法
Oct 27 Python
Python3 实现串口两进程同时读写
Jun 12 Python
Python循环中else,break和continue的用法实例详解
Jul 11 Python
django迁移文件migrations的实现
Mar 31 Python
python中rc1什么意思
Jun 19 Python
再谈python_tkinter弹出对话框创建
Mar 20 Python
pytorch中的inference使用实例
Feb 20 #Python
python encrypt 实现AES加密的实例详解
Feb 20 #Python
Python关于反射的实例代码分享
Feb 20 #Python
Python3监控疫情的完整代码
Feb 20 #Python
浅谈Pytorch torch.optim优化器个性化的使用
Feb 20 #Python
关于torch.optim的灵活使用详解(包括重写SGD,加上L1正则)
Feb 20 #Python
Python sys模块常用方法解析
Feb 20 #Python
You might like
构建简单的Webmail系统
2006/10/09 PHP
PHP图片处理类 phpThumb参数用法介绍
2012/03/11 PHP
浅谈PHP中foreach/in_array的使用
2015/11/02 PHP
PHP实现微信JS-SDK接口选择相册及拍照并上传的方法
2016/12/05 PHP
laravel5.0在linux下解决.htaccess无效和去除index.php的问题
2019/10/16 PHP
thinkphp5 框架结合plupload实现图片批量上传功能示例
2020/04/04 PHP
js表格分页实现代码
2009/09/18 Javascript
JavaScript 对象成员的可见性说明
2009/10/16 Javascript
js下将字符串当函数执行的方法
2011/07/13 Javascript
jQuery源码分析-03构造jQuery对象-源码结构和核心函数
2011/11/14 Javascript
JS获取地址栏参数的几种方法小结
2014/02/28 Javascript
js Calender控件使用详解
2015/01/05 Javascript
jQuery过滤选择器详解
2015/01/13 Javascript
jQuery实现为图片添加镜头放大效果的方法
2015/06/25 Javascript
JavaScript检查子字符串是否在字符串中的方法
2016/02/03 Javascript
javascript的BOM
2016/05/03 Javascript
jQuery基于BootStrap样式实现无限极地区联动
2016/08/26 Javascript
详解基于Node.js的微信JS-SDK后端接口实现代码
2017/07/15 Javascript
详解用函数式编程对JavaScript进行断舍离
2017/09/18 Javascript
微信小程序实现长按删除图片的示例
2018/05/18 Javascript
vue中img src 动态加载本地json的图片路径写法
2019/04/25 Javascript
微信小程序实现简单购物车功能
2020/12/30 Javascript
phpsir 开发 一个检测百度关键字网站排名的python 程序
2009/09/17 Python
Python多线程正确用法实例解析
2020/05/30 Python
Java多线程实现四种方式原理详解
2020/06/02 Python
python numpy实现rolling滚动案例
2020/06/08 Python
python实现发送QQ邮件(可加附件)
2020/12/23 Python
详解如何用HTML5 Canvas API控制图片的缩放变换
2016/03/22 HTML / CSS
我的求职计划书
2014/01/10 职场文书
八年级物理教学反思
2014/01/19 职场文书
十八大感想感言
2014/02/10 职场文书
村班子对照检查材料
2014/08/18 职场文书
群众路线个人对照检查材料2014
2014/09/26 职场文书
2014年预备党员群众路线教育实践活动对照检查材料思想汇报
2014/10/02 职场文书
小时代观后感
2015/06/10 职场文书
为什么node.js不适合大型项目
2021/04/28 Javascript