Errors when I run train.py

[KrystalCWT]

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]

any solution found regarding this?

[alakhrasSandy]

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