调整输入通道数以匹配期望值[4, 512, 64, 64]中的256个通道

【问题解决】

Given groups=1, weight of size [256, 256, 3, 3], expected input[4, 512, 64, 64] to have 256 channels, but got 512 channels instead

这个应该是很常见的问题了。直接翻译过来就是:在给定组=1,大小权重[256,256,3,3],预期输入[4,512,64,64]具有256个通道,但得到了512个通道。直白点说就是第2位置的参数没设置对应上,要么第二个都是256,要么第二个都是512。

Given groups=1, weight of size [256, 256, 3, 3]
代表卷积核的channel 大小为 256 ,大小为3*3,batchsize大小为256。

像下面这种:

RuntimeError: Given groups=1, weight of size [64, 1, 3, 3], expected input[1, 3, 512, 512] to have 1 channels, but got 3 channels instead

第2位置的参数设置对应上的话,要么第二个都是1,要么第二个都是3。

这个问题出现的是因为卷积设置的输入channels和实际输入的尺寸没有匹配上,所以只需要在报错代码前面断点查询卷积设置的输入和实际输入,对其进行修改就可以了。

物联沃分享整理
物联沃-IOTWORD物联网 » 调整输入通道数以匹配期望值[4, 512, 64, 64]中的256个通道

发表评论