Out of memory error while training the model

[Haneesha123]

Hello,
When I try to train your model, I am getting the Runtime error:CUDA: out of memory error. Can you suggest me anything to get rid of this error?

Thank you.

[layumi]

Hi @Haneesha123
You may try fp16 to save the GPU memory cost.