aimagelab/meshed-memory-transformer

RL training stage,out of memory

TBI805 opened this issue · 0 comments

Switching to RL, after every epoch training, the memory usage will increase. Finally, the system will be out of memory, and kill process.
Could you please give me some tips to solve it? Thank you very much!