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

已于 2023-07-13 23:16:23 修改
阅读量194 收藏 1
点赞数
分类专栏： python报错合集 文章标签： 人工智能 机器学习 深度学习
于 2023-07-13 23:06:04 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_45588505/article/details/131713446
版权

在使用pytorch进行模型训练时报错:
ValueError: Expected more than 1 value per channel when training, got input size torch.Size

出现上述问题是因为batchsize=1,因为 BN层里，遇到了batch=1的情况

解决方案：

1、如果是在训练的时候，则drop_last=True

train_loader = DataLoader(trainset, batch_size=batch_size,num_workers=4, shuffle=True,drop_last=True)

2、如果是在测试的时候则加上model.eval()

model.eval()关闭BatchNorm，使用训练时计算得到的均值和方差

零点_ling
关注
专栏目录
解决报错：ValueError: Expected more than 1 value per channel when training
羊城迷鹿的博客
12-21 5716
写了一个图片分类网络，之前跑一直没问题，换了一下训练集占的比例之后跑到最后一个batch突然报了这个错 定位到出错的语句为 我们知道bn操作需要计算一个batch内的均值和方差，所以，最后一个batch内到底包含几个样本呢？ 这样就得每次把这个单独的样本丢掉啦，dataloader改写成下面这种形式，加上drop_last=True ...
YOLOv5-5.x版本报错Expected more than 1 value per channel when training, got input size torch.Size([1, *]
younger14的博客
04-21 1546
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
讲解Expected more than 1 value per channel when training, got input size torch.Size
牛肉胡辣汤
12-12 1616
在训练深度学习模型时，遇到错误消息"Expected more than 1 value per channel when training, got input size torch.Size"时，我们需要检查数据预处理的过程，确保输入数据的形状满足模型的要求。通过检查数据形状、数据预处理代码和模型的输入层，我们可以找出错误的原因并进行修复。这样，我们就可以成功训练模型并获得预期的结果。
Expected more than 1 value per channel when training, got input size torch.Size([1, 256, 1, 1]
It is me 的博客
07-20 2124
SSD Expected more than 1 value per channel when training, got input size torch.Size([1, 256, 1, 1]) 原因：模型中用了batchnomolization，训练中用batch训练的时候当前batch恰好只含一个sample，而由于BatchNorm操作需要多于一个数据计算平均值，因此造成该错误。 解决方法1 ：在torch.utils.data.DataLoader类中或自己创建的继承于DataLo..
BUG-Expected more than 1 value per channel when training, got input size torch.Size([1, xxx, 1, 1])
zwhdldz的博客
12-08 680
当size=Size([1, xxx, 1, 1]), size_prods是size[0] * size[2] * size[3]当size=Size([1, xxx]), size_prods是size[0]1.优先建议,调整输入尺寸或者网络结构使得 size[2]和size[3]不为1。上面的size是Size([1, xxx, 1, 1])2.方法1如果不能用,则batch_size改成大于1。就可以,这就是为什么网上通用解释都是这个.因此,为了避免这个错误,我们把。
python报错系列(7)--ValueError: Expected more than 1 value perchannel when training, got input size ....
热门推荐
用余生去守护的博客
01-24 1万+
活用知识和经验来寻找新点子、新创意，就是培养创造性思考所需的态度。
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.s
04-28
它告诉我们，模型在训练的过程中期望每个通道传入的值应该不止1个（至少2个），但实际上输入的大小却只有torch.s。torch.s代表的是一个不确定的size，这说明了模型并没有正确的接收到输入。 出现这个问题的原因可能...
pytorch中的一个小坑：Expected more than 1 value per channel when training, got input size ...
一点痕迹
09-01 559
无
ValueError:Expected more than 1 value per channel when training, got input size torch.Size([1,1024])
weixin_43941438的博客
11-29 1091
报错内容：ValueError:Expected more than 1 value per channel when training, got input size torch.Size([1,1024])报错原因：网络中含有bn或dropout，训练模式中bn和dropout的行为方式和预测模式是不一样的。在训练部分的代码前面加上：model.train()在测试部分的代码前面加上：model.eval()
ValueError: Expected more than 1 value per channel when training, got input size torch.Size
深度学习炼丹师，偶尔分享炼丹技术!
03-15 619
[问题描述]: ValueError: Expected more than 1 value per channel when training, got input size torch.Size可能原因1：输出张量x尺寸过小解决办法1：调大尺寸。
Expected more than 1 value per channel when training, got input size torch.Size
jacke121的专栏
05-17 8897
Expected more than 1 value per channel when training, got input size torch.Size 简单说batch size必须大于1 2.网上查找的原因为模型中用了batchnomolization，训练中用batch训练的时候当前batch恰好只含一个sample，而由于BatchNorm操作需要多于一个数据计算平均值，因此造成该错误。 ...
raise ValueError(‘Expected more than 1 value per channel when training, got input size {}‘
最新发布
m0_66237895的博客
04-26 1158
可能因为self.training=True 导致BatchNorm操作认为还在训练阶段。把batch_size的值改为大于1的数。
【pytorch】ValueError: Expected more than 1 value per channel when training
做减求空
10-15 3378
ValueError: Expected more than 1 value per channel when training 解决方案(亲测)
ValueError: Expected more than 1 value per channel when training, got input size ...
知识搬运者
12-19 774
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的一个...
ValueError: Expected more than 1 value per channel when training
qq_37668436的博客
09-27 2140
前言 报错这个的可能就三种，我全踩过了，记录一下造福后人 原因 训练的时候，网络有bn层，但是输入的特征的第一个维度是1，例如：[1,2048]这样的就会报错，因为bn层要计算均值方差，一张图没法统计。 出现情况 dataloader没有drop last，可能会出现最后一个batch的数量为1 评估的时候没有写model.eval() model的结构中用列表或者字典存了bn 解决 第一种很好解决，将DataLoader中drop_last设置为True即可 第二种更好解决，直接加一行model.ev
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值