原文:深度学习报错:Expected more than 1 value per channel when training, got input size torch.Size([1, 256])

造成该结果的操作:修改batch size to 根本原因:模型中用了batchnomolization,训练中用batch训练的时候当前batch恰好只含一个sample,而由于BatchNorm操作需要多于一个数据计算平均值,因此造成该错误。 解决方法:在torch.utils.data.DataLoader类中或自己创建的继承于DataLoader的类中设置参数drop last True ...

2022-03-24 16:48 0 2722 推荐指数:

查看详情

 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM