RuntimeError: Given groups=1, weight of size 64 128 1 7, expected input[16,

268 阅读1分钟

如果输入和输出匹配的情况下有这样的错误,请检查在定义卷积的时候有没有名字重复的,

例:

("2conv1,3_bn1", ConvBN(channelNum * 4, channelNum * 2, [1, 3])),

2conv1,3_bn1这个是定义卷积的名字,如果存在重复的,则可能引起这样的错误。