search

dvlab-research / MASA-SR

MASA-SR: Matching Acceleration and Spatial Adaptation for Reference-Based Image Super-Resolution (CVPR2021)
161 stars 20 forks source link

cuda out of memory #9

Open zhh99 opened 3 years ago

zhh99 commented 3 years ago

test时12G显存不够用，报错内存不够，训练才占8G，是什么原因

SkyeLu commented 3 years ago

Since the resolution of testing images is much larger than that of training images. Better to test on GPU whose memory is larger than 16G. Or you may test on CPU only (use command --gpu_ids -1).