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计算圆周率pi的方法
Jul 11 Python
Linux下通过python访问MySQL、Oracle、SQL Server数据库的方法
Apr 23 Python
Python的IDEL增加清屏功能实例
Jun 19 Python
使用python装饰器计算函数运行时间的实例
Apr 21 Python
用TensorFlow实现多类支持向量机的示例代码
Apr 28 Python
Python+selenium 获取一组元素属性值的实例
Jun 22 Python
python 检查是否为中文字符串的方法
Dec 28 Python
python tornado使用流生成图片的例子
Nov 18 Python
python使用配置文件过程详解
Dec 28 Python
python 追踪except信息方式
Apr 25 Python
PyTorch实现重写/改写Dataset并载入Dataloader
Jul 14 Python
Django用内置方法实现简单搜索功能的方法
Dec 18 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中的str_repeat函数在JavaScript中的实现
2013/09/16 PHP
php如何解决无法上传大于8M的文件问题
2014/03/10 PHP
PHP调用.NET的WebService 简单实例
2015/03/27 PHP
PHP使用Mysqli类库实现完美分页效果的方法
2016/04/07 PHP
JS 精确统计网站访问量的实例代码
2013/07/05 Javascript
详解JavaScript的Date对象(制作简易钟表)
2020/04/07 Javascript
浅析JavaScript中作用域和作用域链
2016/12/06 Javascript
浅谈Node.js:理解stream
2016/12/08 Javascript
详解使用Node.js 将txt文件转为Excel文件
2017/07/05 Javascript
JS仿QQ好友列表展开、收缩功能(第二篇)
2017/07/07 Javascript
javascript定时器取消定时器及优化方法
2017/07/08 Javascript
vue组件Prop传递数据的实现示例
2017/08/17 Javascript
AngularJS实现注册表单验证功能
2017/10/16 Javascript
Vue父组件调用子组件事件方法
2018/02/23 Javascript
JS实现的简单折叠展开动画效果示例
2018/04/28 Javascript
node实现生成带参数的小程序二维码并保存到本地功能示例
2018/12/05 Javascript
jQuery实现经典的网页3D轮播图封装功能【附源码下载】
2019/02/15 jQuery
vue+animation实现翻页动画
2020/06/29 Javascript
解决vue单页面应用打包后相对路径、绝对路径相关问题
2020/08/14 Javascript
vue-cli脚手架的.babelrc文件用法说明
2020/09/11 Javascript
[57:59]完美世界DOTA2联赛循环赛 Ink Ice vs LBZS BO2第一场 11.05
2020/11/05 DOTA
python设置windows桌面壁纸的实现代码
2013/01/28 Python
Python运算符重载用法实例分析
2015/06/01 Python
python 删除非空文件夹的实例
2018/04/26 Python
Python异常处理知识点总结
2019/02/18 Python
python实现ip代理池功能示例
2019/07/05 Python
Python-Tkinter Text输入内容在界面显示的实例
2019/07/12 Python
python同步两个文件夹下的内容
2019/08/29 Python
Python绘制三角函数图(sin\cos\tan)并标注特定范围的例子
2019/12/04 Python
Python 通过爬虫实现GitHub网页的模拟登录的示例代码
2020/08/17 Python
计算机专业应届毕业生自荐信
2013/09/26 职场文书
电厂厂长岗位职责
2014/01/02 职场文书
公司担保书范文
2014/05/21 职场文书
大学生就业求职信
2014/06/12 职场文书
基于Redis延迟队列的实现代码
2021/05/13 Redis
VS2019连接MySQL数据库的过程及常见问题总结
2021/11/27 MySQL