search

NVlabs / DG-Net

:couple: Joint Discriminative and Generative Learning for Person Re-identification. CVPR'19 (Oral) :couple:
https://www.zdzheng.xyz/publication/Joint-di2019
Other
1.27k stars 230 forks source link

Out of memory error while training the model #71

Open Haneesha123 opened 3 years ago

Haneesha123 commented 3 years ago

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 commented 3 years ago

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