I am trying to upsample 5D tensor using nn.Upsample:
size2=(16, 56,56)
self.interpolation1 = nn.Upsample(size=size2,mode=‘trilinear’)
and then using it in forward as:
out_interp = self.interpolation1(out_middle_2r_blocks) + out_down_residual_blocks1
I get an error as:
return torch._C._nn.upsample_trilinear3d(input, _output_size(3), align_corners)
RuntimeError: Expected a list of 3 ints but got 2 for argument #2 'output_size'
Please help!