I am getting the error when running the Deep Image Prior’s Super-Resolution. I have attached the screenshots below. Can someone tell me where I am going wrong? Also, some people asked me to reduce the batch size and I dont know how to do it.
p.s. I am using CUDA 9.2 and GPU 12GB which is sufficient to run this.
Screenshot-2018-9-3 super-resolution.png1920×907 107 KB
 
Screenshot-2018-9-3 super-resolution(3).png1920×907 96.7 KB
 
Screenshot-2018-9-3 super-resolution(1).png1920×907 250 KB
 
Screenshot-2018-9-3 super-resolution(2).png1920×907 88.3 KB
 
Screenshot-2018-9-3 super-resolution(4).png1920×907 83.8 KB