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编程实现两个文件夹里文件的对比功能示例【包含内容的对比】
Jun 20 Python
Python原始字符串与Unicode字符串操作符用法实例分析
Jul 22 Python
python实现批量修改文件名代码
Sep 10 Python
Python编程之黑板上排列组合,你舍得解开吗
Oct 30 Python
python 3.7.0 下pillow安装方法
Aug 27 Python
python读取目录下最新的文件夹方法
Dec 24 Python
解决项目pycharm能运行,在终端却无法运行的问题
Jan 19 Python
对python多线程SSH登录并发脚本详解
Feb 14 Python
pandas 中对特征进行硬编码和onehot编码的实现
Dec 20 Python
python数据库编程 Mysql实现通讯录
Mar 27 Python
django 将自带的数据库sqlite3改成mysql实例
Jul 09 Python
Python+Matplotlib+LaTeX玩转数学公式
Feb 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
常用的PHP数据库操作方法(MYSQL版)
2011/06/08 PHP
浅析Yii2中GridView常见操作
2016/04/22 PHP
PHP的mysqli_select_db()函数讲解
2019/01/23 PHP
jQuery学习7 操作JavaScript对象和集合的函数
2010/02/07 Javascript
jquery Mobile入门—多页面切换示例学习
2013/01/08 Javascript
jQuery选择器用法实例详解
2015/12/17 Javascript
BootStrap按钮标签及基本样式
2016/11/23 Javascript
AJAX和jQuery动态加载数据的实现方法
2016/12/05 Javascript
详解微信小程序——自定义圆形进度条
2016/12/29 Javascript
BootStrap Datetimepicker 汉化的实现代码
2017/02/10 Javascript
JQuery 进入页面默认给已赋值的复选框打钩
2017/03/23 jQuery
AngularJS实用基础知识_入门必备篇(推荐)
2017/07/10 Javascript
快速解决vue动态绑定多个class的官方实例语法无效的问题
2018/09/05 Javascript
Angular ui-roter 和AngularJS 通过 ocLazyLoad 实现动态(懒)加载模块和依赖
2018/11/25 Javascript
微信小程序使用setData修改数组中单个对象的方法分析
2018/12/30 Javascript
微信公众平台 发送模板消息(Java接口开发)
2019/04/17 Javascript
javascript写一个ajax自动拦截并下载数据代码实例
2019/09/07 Javascript
vue2.* element tabs tab-pane 动态加载组件操作
2020/07/19 Javascript
基于jquery实现彩色投票进度条代码解析
2020/08/26 jQuery
JS前端基于canvas给图片添加水印
2020/11/11 Javascript
vue实现登录功能
2020/12/31 Vue.js
Vue页面渲染中key的应用实例教程
2021/01/12 Vue.js
[42:27]DOTA2上海特级锦标赛主赛事日 - 3 败者组第三轮#2Fnatic VS OG第三局
2016/03/05 DOTA
[46:57]EG vs Winstrike 2018国际邀请赛小组赛BO2 第二场 8.18
2018/08/19 DOTA
matplotlib subplots 设置总图的标题方法
2018/05/25 Python
Python实现数据结构线性链表(单链表)算法示例
2019/05/04 Python
使用tensorflow框架在Colab上跑通猫狗识别代码
2020/04/26 Python
scrapy与selenium结合爬取数据(爬取动态网站)的示例代码
2020/09/28 Python
python matlab库简单用法讲解
2020/12/31 Python
世界上最大的在线旅行社新加坡网站:Expedia新加坡
2016/08/25 全球购物
澳大利亚最超值的自行车之家:Reid Cycles
2019/03/24 全球购物
eBay奥地利站:eBay.at
2019/07/24 全球购物
干部培训工作总结2015
2015/05/25 职场文书
2015年服务员个人工作总结
2015/05/27 职场文书
美丽心灵观后感
2015/06/01 职场文书
幼儿园大班教师评语
2019/06/21 职场文书