google-research / google-research/circuit_training
CPU RAM Usage
- Dominant language
- Python
- Stars
- 1.7k
- Forks
- 273
- PR merge metrics
- No merged PRs in 30d
Description
I'm having issues with the amount of CPU RAM being used by train_ppo.py. Specifically, over the course of training the memory usage steadily increases until there is no memory left, causing an error. This seems odd as I would expect the total memory usage to be roughly constant over the course of training as the model and dataset are both a fixed size. Does anyone have an idea as to why this is the case? Has anyone else experience similar issues, and been able to alleviate them?
Contributor guide
Research direction
Start by reproducing the memory increase while running train_ppo.py and inspect how CPU RAM changes over the course of training. Trace the components that retain memory, then verify that a representative training run maintains roughly constant usage and finishes without exhausting RAM.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100