[pytorch/onnx]model nn.conv2, in particular dimension 1 conflicts 4096 does not match 512;

I have the same problem. How did you solve it? 我遇到一样的问题,您是怎么解决的?