wjiazheng / SwinPA-Net

SwinPA-Net: Swin Transformer-Based Multiscale Feature Pyramid Aggregation Network for Medical Image Segmentation
https://ieeexplore.ieee.org/document/9895210
5 stars 0 forks source link

Why it is not for binary #2

Open ymmm-4 opened 5 months ago

ymmm-4 commented 5 months ago
size mismatch for patch_embed.proj.weight: copying a param with shape torch.Size([128, 3, 4, 4]) from checkpoint, the shape in current model is torch.Size([128, 1, 4, 4]).
wjiazheng commented 5 months ago

Hi, we may need more details about this bug