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
20anybotics-anymal-cnot in seedalohanot in seedgoogle-barkour-vbnot in seedboston-dynamics-spotnot in seedfranka-fr3not in seedgoogle-barkour-v0not in seedagilex-pipernot in seedberkeley-humanoidnot in seedbitcraze-crazyflie-2not in seedanybotics-anymal-bnot in seedagility-cassienot in seedarx-l5not in seedbooster-t1not in seedfranka-emika-pandanot in seedfranka-fr3-v2not in seeddynamixel-2rnot in seedflexiv-rizon4not in seedassetsnot in seedapptronik-apollonot in seedfourier-n1not in seed
Compatible environments
0No environments list RLHF-Trainer yet.
Datasets that reference this policy
0No datasets reference RLHF-Trainer yet.