DG-Net icon indicating copy to clipboard operation
DG-Net copied to clipboard

Published 20 hours ago • verified publisher icon NVlabs

Out of memory error while training the model

Open Haneesha123 opened this issue 3 years ago • 1 comments

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.

Haneesha123 avatar Apr 04 '21 08:04 Haneesha123

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

layumi avatar May 26 '21 07:05 layumi