search

jbr-ai-labs / mamba

This code accompanies the paper "Scalable Multi-Agent Model-Based Reinforcement Learning".
MIT License
46 stars 9 forks source link

Problem of memory leaking #12

Open BUPT-zeld151 opened 8 months ago

BUPT-zeld151 commented 8 months ago

Hello! I have noticed that the released code has the problem of memory leaking during training. The occupied system memory keeps growing throughout the training phase. Do you have any idea of fixing it? : )