ToughStoneX / Self-Supervised-MVS

Pytorch codes for "Self-supervised Multi-view Stereo via Effective Co-Segmentation and Data-Augmentation"
152 stars 15 forks source link

size incompatibility #21

Closed TWang1017 closed 1 year ago

TWang1017 commented 1 year ago

Hi, thanks for the amazing work. I encountered that error below when run the train.py script. any ideas what happened?

File "c:\Users\xxx\Desktop\jdacs-ms\models\network.py", line 72, in forward conv6 = conv0 + self.conv6(conv5) RuntimeError: The size of tensor a (11) must match the size of tensor b (12) at non-singleton dimension 2