forangel2014/Probabilistic-Uncertain-Reward-Model

Python

8

4 commits

updated Oct 9, 2025

See the code

README

Probilistic Uncertain Reward Model

preparation

pip install -r requirement.txt
python setting.py --reward_model_path <your_reward_model_path> --policy_model_path <your_policy_model_path>
python scripts/process_preference.py
python scripts/process_rlhf.py

reward model training (4 GPUs)

bash run_purm.sh

RLHF with PURM (8 GPUs)

cd OpenRLHF
CUDA_VISIBLE_DEVICES=7 python reward_server.py
bash run.sh

forangel2014/Probabilistic-Uncertain-Reward-Model

Python

8

4 commits

updated Oct 9, 2025

See the code

README

Probilistic Uncertain Reward Model

preparation

pip install -r requirement.txt
python setting.py --reward_model_path <your_reward_model_path> --policy_model_path <your_policy_model_path>
python scripts/process_preference.py
python scripts/process_rlhf.py

reward model training (4 GPUs)

bash run_purm.sh

RLHF with PURM (8 GPUs)

cd OpenRLHF
CUDA_VISIBLE_DEVICES=7 python reward_server.py
bash run.sh