Could you please suggest alternatives for Upsampling, How about I add trainable parameters in the axis which has less dimensions instead of adding zeros (upsampling). I have read that nn.ConvTranspose2d can help ? Will it be helpful. My model hasn’t improved much even after making the architecture better