ReferFormer icon indicating copy to clipboard operation
ReferFormer copied to clipboard

CPU Memory increasing when training

Open YRlin-12 opened this issue 2 years ago • 2 comments

Hi! I'm training the pretraining process on RefCOCO* datasets. But, the CPU memory increasing from 18G to 22G after 1 epoch. I'm worry about after a few epochs later, the CPU MEMORY will face OOM problem. (My device is 32G CPU) Do you have any idea about this?

YRlin-12 avatar Apr 20 '22 01:04 YRlin-12

Hi, I suppose you mean the GPU memory. We also use the GPUs with 32G memory, so we think there won't be a problem.

wjn922 avatar Apr 20 '22 12:04 wjn922

Sorry, I don't mean the GPU memory.

YRlin-12 avatar Apr 21 '22 07:04 YRlin-12