search

MrGiovanni / UNetPlusPlus

[IEEE TMI] Official Implementation for UNet++
Other
2.26k stars 538 forks source link

RuntimeError: Given transposed=1, weight of size [256, 128, 2, 2], expected input[12, 64, 256, 256] to have 256 channels, but got 64 channels instead #73

Open zhouwang0924 opened 2 years ago

LepusMaximus commented 2 years ago

i am getting the same error. i am trying to train a 2d UNetPlusPlus. Can Anyone resolve this? File "/home/maxi/anaconda3/envs/nnUNetpp/lib/python3.9/site-packages/torch/nn/modules/conv.py", line 916, in forward return F.conv_transpose2d( RuntimeError: Given transposed=1, weight of size [256, 128, 2, 2], expected input[5, 64, 384, 384] to have 256 channels, but got 64 channels instead

web-girlfriend commented 2 years ago

i am getting the same error. i am trying to train a 2d UNetPlusPlus. Can Anyone resolve this? File "/home/maxi/anaconda3/envs/nnUNetpp/lib/python3.9/site-packages/torch/nn/modules/conv.py", line 916, in forward return F.conv_transpose2d( RuntimeError: Given transposed=1, weight of size [256, 128, 2, 2], expected input[5, 64, 384, 384] to have 256 channels, but got 64 channels instead

I also encountered the same problem!!! Who can help me