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使用datetime模块计算各种时间间隔的方法
Mar 24 Python
在Django的模型中添加自定义方法的示例
Jul 21 Python
Python的消息队列包SnakeMQ使用初探
Jun 29 Python
Python中格式化format()方法详解
Apr 01 Python
python3安装pip3(install pip3 for python 3.x)
Apr 03 Python
Python爬虫抓取代理IP并检验可用性的实例
May 07 Python
Python3使用SMTP发送带附件邮件
Jun 16 Python
在python中安装basemap的教程
Sep 20 Python
python实现多进程按序号批量修改文件名的方法示例
Dec 30 Python
python实现处理mysql结果输出方式
Apr 09 Python
Python接口测试结果集实现封装比较
May 01 Python
手把手教你怎么用Python实现zip文件密码的破解
May 27 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解析xml提示Invalid byte 1 of 1-byte UTF-8 sequence错误的处理方法
2013/11/14 PHP
destoon网站转移服务器后搜索汉字出现乱码的解决方法
2014/06/21 PHP
PHP实现基于3DES算法加密解密字符串示例
2018/08/24 PHP
Yii2处理密码加密及验证的方法
2019/05/12 PHP
解决PHP使用CURL发送GET请求时传递参数的问题
2019/10/11 PHP
Javascript调用C#代码
2011/01/17 Javascript
jQuery实现contains方法不区分大小写的方法
2015/02/13 Javascript
JavaScript中length属性的使用方法
2015/06/05 Javascript
jquery插件tytabs.jquery.min.js实现渐变TAB选项卡效果
2015/08/25 Javascript
jquery获取css的color值返回RGB的方法
2015/12/18 Javascript
javascript中获取class的简单实现
2016/07/12 Javascript
js中遍历Map对象的简单实例
2016/08/08 Javascript
JavaScript函数节流的两种写法
2017/04/07 Javascript
深入理解jquery的$.extend()、$.fn和$.fn.extend()
2017/07/08 jQuery
jQuery实现可编辑表格并生成json结果(实例代码)
2017/07/19 jQuery
vue-cli 自定义指令directive 添加验证滑块示例
2017/10/19 Javascript
JavaScript私有变量实例详解
2019/01/24 Javascript
Vue数组响应式操作及高阶函数使用代码详解
2020/08/01 Javascript
使用IronPython把Python脚本集成到.NET程序中的教程
2015/03/31 Python
利用python求相邻数的方法示例
2017/08/18 Python
python实现简易内存监控
2018/06/21 Python
Django 解决上传文件时,request.FILES为空的问题
2020/05/20 Python
Python小白学习爬虫常用请求报头
2020/06/03 Python
python3.7.3版本和django2.2.3版本是否可以兼容
2020/09/01 Python
python 常见的反爬虫策略
2020/09/27 Python
CSS3使用transition属性实现过渡效果
2018/04/18 HTML / CSS
提供世界各地便宜的机票:Sky-tours
2016/07/21 全球购物
如何查看在weblogic中已经发布的EJB
2012/06/01 面试题
采购部部门职责
2013/12/15 职场文书
父亲生日宴会答谢词
2014/01/10 职场文书
农民工讨薪标语
2014/06/26 职场文书
搭讪开场白台词大全
2015/05/28 职场文书
音乐剧猫观后感
2015/06/04 职场文书
大学校园餐饮创业计划书
2019/08/07 职场文书
html+css实现分层金字塔的实例
2021/06/02 HTML / CSS
create-react-app开发常用配置教程
2022/06/25 Javascript