通道数不匹配问题

[Luogic]

RuntimeError: Given groups=1, weight of size [256, 3, 5, 5], expected input[128, 32, 3, 32] to have 3 channels, but got 32 channels instead，多个训练和测试脚本都会报这个错，交换了一下前后位置解决了，但不确定是否正确，请问应该如何修正？