pytorch实现从本地加载 .pth 格式模型
作者：卢开毅 发布时间：2021-07-01 18:32:03
标签：pytorch,加载,pth
可以从官网加载预训练好的模型：
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) 为对应模型的函数即可。
来源：https://blog.csdn.net/TomorrowAndTuture/article/details/100219240
0
投稿猜你喜欢
- 背景最近尝试了解Django中ORM实现的原理，发现其用到了metaclass(元类)这一技术，进一步又涉及到Python class中有两
前言Blender 并不是唯一一款允许你为场景编程和自动化任务的3D软件; 随着每一个新版本的推出，Blender 正逐渐成为一个可靠的 C- 数组:【重点1】implode(分隔,arr) 把数组值数据按指定字符连接起来例如：$arr=array('1','
- 问题描述101/100，想要的结果是1，但是算出来的结果是1.01解决方案101 // 100结果图补充：python2 与 3：一般除法、
- 1.绘制面积图面积图常用于描述某指标随时间的变化程度。其面积也通常可以有一定的含义。绘制面积图使用的是plt.stackplot()方法。以
- 这篇文章主要介绍了python matplotlib给图中的点加标签,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习
- User Centered Design 以用户为中心的设计，一说到这个很多人马上想到互联网和软件的设计，因为在IT、互联网行业提得最多。其
- 如何用METADATA替换ADOVBS.INC？ 在ASP中，使用组件时，如ADO，得先包含
- 问题重述实习项目要做安全帽目标检测，拿到了公司给的一些视频数据，使用Opencv读取视频并每隔 1 s 1s 1s存储一副图像，下面是一些视
- 1. 什么是phpDocumentor ? PHPDocumentor 是一个用PHP写的工具，对于有规范注释的php程序，它能够快速生成具
- 在python中，通过内嵌集成re模块，程序媛们可以直接调用来实现正则匹配。本文重点给大家介绍python中正则表达式 re.findall
- 这篇论坛文章(赛迪网技术社区）主要介绍了MySQL数据库主从复制的相关概念及设置方法，详细内容请大家参考下文：MySQL支持单向、异步复制，
- 本文中介绍的系统优化，主要针对前端和后台这两方面（后台方面主要对SQL语句和数据存储进行了优化），下文中我们将介绍一些优化技巧和经验。技巧：
- 1. 数据抽取的概念2. 数据的分类3. JSON数据概述及解析3.1 JSON数据格式3.2 解析库jsonjson模块是Python内置
- 文件上传是所有UI自动化测试都要面对的一个头疼问题，今天博主在这里给大家分享下自己处理文件上传的经验，希望能够帮助到广大被文件上传坑住的se
- 在通信领域，可以通过希尔伯特变换求解解析信号，进而求解窄带信号的包络。实现希尔伯特变换有两种方法，一种是对信号做FFT，单后只保留单边频谱，
- 记得导入包，其他按键可类比def keyPressEvent(self, event): if event.key() == Q
- yolov5返回坐标（v6版）1 、从yolov5文件夹李找到detect.py，按Ctrl+F 输入annotator.box_label
- 1.自定义聚合函数，结合agg使用2. 同时使用多个聚合函数3. 指定某一列使用某些聚合函数4.merge与transform使用impor
- 控制资源访问前文提到threading库在多线程时，对同一资源的访问容易导致破坏与丢失数据。为了保证安全的访问一个资源对象，我们需要创建锁。
