RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1. Got 128 and 256 in dimension 3 at /opt/conda/conda-bld/pytorch_1556653099582/work/aten/src/THC/generic/THCTensorMath.cu:71

/home/shru/abdominal-multi-organ-segmentation-master/net/ResUnet_CE.py:229: UserWarning: nn.init.kaiming_normal is now deprecated in favor of nn.init.kaiming_normal_.
nn.init.kaiming_normal(module.weight.data, 0.25)
/home/shru/abdominal-multi-organ-segmentation-master/net/ResUnet_CE.py:230: UserWarning: nn.init.constant is now deprecated in favor of nn.init.constant_.
nn.init.constant(module.bias.data, 0)
/home/shru/anaconda3/lib/python3.6/site-packages/torch/nn/functional.py:2457: UserWarning: nn.functional.upsample is deprecated. Use nn.functional.interpolate instead.
warnings.warn(“nn.functional.upsample is deprecated. Use nn.functional.interpolate instead.”)
/home/shru/anaconda3/lib/python3.6/site-packages/torch/nn/functional.py:2539: UserWarning: Default upsampling behavior when mode=trilinear is changed to align_corners=False since 0.4.0. Please specify align_corners=True if the old behavior is desired. See the documentation of nn.Upsample for details.
“See the documentation of nn.Upsample for details.”.format(mode))
Traceback (most recent call last):
File “val.py”, line 144, in
outputs = net(ct_tensor)
File “/home/shru/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py”, line 493, in call
result = self.forward(*input, **kwargs)
File “/home/shru/anaconda3/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py”, line 150, in forward
return self.module(*inputs[0], **kwargs[0])
File “/home/shru/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py”, line 493, in call
result = self.forward(*input, **kwargs)
File “/home/shru/abdominal-multi-organ-segmentation-master/net/ResUnet_CE.py”, line 215, in forward
inputs_stage2 = torch.cat((temp, inputs), dim=1)
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1. Got 128 and 256 in dimension 3 at /opt/conda/conda-bld/pytorch_1556653099582/work/aten/src/THC/generic/THCTensorMath.cu:71
Hello, I am very new to pytorch. I am trying run a project available on github for abdominal multi organ segmentation and I keep getting this error. Any help would be great.