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