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 如何提高元组的可读性,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-08-08
这篇文章主要介绍了Django的models中on_delete参数详解，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-07-07
某些条件测试，需要一个简单的web服务器测试一下，为此专门去配置个nginx 或者 apache服务器略显麻烦,这里就为大家介绍一下使用python快速建立超简单的web服务器的方法,需要的朋友可以参考下2018-02-02
今天小编就为大家分享一篇Python解决两个整数相除只得到整数部分的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-11-11
下面小编就为大家分享一篇用pandas按列合并两个文件的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-04-04
python中有一个zfill方法用来给字符串前面补0，非常不错，下面小编给大家分享了实例代码，非常不错，具有参考借鉴价值，需要的朋友参考下吧2018-04-04
pyspark 读取csv文件创建DataFrame的两种方法
今天小编就为大家分享一篇pyspark 读取csv文件创建DataFrame的两种方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-06-06
这篇文章主要介绍了用Python爬取指定关键词的微博，下面文章围绕Python爬取指定关键词的微博的相关资料展开详细内容,需要的朋友可以参考一下2021-11-11
本文主要介绍了Python使用pandasai实现数据分析，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-06-06
这篇文章主要介绍了pycharm 使用conda虚拟环境,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2023-03-03
最新评论