Errors when I run train.py

tengshaofeng / ResidualAttentionNetwork-pytorch

a pytorch code about Residual Attention Network. This code is based on two projects from

681 stars 166 forks source link

Errors when I run train.py #37

Open KrystalCWT opened 3 years ago

KrystalCWT commented 3 years ago

File "/home//ResidualAttentionNetwork-pytorch-master/Residual-Attention-Network/model/attention_module.py", line 249, in forward out_interp3 = self.interpolation3(out_softmax3) + out_softmax2 RuntimeError: The size of tensor a (14) must match the size of tensor b (2) at non-singleton dimension 3

alakhrasSandy commented 3 years ago

any solution found regarding this?

alakhrasSandy commented 3 years ago

Found the issue from my case it was that i didnt resize/scale in the transformer , ( transforms.Scale(224) )