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实现Linux下守护进程的编写方法
Aug 22 Python
在Python下尝试多线程编程
Apr 28 Python
Swift中的协议(protocol)学习教程
Jul 08 Python
Python网络编程详解
Oct 31 Python
解决Python的str强转int时遇到的问题
Apr 09 Python
python检测主机的连通性并记录到文件的实例
Jun 21 Python
解决python中os.listdir()函数读取文件夹下文件的乱序和排序问题
Oct 17 Python
pyqt5与matplotlib的完美结合实例
Jun 21 Python
Flask框架重定向,错误显示,Responses响应及Sessions会话操作示例
Aug 01 Python
TensorFlow 显存使用机制详解
Feb 03 Python
TensorFlow实现checkpoint文件转换为pb文件
Feb 10 Python
Python3变量与基本数据类型用法实例分析
Feb 14 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新手上路(十)
2006/10/09 PHP
php代码中使用换行及(\n或\r\n和br)的应用
2013/02/02 PHP
一个非常完美的读写ini格式的PHP配置类分享
2015/02/12 PHP
php arsort 数组降序排序详细介绍
2016/11/17 PHP
线路分流自动跳转代码;希望对大家有用!
2006/12/02 Javascript
js 文件引入实现代码
2010/04/23 Javascript
javascript parseInt() 函数的进制转换注意细节
2013/01/08 Javascript
JS输入用户名自动显示邮箱后缀列表的方法
2015/01/27 Javascript
nodejs实现bigpipe异步加载页面方案
2016/01/26 NodeJs
AngularJS入门教程之迭代器过滤详解
2016/08/18 Javascript
微信小程序 数据交互与渲染实例详解
2017/01/21 Javascript
javaScript封装的各种写法
2017/08/14 Javascript
解决layui中table异步数据请求不支持自定义返回数据格式的问题
2018/08/19 Javascript
JS造成内存泄漏的几种情况实例分析
2020/03/02 Javascript
解决Nuxt使用axios跨域问题
2020/07/06 Javascript
Python的Django应用程序解决AJAX跨域访问问题的方法
2016/05/31 Python
python下os模块强大的重命名方法renames详解
2017/03/07 Python
python利用smtplib实现QQ邮箱发送邮件
2020/05/20 Python
django-初始配置(纯手写)详解
2019/07/30 Python
nginx黑名单和django限速,最简单的防恶意请求方法分享
2019/08/09 Python
使用pygame写一个古诗词填空通关游戏
2019/12/03 Python
keras 实现轻量级网络ShuffleNet教程
2020/06/19 Python
css3 border-radius属性详解
2017/07/05 HTML / CSS
来自南加州灵感的工作和娱乐服装:TravisMathew
2019/05/01 全球购物
宿舍违规检讨书
2014/01/12 职场文书
教师师德反思材料
2014/02/15 职场文书
技校毕业生个人学习的自我评价
2014/02/21 职场文书
会计人员岗位职责
2014/03/19 职场文书
超市仓管员岗位职责
2014/04/07 职场文书
房产授权委托书范本
2014/09/22 职场文书
经费申请报告范文
2015/05/18 职场文书
2015年学校教科室工作总结
2015/07/20 职场文书
《所见》教学反思
2016/02/23 职场文书
Pillow图像处理库安装及使用
2022/04/12 Python
MySQL池化框架学习接池自定义
2022/07/23 MySQL
win10系统计算机图标怎么调出来?win10调出计算机图标的方法
2022/08/14 数码科技