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的ORM框架SQLAlchemy入门教程
Apr 28 Python
python字典排序实例详解
May 20 Python
Python实现拷贝多个文件到同一目录的方法
Sep 19 Python
OpenCV2.3.1+Python2.7.3+Numpy等的配置解析
Jan 05 Python
Python反转序列的方法实例分析
Mar 21 Python
在pycharm中python切换解释器失败的解决方法
Oct 29 Python
Python 中导入csv数据的三种方法
Nov 01 Python
python自动发送测试报告邮件功能的实现
Jan 22 Python
树莓派采用socket方式文件传输(python)
Jun 22 Python
python3实现绘制二维点图
Dec 04 Python
Django通用类视图实现忘记密码重置密码功能示例
Dec 17 Python
Python tkinter之ComboBox(下拉框)的使用简介
Feb 05 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
谷歌音乐搜索栏的提示功能php修正代码
2011/05/09 PHP
PHP中去除换行解决办法小结(PHP_EOL)
2011/11/27 PHP
php自定义类fsocket模拟post或get请求的方法
2015/07/31 PHP
PHP微信刮刮卡 附微信接口
2016/07/22 PHP
Win10 下安装配置IIS + MySQL + nginx + php7.1.7
2017/08/04 PHP
window.onbeforeunload方法在IE下无法正常工作的解决办法
2010/01/23 Javascript
IE8下String的Trim()方法失效的解决方法
2013/11/08 Javascript
jquery选择器排除某个DOM元素的方法(实例演示)
2014/04/25 Javascript
解决ueditor jquery javascript 取值问题
2014/12/30 Javascript
JavaScript获取一个范围内日期的方法
2015/04/24 Javascript
JavaScript中String.prototype用法实例
2015/05/20 Javascript
JavaScript变量的作用域全解析
2015/08/14 Javascript
详解webpack import()动态加载模块踩坑
2018/07/17 Javascript
使用js实现将后台传入的json数据放在前台显示
2018/08/06 Javascript
Python写入数据到MP3文件中的方法
2015/07/10 Python
python中set()函数简介及实例解析
2018/01/09 Python
python逐行读写txt文件的实例讲解
2018/04/03 Python
详解django+django-celery+celery的整合实战
2019/03/19 Python
基于python的Paxos算法实现
2019/07/03 Python
Python日志无延迟实时写入的示例
2019/07/11 Python
python定义类self用法实例解析
2020/01/22 Python
Python的collections模块真的很好用
2021/03/01 Python
CSS3实现歌词进度文字颜色填充变化动态效果的思路详解
2020/06/02 HTML / CSS
HTML5中meta属性的使用方法
2016/02/29 HTML / CSS
Html5之title吸顶功能
2018/06/04 HTML / CSS
英国最大的电脑零售连锁店集团:PC World
2016/10/10 全球购物
FILA斐乐中国官方商城:意大利运动品牌
2017/01/25 全球购物
澳大利亚手表品牌:Time IV Change
2018/10/06 全球购物
美国乒乓球设备、配件和服装品牌:Killerspin
2020/06/07 全球购物
销售顾问的岗位职责
2013/11/13 职场文书
合伙购房协议样本
2014/10/06 职场文书
党员违纪检讨书怎么写
2014/11/01 职场文书
幼儿园体操比赛口号
2015/12/25 职场文书
三年级作文之趣事作文
2019/11/04 职场文书
《追风筝的人》:人心中的成见是座大山,但请不忘初心
2019/11/15 职场文书
java多态注意项小结
2021/10/16 Java/Android