alibaba / alibaba/FederatedScope

GPU Memory Issue

Open
#775 10 comments 0 reactions 0 assignees View on GitHub
FS-LLM
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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.