search

wjn922 / ReferFormer

[CVPR2022] Official Implementation of ReferFormer
Apache License 2.0
327 stars 25 forks source link

CPU Memory increasing when training #14

Open YRlin-12 opened 2 years ago

YRlin-12 commented 2 years ago

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?

wjn922 commented 2 years ago

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.

YRlin-12 commented 2 years ago

Sorry, I don't mean the GPU memory.