RuntimeError: Given groups=1, weight of size 32 32 1 1, expected input[5, 51, 400, 400] to have 32

用MultiResUnet跑脊柱数据集出现了RuntimeError: Given groups=1, weight of size 32 32 1 1, expected input[5, 51, 400, 400] to have 32 channels, but got 51 channels instead错误，请问怎么修改，具体是修改什么位置。