Super-resolution - Pytorch to caffe2 using ONNX : Wrong output

1

Hi all, I am running this code from official documentation - I am not getting a proper image as the output

Super-resolution - Pytorch to caffe2 using ONNX

https://pytorch.org/tutorials/advanced/super_resolution_with_caffe2.html

I get a Blacked out image as an output instead of the extended size of the image

wrong output image - https://drive.google.com/file/d/1Q46dUuVAjhRGn8uLTUvdi0fnfvfkmJOM/view?usp=sharing