policy

RLHF-Trainer

tb91117 Β· PyTorch

or hover any field below to flag it

Overview

Name
RLHF-Trainer
Author
tb91117
Framework
PyTorch
License
MIT
Skill type
locomotion
Evidence level
untested
Task description
RLHF Trainer implements the complete RLHF pipeline for consumer GPUs: supervised fine-tuning β†’ reward modeling β†’ PPO training. 🎯 Running on 2080Ti 12GB, it brings ChatGPT-style alignment to your local machine. πŸ’»πŸ”₯

Spaces

Action space
other Β· 0-dim Β· 0Hz
Observation space
  • type: other

Links

HuggingFace repo
null
Paper (arXiv)
null

Compatible robots

20

Compatible environments

0

No environments list RLHF-Trainer yet.

Datasets that reference this policy

0

No datasets reference RLHF-Trainer yet.