MASA-SR icon indicating copy to clipboard operation
MASA-SR copied to clipboard

Published 20 hours ago • verified publisher icon dvlab-research

cuda out of memory

Open zhh99 opened this issue 3 years ago • 1 comments

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

zhh99 avatar Oct 25 '21 05:10 zhh99

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).

SkyeLu avatar Oct 25 '21 06:10 SkyeLu