pytorch实现从本地加载 .pth 格式模型
更新时间：2020年02月14日 11:51:34 作者：卢开毅
今天小编就为大家分享一篇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) 为对应模型的函数即可。
以上这篇pytorch实现从本地加载 .pth 格式模型就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了python tkinter事件绑定，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-03-03
今天小编就为大家分享一篇pandas 根据列的值选取所有行的示例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-11-11
编写函数最重要的原则就是：别写太复杂的函数,那什么样的函数才能算是过于复杂？一般会通过两个标准来判断,长度和圈复杂度,下面我们就来看看如何使用Python判断函数的圈复杂度吧2024-04-04
Python的datetime模块是处理日期和时间的强大工具,datetime类可以获取当前时间、指定日期、计算时间差、访问时间属性及格式化时间,这些功能使得在Python中进行时间日期处理变得简单高效,需要的朋友可以参考下2024-09-09
这篇文章主要介绍了python实现三次密码验证的示例，帮助大家更好的理解和学习使用python，感兴趣的朋友可以了解下2021-04-04
这篇文章主要介绍了Django Rest framework认证组件详细用法,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-07-07
本文主要介绍了Python中Matplotlib图像添加标签的方法实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-04-04
Python Pandas教程之使用 pandas.read_csv() 读取 csv
这篇文章主要介绍了Python Pandas教程之使用pandas.read_csv()读取csv，文章通过围绕主题展开详细的内容介绍，具有一定的参考价值，需要的小伙伴可以参考一下2022-09-09
这篇文章主要为大家介绍了使用PyTorch常见4个错误解决示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-10-10
引用和复制是Python处理对象的重要概念,本文主要介绍了Python中对象的引用与复制的使用,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧2023-12-12
最新评论