Comments (2)
I encountered the same issue. Is there a way to reduce memory usage by modifying the code? thx
from trl.
Try setting gradient checkpointing to True, or you could use unsloth for much memory efficient fine tuning
from trl.
Related Issues (20)
- SFT - Dataset Packing causes C10d timeout
- TrlParser not respecting config
- Reproducing LLaVA with Mistral backened
- Bug in calling model.eval() in PPO
- vsft_llava.py:
- Fail to train with ORPOTrainer under multi-GPUs setting HOT 8
- NameError: name 'PeftConfig' is not defined HOT 4
- Loss value returned by the SFT Trainer
- Using PEFT causes model to not predict EOS
- Using SFTTrainer and the DeepSpeed Zero-3 Method training, the data map function runs, but the map function does not run in parallel GPU form. HOT 1
- IPO loss computation vs changes in DPO dataset format
- Anthropic HH new dataset format repeats the prompt HOT 1
- Group Relative Policy Optimization Trainer
- bugs in trained save and evaluation for DPO training with deepspeed_zero3 HOT 3
- config.json file is missing. HOT 3
- [Question] Does TRL support DPO trainer with simulated environment for generating training data like PPO's step-based training? HOT 1
- [Question]How should I combine DPOTrainer and Accelerate for training?
- CLI utils class cases seem to be incorrect HOT 2
- Training stops early
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from trl.