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

报错代码：

File "E:/project/fenge/jacke121-U-2-Net-master_mse/nets/bisnet.py", line 106, in forward
feat = self.bn(feat)
File "D:\Users\Administrator\miniconda3\lib\site-packages\torch\nn\modules\module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "D:\Users\Administrator\miniconda3\lib\site-packages\torch\nn\modules\batchnorm.py", line 136, in forward
self.weight, self.bias, bn_training, exponential_average_factor, self.eps)
File "D:\Users\Administrator\miniconda3\lib\site-packages\torch\nn\functional.py", line 2054, in batch_norm
_verify_batch_size(input.size())
File "D:\Users\Administrator\miniconda3\lib\site-packages\torch\nn\functional.py", line 2037, in _verify_batch_size
raise ValueError('Expected more than 1 value per channel when training, got input size {}'.format(size))
ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 128, 1, 1])

原因：batch_size为1时，bn层报错

解决方法：

网络开启预测模式：

model.eval()

就可以了

或者batch_size大于1.

bn层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
Expected more than 1 value per channel when training, got input size torch.Size 训练的时候,batch size必须大于 ...
Debug Pytorch: ValueError: Expected more than 1 value per channel when training, got input size tor
ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 512] ...
Expected more than 1 value per channel when training, got input size torch.Size([1, **])
目录出现这种问题大概有两种情况如果上面步骤三还没解决问题原因分析先定位下问题出现这种问题大概有两种情况第一种情况是训练时候出现,另一种是测试模型的时候出现出现这种问题一般是你输入的数据只 ...
【pytorch】ValueError: Expected more than 1 value per channel when training
ValueError: Expected more than 1 value per channel when training 一.解决方案二.问题分析三.深入分析,修改源码一.解决方案 da ...
tensorflow中的BN层实现
import tensorflow as tf import numpy as np import matplotlib.pyplot as plt from tensorflow.keras imp ...
网络骨架：Backbone(神经网络基本组成——BN层、全连接层)
BN层为了追求更高的性能,卷积网络被设计得越来越深,然而网络却变得难以训练收敛与调参.原因在于,浅层参数的微弱变化经过多层线性变化与激活函数后会被放大,改变了每一层的输入分布,造成深层的网络需要不断 ...
pytorch 批量归一化BatchNorm1d和BatchNorm2d的用法、BN层参数 running_mean running_var变量计算验证
前提知识 BN层包括mean var gamma beta四个参数,.对于图像来说(4,3,2,2),一组特征图,一个通道的特征图对应一组参数,即四个参数均为维度为通道数的一维向量,图中gamma.b ...
Interview：算法岗位面试—上海某科技公司算法岗位(偏AI算法，国企)技术面试之BN层的认知、BP的推导、GD优化的几种改进等
Interview:算法岗位面试-上海某科技公司算法岗位(偏AI算法,国企)技术面试之BN层的认知.BP的推导.GD优化的几种改进等导读:关于神经网络,问的比较深,因为博主做过总结,所以用自己的语言 ...
DL之DNN优化技术：自定义MultiLayerNetExtend算法(BN层使用/不使用+权重初始值不同)对Mnist数据集训练评估学习过程
DL之DNN优化技术:自定义MultiLayerNetExtend算法(BN层使用/不使用+权重初始值不同)对Mnist数据集训练评估学习过程目录输出结果设计思路核心代码更多输出相关文章: ...

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

bn层Expected more than 1 value per channel when training, got input size torch.Size相关推荐

最新文章

热门文章