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!