pytorch训练时报input size错误问题

错误提示：

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

在用pytorch做模型训练的，之前可以正常运行的代码，突然出现这个问题，一开始被吓一跳。从字面意思看，是输入的size不能为1。

通过一番查找发现，是在训练阶段batch_size不能为1，则就需要总的训练图片数量/batch_size不能余1，因此调整一下train batch_size，则可解决。

而在model.eval()阶段，则没有batch_size一定需要为1的要求。

pytorch训练时报input 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] ...
送你9个快速使用Pytorch训练解决神经网络的技巧（附代码）
来源:读芯术本文约4800字,建议阅读10分钟. 本文为大家介绍9个使用Pytorch训练解决神经网络的技巧图片来源:unsplash.com/@dulgier 事实上,你的模型可能还停留在石器时 ...
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必须大于 ...
torchsummary 中input size 异常的问题
爪爪熊原创文章----谢绝转载本文解决问题: torchsummary针对多个输入模型的时候,其输出信息中input size等存在着错误,这里提供方案解决这个错误. 当我们使用pytorch搭建好 ...
pytorch dali 加速 dali支持的数据处理列表，mxnet tensorflow caff读取数据转换 pytorch训练
使用dali加速,前提是 gpu没有跑满,不然效果也不大 +apex 混合精度训练 5.tf ,mxnet ,caffe, 数据输入,转换成 pytorch 训练 https://github.com ...
关于yolov5训练时报错cuda内存不足问题
RuntimeError: CUDA out of memory. Tried to allocate 4.08 GiB (GPU 0; 10.76 GiB total capacity; 414.9 ...
PyTorch训练加速技巧
PyTorch训练加速技巧由于最近的程序对速度要求比较高,想要快速出结果,因此特地学习了一下混合精度运算和并行化操作,由于已经有很多的文章介绍相关的原理,因此本篇只讲述如何应用PyTorch实现混合 ...
让PyTorch训练速度更快，你需要掌握这17种方法
选自efficientdl.com 作者:LORENZ KUHN 机器之心编译编辑:陈萍掌握这 17 种方法,用最省力的方式,加速你的 Pytorch 深度学习训练. 近日,Reddit 上一个帖 ...
PyTorch训练加速17种技巧
点击上方"小白学视觉",选择加"星标"或"置顶" 重磅干货,第一时间送达文自机器之心作者:LORENZ KUHN 编辑:陈萍掌握这 ...

pytorch训练时报input size错误问题

pytorch训练时报input size错误问题相关推荐

最新文章

热门文章