RuntimeError:通道不一致
问题:
RuntimeError: Given transposed=1, weight of size 53 32 2 2, expected input[1, 74, 256, 256] to have 53 channels, but got 74 channels instead
很明显channels不一致导致的,卷积核size=(2,2),希望53=>>32,但是你输入的通道数是74,对应不上所以报错,53或74有问题
排查:
问题定位,就是这个if else后置的问题,理解错误了,以为这个if else的对象是dim_tag,其实是整个加法式子
源码里的参数都是21,不知道最后输出的到底是哪个了,换个数简单验证一下,over!