pytorch实现从本地加载 .pth 格式模型


Posted in Python onFebruary 14, 2020

可以从官网加载预训练好的模型:

import torchvision.models as models
 
model = models.vgg16(pretrained = True)
print(model)

但是经常会出现因为下载速度太慢而出现requests.exceptions.ConnectionError: ('Connection aborted.', TimeoutError(10060, '由于连接方在一段时间后没有正确答复或连接的主机没有反应,连接尝试失败。', None, 10060, None))这种错误,因此需要我们手动去下载 .pth 文件(百度云也很慢,如果你是SVIP,当我没说;迅雷的速度也还可以),然后从本地加载。

从本地加载只需要把上面的代码换成如下:

import torchvision.models as models
 
model = models.vgg16(pretrained=False)
pre=torch.load(r'.\kaggle_dog_vs_cat\pretrain\vgg16-397923af.pth')
model.load_state_dict(pre)

如果你模型不是用的vgg16,而是用的vgg11或者vgg13,只需要修改语句 model = models.vgg16(pretrained=False) 为对应模型的函数即可。

以上这篇pytorch实现从本地加载 .pth 格式模型就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python ORM框架SQLAlchemy学习笔记之映射类使用实例和Session会话介绍
Jun 10 Python
用python 制作图片转pdf工具
Jan 30 Python
详解在Python和IPython中使用Docker
Apr 28 Python
python日志记录模块实例及改进
Feb 12 Python
Python错误提示:[Errno 24] Too many open files的分析与解决
Feb 16 Python
python互斥锁、加锁、同步机制、异步通信知识总结
Feb 11 Python
浅谈Python中的私有变量
Feb 28 Python
pandas中apply和transform方法的性能比较及区别介绍
Oct 30 Python
python3 反射的四种基本方法解析
Aug 26 Python
python面向对象之类属性和类方法案例分析
Dec 30 Python
python3 实现口罩抽签的功能
Mar 11 Python
常用的Python代码调试工具总结
Jun 23 Python
python 实现线程之间的通信示例
Feb 14 #Python
Python3变量与基本数据类型用法实例分析
Feb 14 #Python
OpenCV中VideoCapture类的使用详解
Feb 14 #Python
Python networkx包的实现
Feb 14 #Python
python常用运维脚本实例小结
Feb 14 #Python
Python如何在DataFrame增加数值
Feb 14 #Python
python turtle工具绘制四叶草的实例分享
Feb 14 #Python
You might like
修改php.ini以达到屏蔽错误信息并记录日志
2013/06/16 PHP
thinkphp中ajax与php响应过程详解
2014/12/08 PHP
PHP根据session与cookie用户登录状态操作类的代码
2016/05/13 PHP
php实现在线通讯录功能(附源码)
2016/05/13 PHP
ThinkPHP like模糊查询,like多匹配查询,between查询,in查询,一般查询书写方法
2018/09/26 PHP
javascript 隔行换色函数代码
2010/10/24 Javascript
JavaScript 判断日期格式是否正确的实现代码
2011/07/04 Javascript
javascript中关于break,continue的特殊用法与介绍
2012/05/24 Javascript
实例说明为什么不要行内使用javascript
2014/04/18 Javascript
JavaScript阻止回车提交表单的方法
2015/12/30 Javascript
分享有关jQuery中animate、slide、fade等动画的连续触发、滞后反复执行的bug
2016/01/10 Javascript
javascript实现数组去重的多种方法
2016/03/14 Javascript
js图片切换具体实现代码
2016/10/13 Javascript
vue实现商品加减计算总价的实例代码
2018/08/12 Javascript
浅谈js闭包理解
2019/03/28 Javascript
angular4+百分比进度显示插件用法示例
2019/05/05 Javascript
Vue使用轮询定时发送请求代码
2020/08/10 Javascript
如何在现代JavaScript中编写异步任务
2021/01/31 Javascript
[01:58]DOTA2上海特级锦标赛现场采访:RTZ这个ID到底好不好
2016/03/25 DOTA
Python中optionParser模块的使用方法实例教程
2014/08/29 Python
在Python编程过程中用单元测试法调试代码的介绍
2015/04/02 Python
Python写入数据到MP3文件中的方法
2015/07/10 Python
Python wxPython库Core组件BoxSizer用法示例
2018/09/03 Python
python turtle库画一个方格和圆实例
2019/06/27 Python
对Python3中列表乘以某一个数的示例详解
2019/07/20 Python
python中利用numpy.array()实现俩个数值列表的对应相加方法
2019/08/26 Python
python 判断txt每行内容中是否包含子串并重新写入保存的实例
2020/03/12 Python
HTML5 文件域+FileReader 分段读取文件并上传到服务器
2017/10/23 HTML / CSS
黄色火烈鸟:De Gele Flamingo
2019/03/18 全球购物
Java面试笔试题大全
2016/11/23 面试题
员工保密承诺书
2014/05/28 职场文书
校园文化标语
2014/06/18 职场文书
2014年领导班子专项整治整改方案
2014/09/28 职场文书
营运督导岗位职责
2015/04/10 职场文书
2019七夕节祝福语36句,快来收藏吧
2019/08/06 职场文书
mysql分组后合并显示一个字段的多条数据方式
2022/01/22 MySQL