Thank you @JuanFMontesinos . it is causing an error
if input.dim() == 3 and mode == 'nearest': AttributeError: 'tuple' object has no attribute 'dim'
if I remove the torch.split like imgs = activations['up2'].squeeze() then, the size of img is torch.Size([128, 120, 160]) but it’s causing an error that Got 3D input, but bilinear mode needs 4D input