IanTaehoonYoo / semantic-segmentation-pytorch

Pytorch implementation of FCN, UNet, PSPNet, and various encoder models.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

: Given transposed=1, weight of size 1024 2048 4 4, expected input[8, 2048, 10, 10] to have 1024 channels, but got 2048 channels instead

jiafengshen opened this issue · comments

Hi, @jiafengshen
Could you give me some information you used the model and input data?

Thanks
Best regards,

I found that problem when testing the model 'unet_resnet34' and I fixed it.
I update the project, so you should update it. use the command: pip install --upgrade seg-torch

Thanks