LSGM icon indicating copy to clipboard operation
LSGM copied to clipboard

CUDA out of memory

Open jeeyung opened this issue 2 years ago • 0 comments

Hi, I used 32x32 size images. With 12GB GPUs, I reduce the batch size from 16 to 2 or 1(per gpu).

However, still I got CUDA out of memory error. Could you explain to me how to deal with it?

I used the same hyper parameters that you suggested for CIFAR10.

jeeyung avatar Jul 01 '22 15:07 jeeyung