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

训练的时候，batch size必须大于1

但是预测的时候，batch size可以等于1

解决方法：

1. 加大batch_size

2.网络设置eval模式：

model=MyModel()
model.eval()

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

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
报错代码: File "E:/project/fenge/jacke121-U-2-Net-master_mse/nets/bisnet.py", line 106, in for ...
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 ...
Fabric2.2 error 总结：WARN 0cf access denied: channel expected MSP ID Org1MSP, received Org3MSP channel
在搭好网络和部署好链码的情况下,使用Org1的锚节点进行初始化账本 peer chaincode invoke -n boxcc -c '{"Args":["Init&q ...
PyTorch踩过的12坑 | CSDN博文精选
作者 | hyk_1996 来源 | CSDN博客 1. nn.Module.cuda() 和 Tensor.cuda() 的作用效果差异无论是对于模型还是数据,cuda()函数都能实现从CPU到G ...
pytorch统计矩阵非0的个数_PyTorch常见的12坑
原标题:PyTorch常见的12坑 1. nn.Module.cuda 和 Tensor.cuda 的作用效果差异无论是对于模型还是数据,cuda函数都能实现从CPU到GPU的内存迁移,但是他们的作 ...
【Pytorch神经网络实战案例】27 MaskR-CNN内置模型实现语义分割
1 PyTorch中语义分割的内置模型在torchvision库下的models\segmentation目录中,找到segmentation.Py文件.该文件中存放着PyTorch内置的语义分割模 ...
pytorch常见错误（持续更新）
1. nn.Module.cuda() 和 Tensor.cuda() 的作用效果差异无论是对于模型还是数据,cuda()函数都能实现从CPU到GPU的内存迁移,但是他们的作用效果有所不同. 对于n ...

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相关推荐

最新文章

热门文章