Skip to content

histmeisah/Search_r1_multinode

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

371 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

We reimplement the Search-r1 base on the newest verl.

We create a new vllm_rollout: /verl/workers/rollout/vllm_rollout/search_vllm_rollout.py . This rollout can generation answer and use search engine which developed by search-r1.

So the edited files are: Train related

  1. create search_vllm_rollout.py
  2. modify /verl/trainer/main_ppo.py
  3. modify verl/trainer/ppo/ray_trainer.py
  4. modify verl/trainer/config/ppo_trainer.yaml Reward related :
  5. modify verl/utils/reward_score/init.py
  6. modify verl/utils/reward_score/qa_em.py
  7. modify verl/workers/reward_manager/naive.py

If you want to try this repo, please run head_ray.sh on the head node and node_ray.sh on the the worker node. Than run the test_search_r1.sh

About

Multinode reimplement for search-r1

Resources

License

Stars

4 stars

Watchers

0 watching

Forks

Releases

No releases published

Packages

 
 
 

Contributors