Hi, thank you for sharing the code.
in your cuda implementation, if I comment the image resize function, the output image will only display the area of (224,224) and the remaining is black. I wonder why ? is it possible to display the real size of reconstructed image without reshape?