The size of tensor a (7) must match the size of tensor b (8) at non-singleton dimension 3

UARK-AICV / MEGANet

[WACV 2024] An implementation of MEGANet for polyp segmentation with multi-scale edge-guided attention

https://arxiv.org/abs/2309.03329

43 stars 3 forks source link

Closed wuhuqifei0 closed 4 months ago

wuhuqifei0 commented 4 months ago

Hello author, I encountered an issue with mismatched tensors during training. How can I resolve it?