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 相关文章推荐
Win10下Python环境搭建与配置教程
Nov 18 Python
python编程实现随机生成多个椭圆实例代码
Jan 03 Python
python2.7安装图文教程
Mar 13 Python
漂亮的Django Markdown富文本app插件的实现
Jan 02 Python
python使用phoenixdb操作hbase的方法示例
Feb 28 Python
详解Python读取yaml文件多层菜单
Mar 23 Python
python3实现二叉树的遍历与递归算法解析(小结)
Jul 03 Python
Pandas透视表(pivot_table)详解
Jul 22 Python
详解用python生成随机数的几种方法
Aug 04 Python
Python 多线程,threading模块,创建子线程的两种方式示例
Sep 29 Python
Python可以实现栈的结构吗
May 27 Python
总结Pyinstaller打包的高级用法
Jun 28 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
收音机术语解释
2021/03/01 无线电
生成随机字符串和验证码的类的PHP实例
2013/12/24 PHP
php使用fgetcsv读取csv文件出现乱码的解决方法
2014/11/08 PHP
smarty高级特性之过滤器的使用方法
2015/12/25 PHP
JavaScript具有类似Lambda表达式编程能力的代码(改进版)
2010/09/14 Javascript
js自动生成的元素与页面原有元素发生堆叠的解决方法
2014/09/04 Javascript
JS获取网页图片name属性的方法
2015/04/01 Javascript
select隐藏选中值对应的id,显示其它id的简单实现方法
2016/08/25 Javascript
Underscore之Array_动力节点Java学院整理
2017/07/10 Javascript
python网络编程学习笔记(六):Web客户端访问
2014/06/09 Python
python实现二分查找算法
2017/09/21 Python
Python标准模块--ContextManager上下文管理器的具体用法
2017/11/27 Python
python之matplotlib学习绘制动态更新图实例代码
2018/01/23 Python
Python 读取图片文件为矩阵和保存矩阵为图片的方法
2018/04/27 Python
Django框架使用富文本编辑器Uedit的方法分析
2018/07/31 Python
Python爬虫常用小技巧之设置代理IP
2018/09/13 Python
对pandas的算术运算和数据对齐实例详解
2018/12/22 Python
Python 互换字典的键值对实例
2019/02/12 Python
在Python中如何传递任意数量的实参的示例代码
2019/03/21 Python
Python OpenCV实现鼠标画框效果
2020/08/19 Python
pytorch实现建立自己的数据集(以mnist为例)
2020/01/18 Python
如何利用Python matplotlib绘制雷达图
2020/12/21 Python
Python的collections模块真的很好用
2021/03/01 Python
奖学金自我鉴定范文
2013/10/03 职场文书
中专毕业生自我鉴定
2013/11/21 职场文书
网络宣传方案
2014/03/15 职场文书
2014年单位植树节活动方案
2014/03/23 职场文书
青年文明号服务承诺
2014/03/31 职场文书
求职个人评价范文
2014/04/09 职场文书
2014预防青少年违法犯罪工作总结
2014/12/10 职场文书
学生退学证明
2015/06/23 职场文书
2015最新民情日记范文
2015/06/26 职场文书
三八妇女节新闻稿
2015/07/17 职场文书
《折线统计图》教学反思
2016/02/22 职场文书
解决Vue+SpringBoot+Shiro跨域问题
2021/06/09 Vue.js
在redisCluster中模糊获取key方式
2021/07/09 Redis