dataset
RLHF-Dialogue-Summarizer-with-PPO-and-Reward-Model
ivanluk914
or hover any field below to flag it
Overview
Name
RLHF-Dialogue-Summarizer-with-PPO-and-Reward-Model
Source
ivanluk914
Episodes
0
Robot count
0
Format
other
Description
This project implements RLHF to fine-tune a FLAN-T5 language model for dialogue summarization. The training pipeline involves previous PEFT model, dataset preparation, reward modeling that leveraging a toxicity classifier, implementing PPO-based RL with trl library and evaluation before and after RL
Robots used
null
Links
HuggingFace dataset
null