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

最新推荐文章于 2023-12-12 11:59:42 发布
最新推荐文章于 2023-12-12 11:59:42 发布
阅读量6.8k 收藏 1
点赞数 5
分类专栏： 工具的使用 文章标签： 深度学习 pytorch 机器学习 神经网络
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/yy19980310/article/details/117827682
版权

由于BatchNorm层需要大于一个图像输入，当最后一个批数据恰好只剩一个图像就会报此错误，解决方案：

if num_of_imgs > 1:
'''continue training'''

认为设置长度小于2丢掉就好了

补充：

2.BN带来的好处。

(1) 减轻了对参数初始化的依赖，这是利于调参的朋友们的。

(2) 训练更快，可以使用更高的学习率。

(3) BN一定程度上增加了泛化能力，dropout等技术可以去掉。

3.BN的缺陷

从上面可以看出，batch normalization依赖于batch的大小，当batch值很小时，计算的均值和方差不稳定。



作者：言有三
链接：https://www.zhihu.com/question/38102762/answer/607815171
来源：知乎
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

pytorch BN层报错：ValueError: expected 4D input (got 2D input)
Hello Word!
05-20 9048
报错： Traceback (most recent call last): File "train_noPfc.py", line 201, in <module> main(args_) File "train_noPfc.py", line 160, in main f_masked, focc_masked, output, output_occ, f_diff, out = backbone(img1, img2) File "/home/user1/min
解决报错：ValueError: Expected more than 1 value per channel when training
羊城迷鹿的博客
12-21 5735
写了一个图片分类网络，之前跑一直没问题，换了一下训练集占的比例之后跑到最后一个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 1592
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遇到报错Expected more than 1 value per channel when training, got input size torch.Size
hjxu2016的博客
11-01 2万+
Pytorch遇到 Expected more than 1 value per channel when training, got input size torch.Size([1, 256, 1, 1]) 楼主遇到这个报错，通过pycharm定位到断言的位置 def _verify_batch_size(size: List[int]) -> None: # XXX: JIT script does not support the reduce from functools, and m
Expected more than 1 value per channel when training, got input size torch.Size
小李的研究生学习日记
04-15 6056
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 1619
在训练深度学习模型时，遇到错误消息"Expected more than 1 value per channel when training, got input size torch.Size"时，我们需要检查数据预处理的过程，确保输入数据的形状满足模型的要求。通过检查数据形状、数据预处理代码和模型的输入层，我们可以找出错误的原因并进行修复。这样，我们就可以成功训练模型并获得预期的结果。
【Pytorch】报错：ValueError: Expected more than 1 value per channel when training, got input size [1, 768
mjiansun的专栏
07-04 1万+
Traceback (most recent call last): File "train_ammeter_twoclass.py", line 189, in <module> train(epoch) File "train_ammeter_twoclass.py", line 133, in train outputs = net(inputs) ...
pytorch的运行错误及解决办法（一）RuntimeError: Expected object of scalar type Float but got scalar type Double...
weixin_43681705的博客
04-24 2万+
RuntimeError: Expected object of scalar type Float but got scalar type Double for sequence elment 2 in sequence at position #1 ‘tensors’ 在跑python代码的时候，遇到的有关pytorch的错误，RuntimeError: Expected object of ...
OSError Unable to load weights from pytorch checkpoint file for pytorch_model.bin解决方案
weixin_43178406的博客
07-07 6万+
 本文主要介绍了OSError Unable to load weights from pytorch checkpoint file for pytorch_model-00003-of-00003.bin解决方案，希望能对学习大模型的同学们有所帮助。 文章目录 1. 问题描述 2. 解决方案
ImportError: libcudart.so.10.2: cannot open shared object file: No such file or directory解决方案
热门推荐
weixin_43178406的博客
06-05 9万+
本文主要介绍了ImportError: libcudart.so.10.2: cannot open shared object file: No such file or directory解决方案，希望能对学习paddlepaddle的同学们有所帮助。 文章目录 1. 问题描述 2. 解决方案
ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 256,
qq_42191909的博客
03-04 576
pytorch在对以经训练好的网络进行测试时发现以上问题，在装载网络后添加model.eval（），问题得到解决。 model = nn.DataParallel(model).cpu() model.load_state_dict(torch.load(path, map_location=torch.device('cpu')), False) model.eval() 此代码段有三个需要注意的地方 nn.DataParallel(model) 如果在训练网络时加入了这个函数，在使用训练好
【Pytorch】ValueError: Expected more than 1 value per channel when training，got input size...
Asunnayy的博客
03-09 677
ValueError: Expected more than 1 value per channel when training，got input size… 在 BatchNorm 的forward地方报错 ValueError: Expected more than 1 value per channel when training，got input size... 如果输入是[N,X,1]需要[N,X,1+] 流行解释：训练中用batch训练的时候当前batch恰好只含一个sample，而由于B
ValueError(‘Expected more than 1 value per channel when training, got input size {}‘.format(size))
jiayoudangdang的博客
07-23 959
报错：ValueError('Expected more than 1 value per channel when training, got input size {}'.format(size)) 错误原因：batch中仅有一个样本，调用batch_norm就会有这个报错： 解决办法：重新计算num_samples和batchsize，保证num_samples%batchsi不是1即可 ...
Debug Pytorch: ValueError: Expected more than 1 value per channel when training, got input size tor
yuetan的博客
04-01 649
ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 512]) 当模型中有nn.BatchNorm时，batch不能设置为1，也不能出现样本最后一个batch可能就剩一个样本 解决方法 batch_size > 1 DataLoader中设置 drop_last=True ......
Expected more than 1 value per channel when training, got input size torch.Size......
张帅影&RS&DeepLearning
06-12 6310
完整问题描述 在将自己制作的数据集输入到模型中出现如下报错： ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 512, 1, 1]) 解决方案 由于我自己也没有设置batchsize，所以并没有网上提供的那种含有DataLoader的类，看了不少经验帖之后，发现有种方法能够很好的解决，就是在训练模型前加入： model.eval() 如图所示： 然后就可以解
Expected more than 1 value per channel when training, got input size torch.Size([1, 64, 1, 1])
sinat_42367115的博客
08-16 893
记录一次因BN层和Batch_size=1出现的错误和检测神经网络模型框架的有用方法
Expected more than 1 value per channel when training, got input size torch.Size([1, 256, 1, 1]
It is me 的博客
07-20 2134
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..
ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 27, 1, 1])
最新发布
01-30
这个错误通常出现在使用深度学习框架PyTorch进行训练时。它表示在训练过程中，输入的通道数不符合预期。具体来说，它要求每个通道至少有两个值，但是你的输入只有一个值。 解决这个问题的方法是检查你的输入数据的维度是否正确，并确保每个通道至少有两个值。你可以使用`torch.Size`对象来检查输入数据的维度。 以下是一个解决这个问题的示例代码： ```python import torch input_size = torch.Size([1, 27, 1, 1]) # 检查每个通道的值数量 for channel in input_size[1:]: if channel < 2: raise ValueError("Expected more than 1 value per channel when training") # 继续进行训练的代码... ``` 请注意，这只是一个示例代码，你需要根据你的具体情况进行相应的修改。
评论 4
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值