alibaba / alibaba/FederatedScope
GPU Memory Issue
- Dominant language
- Python
- Stars
- 1.5k
- Forks
- 261
- PR merge metrics
- No merged PRs in 30d
Description
The gpu memory usage continues to increase after each round while finetuning LLM with an adapter. The gpu memory increment after each round was approximately the same. I speculate it's because that there are new clients joining in each round and there would be new model parameters. I've already set share_local_model and llm.adapter.mv_to_cpu to True, it should move the adapter to cpu after each round but why would the gpu memory still increase? I'd appreaciate it if anyone could help me with this issue. Thanks in advance!
Contributor guide
No contributing guide indexed for this repository
Research direction
The report names no files, tests, or entry points. Begin by reproducing the GPU-memory growth in the adapter fine-tuning path with share_local_model and llm.adapter.mv_to_cpu enabled; trace memory across rounds, and consider the work done when memory remains stable and the regression is covered by a test.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- distributed-systems, machine-learning
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100