nikhilroxtomar/Retina-Blood-Vessel-Segmentation-in-PyTorch

Bug

Closed this issue · 1 comments

MarkusKhoa commented

RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 56 but got size 64 for tensor number 1 in the list.

You should check the tensors' shape in decoding steps, when you shall crop the tensors' sizes in the encoder block before doing skip connection to tensor at the decoder block.

MarkusKhoa commented

I've noticed that if I must the input size of 512, this works. Sorry for my fault.

  • 👍1