pytorch实现从本地加载 .pth 格式模型

 更新时间:2020年02月14日 11:51:35   作者:卢开毅   我要评论
今天小编就为大家分享一篇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 文件操作的详解及实例

    Python 文件操作的详解及实例

    这篇文章主要介绍了Python 文件操作的详解及实例的相关资料,希望通过本文大家能够理解掌握Python 文件操作的知识,需要的朋友可以参考下
    2017-09-09
  • 详解Python字典小结

    详解Python字典小结

    这篇文章主要介绍了详解Python字典小结,详细的介绍了什么是字典且创建字典和示例,非常具有实用价值,需要的朋友可以参考下
    2018-10-10
  • python pygame实现方向键控制小球

    python pygame实现方向键控制小球

    这篇文章主要为大家详细介绍了python pygame实现方向键控制小球,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2019-05-05
  • python简易实现任意位数的水仙花实例

    python简易实现任意位数的水仙花实例

    今天小编就为大家分享一篇python简易实现任意位数的水仙花实例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2018-11-11
  • Python爬虫实现的根据分类爬取豆瓣电影信息功能示例

    Python爬虫实现的根据分类爬取豆瓣电影信息功能示例

    这篇文章主要介绍了Python爬虫实现的根据分类爬取豆瓣电影信息功能,结合完整实例形式分析了Python针对电影信息分类抓取的相关实现技巧,需要的朋友可以参考下
    2019-09-09
  • Python实现Youku视频批量下载功能

    Python实现Youku视频批量下载功能

    前段时间由于收集视频数据的需要,自己捣鼓了一个YouKu视频批量下载的程序。下面小编把实现过程分享到脚本之家平台,供大家参考
    2017-03-03
  • Django实现简单网页弹出警告代码

    Django实现简单网页弹出警告代码

    今天小编就为大家分享一篇Django实现简单网页弹出警告代码,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2019-11-11
  • pytorch 调整某一维度数据顺序的方法

    pytorch 调整某一维度数据顺序的方法

    今天小编就为大家分享一篇pytorch 调整某一维度数据顺序的方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2018-12-12
  • 解决Pycharm的项目目录突然消失的问题

    解决Pycharm的项目目录突然消失的问题

    今天小编就为大家分享一篇解决Pycharm的项目目录突然消失的问题,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2020-01-01
  • python基于property()函数定义属性

    python基于property()函数定义属性

    这篇文章主要介绍了python基于property()函数定义属性,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-01-01

最新评论