daitao / SAN

Second-order Attention Network for Single Image Super-resolution (CVPR-2019)
557 stars 105 forks source link

Attempting to Upscale 720p images, CUDA Out of Memory #31

Closed nelaturuharsha closed 4 years ago

nelaturuharsha commented 4 years ago

Hello,

Thank you so much for the excellent contribution! I am attempting to upscale a standard 720p image using SAN. I am using the 3x model, and have tried the following GPUs

  1. 2060
  2. 2x 2080Tis
  3. Tesla V100

And the model is OOM (out of memory) in all of them. I'm actually at a loss, how should I proceed with regards to the same in order to run inference?

Thank you.

Akshaypatil7 commented 4 years ago

Hi, did you solved this error?

nelaturuharsha commented 4 years ago

You can SR on patches and then stitch them together.