google-research / google-research/circuit_training

CPU RAM Usage

Open
#44 1 comment 0 reactions 0 assignees View on GitHub
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

Open the contributing 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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.