I am getting the following error in concat2 = torch.cat([x2, deconv2, upsamp3to2], dim=1) this line of code from VTN function.
torch.cat(): Sizes of tensors must match except in dimension 1. Got 23 and 24 in dimension 2 (The offending index is 1)