search

wyf0912 / SinSR

[CVPR 2024] SinSR: Diffusion-Based Image Super-Resolution in a Single Step
Other
327 stars 19 forks source link

Asking about performance of SinSR #5

Closed pvtoan closed 7 months ago

pvtoan commented 7 months ago

Hi,

I have just tested the inference.py file (using SinSR-v1.pth) with a LR image. However, the result is poor compared to ResShift and the output image also includes something weird.

  1. I reduced chopsize to 256 because my GPU is only 24GB and it caused an Out-Of-Memory when I kept chopsize=512. Can I know what exact GPU VRAM you used when testing with chopsize=512

  2. I used the same settings as in defaults but the output image includes weird color .... Do you know if any settings I need to modify? 1st-lr lr3

wyf0912 commented 7 months ago

Please make sure that you have installed xformers successfully

24G VRAM shall be enough for inference with a chop size of 512.

image
pvtoan commented 7 months ago

Hi,

I found the problem and it works as normal now.

Thank you for your help!