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
投稿猜你喜欢
vue卡片翻转轮播展示，同时在翻转时切换数据，供大家参考，具体内容如下效果及代码代码：<template> <- 本文实例讲述了JavaScript函数参数使用带参数名的方式赋值传入的方法。分享给大家供大家参考。具体分析如下：这里其实就是在给函数传递参数
- 本文介绍基于Python语言，按照一定命名规则批量修改多个文件的文件名的方法。已知现有一个文件夹，其中包括班级所有同学上交的作业文件，每人一
- 简介黑洞图像大家都知道，毕竟前几年刚发布的时候曾火遍全网，甚至都做成表情包了。问题在于，凭什么认为这就是黑洞的照片，而不是一个甜甜圈啥的给整
- asp之家注：有时候我们想让程序运行变慢下来，asp中该怎么做呢？原理很简单就是在运行程序前运行一段无关紧要的程序就可以了，要实现加长程序的
- select serverproperty('productlevel') 如果没有打过SP，则为RTM，否则显示的是SP版
- 在今天网络开发方面，JavaScript起了很关键的作用；像jQuery, MooTools, Prototype等等JavaScript框
- 本文实例讲述了Go语言写入字符串到文件的方法。分享给大家供大家参考。具体实现方法如下：package mainimport &q
- 缘由新手学习 Django 当配置好 HTML 页面后，就需要使用一些静态资源，如图片，JS 文件，CSS 样式等，但是 Django 里面
- 在Python2中datetime对象没有timestamp方法，不能很方便的生成epoch，现有方法没有处理很容易导致错误。关于Epoch
- 本文实例为大家分享了tkinter+pygame+spider实现音乐播放器，供大家参考，具体内容如下1.确定页面SongSheet&nbs
- //主键 ALTER TABLE tablename add new_field_id int(5) UNSIGNED DEFAULT 0
- 比较说明1、break和continue是python两个关键字2、break和continue只能用在循环中3、break是终止循环的执行
- 当管理SQL Server内在的帐户和密码时，我们很容易认为这一切都相当的安全。但实际上并非如此。在这里，我们列出了一些对于SQL Serv
- 函数最重要的目的是方便我们重复使用相同的一段程序。将一些操作隶属于一个函数，以后你想实现相同的操作的时候，只用调用函数名就可以，而不需要重复
- 一种很常见的写法： document.write('<scr'+'ipt src=&quo
- 自上一篇文章 Z Story : Using Django with GAE Python 后台抓取多个网站的页面全文 后，大体的进度如下：
- 测试浏览器的版本: IETester 6 ,7 IE 8.0 Firefox 3.5.5 Chrome 4.1.249.1064 (4537
- 搞一个图形化界面还是挺酷的，是吧 安装库什么的应该不用多说了吧。。一般来说会让你把 designer.exe（编辑图形化界面的东西，跟vb差
- 在 Python 2.5 中， with 关键字被加入。它将常用的 try ... except ... finally ... 模式很方便
