【问题解决】
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和实际输入的尺寸没有匹配上,所以只需要在报错代码前面断点查询卷积设置的输入和实际输入,对其进行修改就可以了。
来源地址:https://blog.csdn.net/weixin_43266423/article/details/128091945