bigmb/Unet-Segmentation-Pytorch-Nest-of-Unets

dim

giter915 opened this issue · 1 comments

giter915 commented

File "F:\some_project\Unet-Segmentation-Pytorch-Nest-of-Unets-master\Unet-Segmentation-Pytorch-Nest-of-Unets-master\Models.py", line 102, in forward
d5 = torch.cat((e4, d5), dim=1)
RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 61 but got size 60 for tensor number 1 in the list.

bigmb commented

Whats the data shape?
Make sure its in muliple of 2^4.
Or reduce the depth of the model.
(You can check other closed issue for this too.)