IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
 
   -> 人工智能 -> ValueError: Expected input batch_size (5408) to match target batch_size (32). -> 正文阅读

[人工智能]ValueError: Expected input batch_size (5408) to match target batch_size (32).

----------初识PYTORCH————(小白篇)

 这是第一次接触pytorch,想把自己踩过的坑写出来希望能给大家帮助,同时也可以加深对它的理解。

这个报错表达的意思是:预期输入批次大小(5408)与目标批次大小(32)匹配错误。
即在你所构建的神经网络,层与层之间有输入与接受不匹配的缘故,比如:

        self.conv1 = nn.Conv2d(3,6,kernel_size=5,stride=1,padding=0)#第一层卷积层
        self.pool = nn.MaxPool2d(kernel_size=2,stride=2,padding=0)#池化层
        self.conv2 = nn.Conv2d(6,32,kernel_size=5,stride=1,padding=0)#第二层卷积层
    
        self.fc1 = nn.Linear(32*1*1,120)#第一层全连接层
        self.fc2 = nn.Linear(120, 84)#第二层全连接层
        self.fc3 = nn.Linear(84, 14)#第三层全连接层

上述代码的问题出在第二层卷积层与第一层全连接层之间,即第二层卷积层输出的批次大小为与第一层全连接的接受目标批次不匹配,所以我们应该调整第一层的全连接层的输入通道,其中,即调整下面代码的参数:

 self.fc1 = nn.Linear(32*1*1,120)#第一层全连接层

上述nn.Linear有两个参数,分别为输入特征的大小和输出特征的大小;
而第一个参数32为通道数量,1*1为通道大小,对应上面的第二层卷积层
那么如何得到正确的参数呢?
我们完全可以看看从第二层卷积层输出的批次是什么样的:

	·   x = torch.randn(256,3,64,64)#(这里是输入的图片格式256张:3通道  64*64的图片,大家根据自己的输入来写)
        x = self.pool(F.relu(self.conv1(x)))
        out = self.pool(F.relu(self.conv2(x)))
        print(out.shape)

最后显示【256,32,13,13】,所以将 nn.Linear(3211,120)中的第一个参数改为 nn.Linear(321313,120)即可。

  人工智能 最新文章
2022吴恩达机器学习课程——第二课(神经网
第十五章 规则学习
FixMatch: Simplifying Semi-Supervised Le
数据挖掘Java——Kmeans算法的实现
大脑皮层的分割方法
【翻译】GPT-3是如何工作的
论文笔记:TEACHTEXT: CrossModal Generaliz
python从零学(六)
详解Python 3.x 导入(import)
【答读者问27】backtrader不支持最新版本的
上一篇文章      下一篇文章      查看所有文章
加:2021-08-06 09:37:50  更:2021-08-06 09:50:46 
 
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁

360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年5日历 -2024/5/4 19:06:28-

图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码