I have changed the input to gray scale image. but the outputs are still zeros. It would be very helpful if you can lighten-up that ‘dim2’ and ‘image.permute’ .
After first optimizer step in my model, the values of tensor becomes zero…
Thank you