SR_scale=2 has problems

dvlab-research / MASA-SR

MASA-SR: Matching Acceleration and Spatial Adaptation for Reference-Based Image Super-Resolution (CVPR2021)

160 stars 20 forks source link

SR_scale=2 has problems #20

Closed xianrui-luo closed 2 years ago

xianrui-luo commented 2 years ago

Hi, when i set the sr_scale from 4 to 2, there would be an error: RuntimeError: Caught RuntimeError in replica 0 on device 4. RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

It is in MASA_arch.py", line 122, fea_L1 = self.blk_L1(self.act(self.conv_L1(x))) I wonder what is happening

SkyeLu commented 2 years ago

Hi, our model is only designed for x4 SR. If you want to test on x2 SR, the model architecture should be modified and retrained.

xianrui-luo commented 2 years ago

Thank you for replying.