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中if语句的嵌套用法
May 14 Python
Python自动调用IE打开某个网站的方法
Jun 03 Python
Python中列表元素转为数字的方法分析
Jun 14 Python
PyQt5每天必学之布局管理
Apr 19 Python
Python 数据处理库 pandas进阶教程
Apr 21 Python
python 集合 并集、交集 Series list set 转换的实例
May 29 Python
python3.5安装python3-tk详解
Apr 26 Python
深入浅析python3中的unicode和bytes问题
Jul 03 Python
Pycharm中切换pytorch的环境和配置的教程详解
Mar 13 Python
记一次Django响应超慢的解决过程
Sep 17 Python
Python如何获取文件路径/目录
Sep 22 Python
python3爬虫中多线程的优势总结
Nov 24 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
ThinkPHP的截取字符串函数无法显示省略号的解决方法
2014/06/25 PHP
Thinkphp使用mongodb数据库实现多条件查询方法
2014/06/26 PHP
ThinkPHP中处理表单中的注意事项
2014/11/22 PHP
php实现数组按指定KEY排序的方法
2015/03/30 PHP
PHP Oauth授权和本地加密实现方法
2016/08/12 PHP
PHP远程连接oracle数据库操作实现方法图文详解
2019/04/11 PHP
js获取提交的字符串的字节数
2009/02/09 Javascript
基于node.js的快速开发透明代理
2010/12/25 Javascript
浅谈Javascript事件模拟
2012/06/27 Javascript
JS 打印界面的CSS居中代码适用所有浏览器
2014/03/19 Javascript
jQuery获取动态生成的元素示例
2014/06/15 Javascript
JS实现让网页背景图片斜向移动的方法
2015/02/25 Javascript
JavaScript阻止浏览器返回按钮的方法
2015/03/18 Javascript
javascript弹性运动效果简单实现方法
2016/01/08 Javascript
JavaScript全屏和退出全屏事件总结(附代码)
2017/08/17 Javascript
JS手机端touch事件计算滑动距离的方法示例
2017/10/26 Javascript
jQuery.parseJSON()函数详解
2019/02/28 jQuery
Layui 导航默认展开和菜单栏选中高亮设置的方法
2019/09/04 Javascript
基于js实现抽红包并分配代码实例
2019/09/19 Javascript
vue双向绑定数据限制长度的方法
2019/11/04 Javascript
用Python的Django框架完成视频处理任务的教程
2015/04/02 Python
Python3.x版本中新的字符串格式化方法
2015/04/24 Python
简单了解Python下用于监视文件系统的pyinotify包
2015/11/13 Python
python中的lambda表达式用法详解
2016/06/22 Python
在Python 中同一个类两个函数间变量的调用方法
2019/01/31 Python
Python中请不要再用re.compile了
2019/06/30 Python
用vue.js组件模拟v-model指令实例方法
2019/07/05 Python
python manage.py runserver流程解析
2019/11/08 Python
python 实现任务管理清单案例
2020/04/25 Python
如何把python项目部署到linux服务器
2020/08/26 Python
详解pandas中利用DataFrame对象的.loc[]、.iloc[]方法抽取数据
2020/12/13 Python
计算机专业推荐信范文
2013/11/20 职场文书
资源工程专业毕业生求职信
2014/02/27 职场文书
财务部绩效考核方案
2014/05/04 职场文书
开学第一周值周总结
2015/07/16 职场文书
浅谈mysql返回Boolean类型的几种情况
2021/06/04 MySQL