RuntimeError: CUDA out of memory.

[MrRainbowYoo]

I had set the batchsize is ONE, but there is still “CUDA out of memory”.

My GPU is 3060 and the memory is 12GB, so how could I train the model, thanks!

[LouisRouss]

Hi, thanks for your message.
Is your memory totally filled by the model itself or is it happening during the training loop ?
You can maybe start by reducing the size of the model via the conf file (number of res block for exemple or attention)

[MrRainbowYoo]

Thanks for your reply!
My configuration： BatchSize is 1，IMAGE_SIZE is 64，NUM_RESBLOCKS is 4，now the model is begin to train without errors.
And could you tell me how could I colorize a gray image using the model after completing training~