错误:Given groups=1, weight of size [256,1024,1, 1], expected input[1, 256, 64, 64] to have 1024,但是通道数是256
解决问题:
将变通道代码不在for循环即可,就解决此问题了。
错误:Given groups=1, weight of size [256,1024,1, 1], expected input[1, 256, 64, 64] to have 1024,但是通道数是256
解决问题:
将变通道代码不在for循环即可,就解决此问题了。