Comments (3)
lvwerra
commented on September 26, 2024
Will have a look!
from trl.
22Mukesh22
commented on September 26, 2024
Hi @lvwerra Any fix on the above error ?
I was running the notebook '04-gpt2-sentiment-ppo-training.ipynb' for the first time, and received a Key Error when running the training loop section. It was in this line:
rewards = torch.tensor([output[1]["score"] for output in pipe_outputs]).to(device)
I presume it is safe to omit the '[1]'?
rewards = torch.tensor([output["score"] for output in pipe_outputs]).to(device)
from trl.
lvwerra
commented on September 26, 2024
It should be fixed now!
from trl.
Related Issues (20)
- Different with online dpo papers HOT 1
- finetuning gemma2-2b with multi-gpu get OOM, how do i only do model sharding and no data parallel(i guess it's going into DDP). HOT 2
- I'm closing this issue because no one has provided any code to reproduce the error. If you think you are facing the same problem, please open another issue (link this one) with a precise description and a code to reproduce the error.
- Negative Entropy in TRL PPOv2Trainer TLDR Example HOT 3
- Clarification of 2 Entropies in PPOv2Trainer Documentation
- `OnPolicyConfig` - Rename or revise `num_sample_generations` HOT 1
- AttributeError: 'PPOv2Trainer' object has no attribute 'deepspeed' HOT 3
- Incorrect Doc String for `SFTConfig` HOT 2
- plz make GPOTrainer! (Generalized Preference Optimization)
- AttributeError: 'PPOv2Trainer' object has no attribute 'deepspeed' HOT 3
- ConstantLengthDataset should shuffle the order of samples before packing HOT 2
- The ppo_trainer.generate() call results in an error. HOT 2
- DPOTrainer failed on training Custom Mixture of Experts model with config output_router_logits=True
- Disable the dropout by default in Online DPO HOT 1
- RLOOTrainer & PPOv2Trainer - Modify Name for W&B Logged Table HOT 3
- PPOv2Trainer & RLOOTrainer - Add Safety Check that `policy` object != `ref_policy` object HOT 2
- Always allow `ref_model=None` HOT 5
- Reward model HOT 4
- [DPOTrainer] Tokenizer calculation fail during Q+A concat HOT 3
- Model does not generate eos when SFTTrainer with setup_chat_format is used HOT 1
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.