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实现人人网登录示例分享
Jan 19 Python
Python urllib、urllib2、httplib抓取网页代码实例
May 09 Python
python对excel文档去重及求和的实例
Apr 18 Python
Python实现压缩文件夹与解压缩zip文件的方法
Sep 01 Python
Python Series从0开始索引的方法
Nov 06 Python
python实现n个数中选出m个数的方法
Nov 13 Python
对python pandas 画移动平均线的方法详解
Nov 28 Python
python print输出延时,让其立刻输出的方法
Jan 07 Python
Python解释器以及PyCharm的安装教程图文详解
Feb 26 Python
Pycharm连接gitlab实现过程图解
Sep 01 Python
Python函数调用追踪实现代码
Nov 27 Python
安装pytorch时报sslerror错误的解决方案
May 17 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
linux下编译安装memcached服务
2014/08/03 PHP
PHP中仿制 ecshop验证码实例
2017/01/06 PHP
数组任意位置插入元素,删除特定元素的实例
2017/03/02 PHP
深入理解Yii2.0乐观锁与悲观锁的原理与使用
2017/07/26 PHP
PHP+redis实现的限制抢购防止商品超发功能详解
2019/09/19 PHP
4种Windows系统下Laravel框架的开发环境安装及部署方法详解
2020/04/06 PHP
JS小框架 fly javascript framework
2009/11/26 Javascript
javascript 语法基础 想学习js的朋友可以看看
2009/12/16 Javascript
JavaScript window.document的属性、方法和事件小结
2012/10/24 Javascript
Jquery选中或取消radio示例
2013/09/29 Javascript
js查找某元素中的所有图片地址的方法
2014/01/16 Javascript
了不起的node.js读书笔记之node.js中的特性
2014/12/22 Javascript
AngularJS 让人爱不释手的八种功能
2016/03/23 Javascript
有关JavaScript中call()和apply() 的一些理解
2016/05/20 Javascript
JavaScript实现通过select标签跳转网页的方法
2016/09/29 Javascript
浅析location.href跨窗口调用函数
2016/11/22 Javascript
浅谈React之状态(State)
2018/09/19 Javascript
Node.js模拟发起http请求从异步转同步的5种用法
2018/09/26 Javascript
vue使用微信JS-SDK实现分享功能
2019/08/23 Javascript
Node.js 在本地生成日志文件的方法
2020/02/07 Javascript
python进阶教程之模块(module)介绍
2014/08/30 Python
Python实现拷贝多个文件到同一目录的方法
2016/09/19 Python
Python 转义字符详细介绍
2017/03/21 Python
Python三级菜单的实例
2017/09/13 Python
python可视化爬虫界面之天气查询
2019/07/03 Python
Django模板Templates使用方法详解
2019/07/19 Python
Python-jenkins模块之folder相关操作介绍
2020/05/12 Python
python3 中时间戳、时间、日期的转换和加减操作
2020/07/14 Python
三年级班级文化建设方案
2014/05/04 职场文书
没有孩子的离婚协议书怎么写
2014/09/17 职场文书
学院党的群众路线教育实践活动第一阶段情况汇报
2014/10/25 职场文书
礼貌问候语大全
2015/11/10 职场文书
党章学习心得体会2016
2016/01/14 职场文书
2016年第十九届推普周活动总结
2016/04/06 职场文书
Python 阶乘详解
2021/10/05 Python
python机器学习实现oneR算法(以鸢尾data为例)
2022/03/03 Python