Expected more than 1 value per channel when training, got input size torch.Size......

最新推荐文章于 2024-07-03 18:00:00 发布

毛毛真nice

最新推荐文章于 2024-07-03 18:00:00 发布

阅读量6.2k

点赞数 4

分类专栏： python专栏文章标签：模型训练 BatchNorm ValueError 单样本解决方案

本文链接：https://blog.csdn.net/qq_37554556/article/details/117849364

版权

python专栏专栏收录该内容

11 篇文章 1 订阅

订阅专栏

完整问题描述

在将自己制作的数据集输入到模型中出现如下报错：

ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 512, 1, 1])

在这里插入图片描述

问题分析

模型中用了batchnomolization，训练中用batch训练的时候当前batch恰好只含一个sample，而由于BatchNorm操作需要多于一个数据计算平均值，因此造成该错误。

解决方案

由于我自己也没有设置batchsize，所以并没有网上提供的那种含有DataLoader的类，看了不少经验帖之后，发现有种方法能够很好的解决，就是在训练模型前加入：

model.eval()

如图所示：
在这里插入图片描述

然后就可以解决该问题了，完美！！！

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

毛毛真nice

关注关注

4
点赞
踩
4

收藏

觉得还不错? 一键收藏
打赏
7
评论
复制链接

分享到 QQ

分享到新浪微博

扫一扫

专栏目录

YOLOv5-5.x版本报错Expected more than 1 value per channel when training, got input size torch.Size([1, *]

younger14的博客

04-21

1510

yolov5在运行之前会先用全0的数据跑一遍，确保我们的网络配置能够跑通，但是源码设置的全0数据的shape为[1,3,256,256],batch size为1，这就导致了我们的有些Batch Normalization操作会出现以下报错： Expected more than 1 value per channel when training, got input size torch.Size([1, *]) 因为batch size==1是不能进行BN操作的，所以我们需要修改源码给的全零数据的sha

pytorch报错：ValueError: Expected more than 1 value per channel when training, got input size torch.Siz

qq_38504109的博客

12-01

1875

pytorch报错：ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1,256,1,1]) 通过查阅之后：解决方法1.在torch.utils.data.DataLoader类中或自己创建的继承于DataLoader的类中设置参数drop_last=True，把不够一个batch_size的数据丢弃。（实际操作不行）解决方法2，batch_size大于等于2 分析：感

7 条评论您还未登录，请先登录后发表或查看评论

已解决: ValueError: Error when checking input: expected input_1 to have 4 dimensions, but got array wit

猫头虎

11-10

908

喵，AI研发的小伙伴们，猫头虎博主今天要和大家探讨一个深度学习模型中常见的维度错误。🐱 🤖 我们会钻研这个错误背后的原因，并展示一些炫酷的代码，以确保你的神经网络不会在这种小错误上摔跤。准备好你的笔记本，让我们一起解决这个四维问题！📔在人工智能的世界里，正确的数据形状是模型能否成功学习的关键。通过仔细的数据预处理和严谨的测试，我们可以确保模型得到所需的正确输入，就像猫头虎精确地跳跃到它想要的地方一样。🐱。

报错：ValueError_ Expected more than 1 value per channel when training, got input size torch.Size

最新发布

一只努力学习的小碗~

07-03

153

记录报错解决方案

ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 8, 1

zy的博客

03-08

1319

深度学习

Expected more than 1 value per channel when training, got input size torch.Size([1, **])

程序员的博客

05-10

2932

目录出现这种问题大概有两种情况如果上面步骤三还没解决问题原因分析先定位下问题出现这种问题大概有两种情况第一种情况是训练时候出现，另一种是测试模型的时候出现出现这种问题一般是你输入的数据只有一张图片例如 [1,3,256,256] 步骤1 你试下改成 [2,3,256,256] 能否解决（如果无法加，请看下一个步骤），如果你是在训练模型，看步骤2 如果你是在测试模型，看步骤3 步骤2 如果你是在训练模型，在获取数据集时，将DataLoader中drop_las...

ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 256,

AliceH1226的博客

12-06

590

expected more than 1 value per channel batchsize设为2或以上可以解决

ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 2048

weixin_43925119的博客

11-18

1450

在多块GPU上训练模型时，出现如下错误：看很多博客说是：由于batchnorm层需要大于一个样本去计算其中的参数，解决方法是将dataloader的一个丢弃参数设置为true。网上查找的原因为模型中用了batchnomolization，训练中用batch训练的时候当前batch恰好只含一个sample，而由于BatchNorm操作需要多于一个数据计算平均值，因此造成该错误。于是查看我的训练batch_size=16, 训练集样本：验证集样本：测试集样本：计算了一下均没有剩下一个样本的情况，

expected more than 1 value per channel when training, got input size torch.s

04-28

它告诉我们，模型在训练的过程中期望每个通道传入的值应该不止1个（至少2个），但实际上输入的大小却只有torch.s。torch.s代表的是一个不确定的size，这说明了模型并没有正确的接收到输入。出现这个问题的原因可能...

ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 27, 1, 1])

01-30

这个错误通常出现在使用深度学习... raise ValueError("Expected more than 1 value per channel when training") # 继续进行训练的代码... ``` 请注意，这只是一个示例代码，你需要根据你的具体情况进行相应的修改。

ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 512, 1, 1])

07-28

- *1* *3* [ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1,](https://blog.csdn.net/weixin_47725177/article/details/128460187)[target="_blank" data-report...

ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 64, 1, 1])

08-15

- *1* [ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1,](https://blog.csdn.net/weixin_47725177/article/details/128460187)[target="_blank" data-report-...

Expected more than 1 value per channel when training, got input size torch.Size

小李的研究生学习日记

04-15

6047

Expected more than 1 value per channel when training, got input size torch.Size 训练的时候，batch size必须大于1，但是预测的时候，batch size可以等于1 原因：因为模型中用了batchnomolization，训练中用batch训练的时候当前batch恰好只含一个sample，而由于BatchNorm操作需要多于一个数据计算平均值，因此造成该错误。解决方法： 1. 加大batch_size 2.

讲解Expected more than 1 value per channel when training, got input size torch.Size

牛肉胡辣汤

12-12

1603

在训练深度学习模型时，遇到错误消息"Expected more than 1 value per channel when training, got input size torch.Size"时，我们需要检查数据预处理的过程，确保输入数据的形状满足模型的要求。通过检查数据形状、数据预处理代码和模型的输入层，我们可以找出错误的原因并进行修复。这样，我们就可以成功训练模型并获得预期的结果。

pytorch中的一个小坑：Expected more than 1 value per channel when training, got input size ...

一点痕迹

09-01

547

无

测试时模型时出现：Expected more than 1 value per channel when training, got input size torch.Size([])问题

YancyBill的博客

07-25

297

在测试单个实验数据时报错Expected more than 1 value per channel when training, got input size torch.Size([])，猜测应该是BN层出现了问题。并且通过model.eval()后发现没有解决该问题。

ValueError: Expected more than 1 value per channel when training, got input size [1, 256, 1, 1]的解决方案

weixin_47142735的博客

03-07

1227

在运行pytorch代码的时候，当其中存在BN层，并且你还存在输出的结果为1x1卷积，（如全局平均池化），在验证集的时候就很会出现这个问题： ValueError: Expected more than 1 value per channel when training, got input size [1, 256, 1, 1] 可是这个时候检查一下batch_sizes 发现明明是大于1的，之所以还会出现这个，就是你数据的总数除以batch_size不能整除，所以需要我们在dataloader中将dr

ValueError: Expected more than 1 value per channel when training, got input size ...

知识搬运者

12-19

764

ValueError: Expected more than 1 value per channel when training, got input size… pytorch 在训练时使用多卡，会出现ValueError: Expected more than 1 value per channel when training, got input 这个错误由于batchnorm层需要大于一个样本去计算其中的参数，解决方法是将dataloader的一个...

Expected more than 1 value per channel when training, got input size torch.Size([1, 64, 1, 1])

sinat_42367115的博客

08-16

878

记录一次因BN层和Batch_size=1出现的错误和检测神经网络模型框架的有用方法