Examples

Table of Contents

Install requirements

Get Start with ColossalRun

Training Configuration

RLHF Training Stage1 - Supervised Instructs Tuning

Step 1: Data Collection

Step 2: Preprocessing

Step 3: Training

RLHF Training Stage2 - Training Reward Model

Step 1: Data Collection

Step 2: Preprocessing

Step 3: Training

Features and Tricks in RM Training

Note on Reward Model Training

RLHF Training Stage3 - Proximal Policy Optimization

Step 1: Data Collection

Step 2: Preprocessing

Step 3: Training

Sample Training Results Using Default Script

Reward

Note on PPO Training

Q1: My reward is negative

Q2: My actor loss is negative

Q3: My reward doesn't go up (decreases)

Q4: Generation is garbage

Alternative Option For RLHF: Direct Preference Optimization

DPO Training Stage1 - Supervised Instructs Tuning

DPO Training Stage2 - DPO Training

Step 1: Data Collection & Preparation

Step 2: Training

DPO Result

Hardware Requirements

List of Supported Models

Inference example

Attention