Add files using upload-large-folder tool
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- .gitattributes +30 -0
- atpo_hotpotqa_qwen4b/20260506_102808/global_step_250/actor/optim_world_size_8_rank_0.pt +3 -0
- atpo_hotpotqa_qwen4b/20260506_102808/global_step_250/actor/optim_world_size_8_rank_4.pt +3 -0
- atpo_hotpotqa_qwen4b/20260506_102808/global_step_250/actor/optim_world_size_8_rank_5.pt +3 -0
- atpo_hotpotqa_qwen4b/20260506_102808/outputs/.hydra/overrides.yaml +70 -0
- atpo_hotpotqa_qwen4b/20260506_102808/rollout/220.jsonl +3 -0
- atpo_hotpotqa_qwen4b/20260506_102808/rollout/222.jsonl +3 -0
- atpo_hotpotqa_qwen4b/20260506_102808/rollout/231.jsonl +3 -0
- atpo_hotpotqa_qwen4b/20260506_102808/rollout/232.jsonl +3 -0
- atpo_hotpotqa_qwen4b/20260506_102808/rollout/235.jsonl +3 -0
- atpo_hotpotqa_qwen4b/20260506_102808/rollout/240.jsonl +3 -0
- atpo_hotpotqa_qwen4b/20260506_102808/validation/200.jsonl +0 -0
- atpo_hotpotqa_qwen4b/20260506_102808/validation/240.jsonl +0 -0
- atpo_hotpotqa_qwen8b/20260506_124901/outputs/.hydra/config.yaml +319 -0
- atpo_hotpotqa_qwen8b/20260506_124901/outputs/.hydra/hydra.yaml +226 -0
- atpo_hotpotqa_qwen8b/20260506_124901/outputs/.hydra/overrides.yaml +70 -0
- atpo_hotpotqa_qwen8b/20260506_124901/outputs/main_ppo.log +0 -0
- atpo_hotpotqa_qwen8b/20260506_124901/validation/200.jsonl +0 -0
- atpo_hotpotqa_qwen8b/20260506_124901/validation/210.jsonl +0 -0
- atpo_hotpotqa_qwen8b/20260506_124901/validation/220.jsonl +0 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/added_tokens.json +28 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/config.json +31 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/generation_config.json +13 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/merges.txt +0 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/model_world_size_8_rank_0.pt +3 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/model_world_size_8_rank_2.pt +3 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/model_world_size_8_rank_4.pt +3 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/model_world_size_8_rank_6.pt +3 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/optim_world_size_8_rank_3.pt +3 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/optim_world_size_8_rank_7.pt +3 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/special_tokens_map.json +31 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/tokenizer_config.json +240 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/vocab.json +0 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/latest_checkpointed_iteration.txt +1 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/1.jsonl +3 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/10.jsonl +3 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/101.jsonl +3 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/118.jsonl +3 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/122.jsonl +3 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/129.jsonl +3 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/13.jsonl +3 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/130.jsonl +3 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/136.jsonl +3 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/143.jsonl +3 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/150.jsonl +3 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/156.jsonl +3 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/157.jsonl +3 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/160.jsonl +3 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/163.jsonl +3 -0
- cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/176.jsonl +3 -0
.gitattributes
CHANGED
|
@@ -994,3 +994,33 @@ cache_nq_qwen4b/20260504_160206/rollout/239.jsonl filter=lfs diff=lfs merge=lfs
|
|
| 994 |
cache_nq_qwen4b/20260504_160206/rollout/225.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 995 |
cache_nq_qwen4b/20260504_160206/rollout/251.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 996 |
cache_nq_qwen4b/20260504_160206/rollout/214.jsonl filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 994 |
cache_nq_qwen4b/20260504_160206/rollout/225.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 995 |
cache_nq_qwen4b/20260504_160206/rollout/251.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 996 |
cache_nq_qwen4b/20260504_160206/rollout/214.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 997 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/209.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 998 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/129.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 999 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/157.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1000 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/208.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1001 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/237.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1002 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/221.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1003 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/143.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1004 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/213.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1005 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/160.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1006 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/195.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1007 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/215.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1008 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/177.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1009 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/178.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1010 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/185.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1011 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/156.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1012 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/176.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1013 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/130.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1014 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/163.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1015 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/150.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1016 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/200.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1017 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/199.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1018 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/194.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1019 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/233.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1020 |
+
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/136.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1021 |
+
atpo_hotpotqa_qwen4b/20260506_102808/rollout/235.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1022 |
+
atpo_hotpotqa_qwen4b/20260506_102808/rollout/232.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1023 |
+
atpo_hotpotqa_qwen4b/20260506_102808/rollout/231.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1024 |
+
atpo_hotpotqa_qwen4b/20260506_102808/rollout/240.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1025 |
+
atpo_hotpotqa_qwen4b/20260506_102808/rollout/222.jsonl filter=lfs diff=lfs merge=lfs -text
|
| 1026 |
+
atpo_hotpotqa_qwen4b/20260506_102808/rollout/220.jsonl filter=lfs diff=lfs merge=lfs -text
|
atpo_hotpotqa_qwen4b/20260506_102808/global_step_250/actor/optim_world_size_8_rank_0.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7573736c838c607348eec7d642227db2a8720a5235f429c8912dc52ba8d37371
|
| 3 |
+
size 4022501290
|
atpo_hotpotqa_qwen4b/20260506_102808/global_step_250/actor/optim_world_size_8_rank_4.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c716c3cdf6ad8f7b3e86e695f015c681cf42f60200c98a8bf12204dd85d0270f
|
| 3 |
+
size 4022501290
|
atpo_hotpotqa_qwen4b/20260506_102808/global_step_250/actor/optim_world_size_8_rank_5.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8b3dde2b35db02b1206aae69aa4225e46aec6a68c07705ca6df36303102d1c87
|
| 3 |
+
size 4022501290
|
atpo_hotpotqa_qwen4b/20260506_102808/outputs/.hydra/overrides.yaml
ADDED
|
@@ -0,0 +1,70 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
- algorithm.adv_estimator=grpo
|
| 2 |
+
- algorithm.kl_ctrl.kl_coef=0.0
|
| 3 |
+
- data.train_files=/scratch/user/sushil22_tamu.edu/projects/ATPO/rl_datasets/hotpotqa/train.parquet
|
| 4 |
+
- data.val_files=["/scratch/user/sushil22_tamu.edu/projects/ATPO/rl_datasets/hotpotqa/test_512.parquet"]
|
| 5 |
+
- data.prompt_key=prompt
|
| 6 |
+
- data.train_batch_size=64
|
| 7 |
+
- data.max_prompt_length=2000
|
| 8 |
+
- data.max_response_length=6192
|
| 9 |
+
- actor_rollout_ref.model.path=/scratch/user/sushil22_tamu.edu/projects/ATPO/models/Qwen3-4B
|
| 10 |
+
- actor_rollout_ref.model.enable_gradient_checkpointing=True
|
| 11 |
+
- actor_rollout_ref.model.use_remove_padding=True
|
| 12 |
+
- actor_rollout_ref.actor.policy_loss=gspo_turn
|
| 13 |
+
- actor_rollout_ref.actor.clip_ratio_low=3e-3
|
| 14 |
+
- actor_rollout_ref.actor.clip_ratio_high=4e-3
|
| 15 |
+
- actor_rollout_ref.actor.enable_entropy_balanced_clipping=False
|
| 16 |
+
- actor_rollout_ref.actor.enable_entropy_balanced_advantage=False
|
| 17 |
+
- actor_rollout_ref.actor.optim.lr=1e-6
|
| 18 |
+
- actor_rollout_ref.actor.ppo_mini_batch_size=8
|
| 19 |
+
- actor_rollout_ref.actor.use_dynamic_bsz=True
|
| 20 |
+
- actor_rollout_ref.actor.ppo_max_token_len_per_gpu=16384
|
| 21 |
+
- actor_rollout_ref.actor.use_kl_loss=True
|
| 22 |
+
- actor_rollout_ref.actor.kl_loss_coef=0.0
|
| 23 |
+
- actor_rollout_ref.actor.kl_loss_type=low_var_kl
|
| 24 |
+
- actor_rollout_ref.actor.fsdp_config.param_offload=False
|
| 25 |
+
- actor_rollout_ref.actor.fsdp_config.optimizer_offload=False
|
| 26 |
+
- actor_rollout_ref.rollout.enable_dynamic_rollouts=False
|
| 27 |
+
- actor_rollout_ref.rollout.log_prob_max_token_len_per_gpu=32768
|
| 28 |
+
- actor_rollout_ref.rollout.tensor_model_parallel_size=1
|
| 29 |
+
- actor_rollout_ref.rollout.name=vllm
|
| 30 |
+
- actor_rollout_ref.rollout.mode=sync_with_tool_tree
|
| 31 |
+
- actor_rollout_ref.rollout.gpu_memory_utilization=0.6
|
| 32 |
+
- actor_rollout_ref.rollout.n=22
|
| 33 |
+
- actor_rollout_ref.rollout.initial_rollouts=10
|
| 34 |
+
- actor_rollout_ref.rollout.expansion_mode=entropy
|
| 35 |
+
- actor_rollout_ref.rollout.expansion_iterations=2
|
| 36 |
+
- actor_rollout_ref.rollout.beam_size=6
|
| 37 |
+
- actor_rollout_ref.rollout.samples_per_tree=22
|
| 38 |
+
- actor_rollout_ref.rollout.branch_probability=0.5
|
| 39 |
+
- actor_rollout_ref.rollout.entropy_weight=0.2
|
| 40 |
+
- actor_rollout_ref.rollout.leaf_value_norm=True
|
| 41 |
+
- actor_rollout_ref.rollout.node_value_mode=child_softmax
|
| 42 |
+
- actor_rollout_ref.rollout.node_adv_mode=node_value
|
| 43 |
+
- actor_rollout_ref.rollout.tools.call_limit=6
|
| 44 |
+
- ++actor_rollout_ref.rollout.tools.tool_instances.search.params.cache_file=/scratch/user/sushil22_tamu.edu/projects/ATPO/search_cache/search_cache.json
|
| 45 |
+
- ++actor_rollout_ref.rollout.tools.tool_instances.search.params.api_key=unused_local_rag_server
|
| 46 |
+
- actor_rollout_ref.rollout.multi_turn.enable=True
|
| 47 |
+
- actor_rollout_ref.rollout.multi_turn.tool_config_path=/scratch/user/sushil22_tamu.edu/projects/ATPO/ATPO/verl_atpo/examples/sglang_multiturn/config/tool_config/search_tool_config.yaml
|
| 48 |
+
- actor_rollout_ref.ref.log_prob_max_token_len_per_gpu=32768
|
| 49 |
+
- actor_rollout_ref.ref.fsdp_config.param_offload=True
|
| 50 |
+
- reward_model.reward_manager=naive
|
| 51 |
+
- custom_reward_function.path=/scratch/user/sushil22_tamu.edu/projects/ATPO/ATPO/verl_atpo/verl/utils/reward_score/deep_research_em.py
|
| 52 |
+
- custom_reward_function.name=compute_score
|
| 53 |
+
- trainer.critic_warmup=0
|
| 54 |
+
- trainer.logger=[console, wandb]
|
| 55 |
+
- trainer.project_name=Neurips-final-runs
|
| 56 |
+
- trainer.experiment_name=atpo_hotpotqa_qwen4b_ablation_noadvantage
|
| 57 |
+
- trainer.n_gpus_per_node=8
|
| 58 |
+
- trainer.nnodes=1
|
| 59 |
+
- trainer.save_freq=50
|
| 60 |
+
- trainer.test_freq=10
|
| 61 |
+
- trainer.max_actor_ckpt_to_keep=2
|
| 62 |
+
- +actor_rollout_ref.actor.checkpoint.save_contents=[hf_model,extra]
|
| 63 |
+
- trainer.total_epochs=1
|
| 64 |
+
- trainer.default_local_dir=/scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo_hotpotqa_qwen4b_ablation_noadvantage/20260506_102808/
|
| 65 |
+
- trainer.val_before_train=True
|
| 66 |
+
- trainer.rollout_data_dir=/scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo_hotpotqa_qwen4b_ablation_noadvantage/20260506_102808//rollout
|
| 67 |
+
- trainer.validation_data_dir=/scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo_hotpotqa_qwen4b_ablation_noadvantage/20260506_102808//validation
|
| 68 |
+
- ray_init.num_cpus=null
|
| 69 |
+
- trainer.resume_mode=resume_path
|
| 70 |
+
- trainer.resume_from_path=/scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo_hotpotqa_qwen4b/20260430_082825/global_step_200
|
atpo_hotpotqa_qwen4b/20260506_102808/rollout/220.jsonl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9ea3cd3f6d23f18ea1858066864051a264effe053e0519610d64ef0f15b77fc9
|
| 3 |
+
size 14714497
|
atpo_hotpotqa_qwen4b/20260506_102808/rollout/222.jsonl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:88ee49045b8af63c5b4999b806019a1e48233e7092558a56b1cf72644851b926
|
| 3 |
+
size 13183027
|
atpo_hotpotqa_qwen4b/20260506_102808/rollout/231.jsonl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:02495cfd1f50cb81e1ada4a561143f7c26036c63d93c501fe60f20b1a60ada57
|
| 3 |
+
size 14235762
|
atpo_hotpotqa_qwen4b/20260506_102808/rollout/232.jsonl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bd5453481bde797d6c360a3c58571d0b9ab1f2eb355ea50cb310399404345ae2
|
| 3 |
+
size 13841430
|
atpo_hotpotqa_qwen4b/20260506_102808/rollout/235.jsonl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a88baa8399c639b0856cd3cb99eb11548d082753cd6bf5231e737f5286c13e1d
|
| 3 |
+
size 15103381
|
atpo_hotpotqa_qwen4b/20260506_102808/rollout/240.jsonl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:02859f96249c3db635d7e112a85c9b5f252526e547df1b8370c2f00936e7c847
|
| 3 |
+
size 13592804
|
atpo_hotpotqa_qwen4b/20260506_102808/validation/200.jsonl
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
atpo_hotpotqa_qwen4b/20260506_102808/validation/240.jsonl
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
atpo_hotpotqa_qwen8b/20260506_124901/outputs/.hydra/config.yaml
ADDED
|
@@ -0,0 +1,319 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
data:
|
| 2 |
+
tokenizer: null
|
| 3 |
+
use_shm: false
|
| 4 |
+
train_files: /scratch/user/sushil22_tamu.edu/projects/ATPO/rl_datasets/hotpotqa/train.parquet
|
| 5 |
+
val_files:
|
| 6 |
+
- /scratch/user/sushil22_tamu.edu/projects/ATPO/rl_datasets/hotpotqa/test_512.parquet
|
| 7 |
+
prompt_key: prompt
|
| 8 |
+
reward_fn_key: data_source
|
| 9 |
+
max_prompt_length: 2000
|
| 10 |
+
max_response_length: 6192
|
| 11 |
+
train_batch_size: 64
|
| 12 |
+
val_batch_size: null
|
| 13 |
+
return_raw_input_ids: false
|
| 14 |
+
return_raw_chat: false
|
| 15 |
+
return_full_prompt: false
|
| 16 |
+
shuffle: true
|
| 17 |
+
filter_overlong_prompts: false
|
| 18 |
+
filter_overlong_prompts_workers: 1
|
| 19 |
+
truncation: error
|
| 20 |
+
image_key: images
|
| 21 |
+
video_key: videos
|
| 22 |
+
trust_remote_code: false
|
| 23 |
+
custom_cls:
|
| 24 |
+
path: null
|
| 25 |
+
name: null
|
| 26 |
+
algorithm:
|
| 27 |
+
gamma: 1.0
|
| 28 |
+
lam: 1.0
|
| 29 |
+
adv_estimator: grpo
|
| 30 |
+
norm_adv_by_std_in_grpo: true
|
| 31 |
+
use_kl_in_reward: false
|
| 32 |
+
kl_penalty: kl
|
| 33 |
+
kl_ctrl:
|
| 34 |
+
type: fixed
|
| 35 |
+
kl_coef: 0.0
|
| 36 |
+
horizon: 10000
|
| 37 |
+
target_kl: 0.1
|
| 38 |
+
use_pf_ppo: false
|
| 39 |
+
pf_ppo:
|
| 40 |
+
reweight_method: pow
|
| 41 |
+
weight_pow: 2.0
|
| 42 |
+
actor_rollout_ref:
|
| 43 |
+
hybrid_engine: true
|
| 44 |
+
model:
|
| 45 |
+
path: /scratch/user/sushil22_tamu.edu/projects/ATPO/models/Qwen3-8B
|
| 46 |
+
use_shm: false
|
| 47 |
+
external_lib: null
|
| 48 |
+
override_config: {}
|
| 49 |
+
enable_gradient_checkpointing: true
|
| 50 |
+
enable_activation_offload: false
|
| 51 |
+
use_remove_padding: true
|
| 52 |
+
lora_rank: 0
|
| 53 |
+
lora_alpha: 16
|
| 54 |
+
target_modules: all-linear
|
| 55 |
+
use_liger: false
|
| 56 |
+
use_fused_kernels: false
|
| 57 |
+
trust_remote_code: false
|
| 58 |
+
actor:
|
| 59 |
+
policy_loss: gspo_turn
|
| 60 |
+
enable_entropy_balanced_clipping: false
|
| 61 |
+
enable_entropy_balanced_advantage: false
|
| 62 |
+
strategy: fsdp
|
| 63 |
+
ppo_mini_batch_size: 8
|
| 64 |
+
ppo_micro_batch_size: null
|
| 65 |
+
ppo_micro_batch_size_per_gpu: null
|
| 66 |
+
use_dynamic_bsz: true
|
| 67 |
+
ppo_max_token_len_per_gpu: 16384
|
| 68 |
+
grad_clip: 1.0
|
| 69 |
+
clip_ratio: 0.2
|
| 70 |
+
clip_ratio_low: 0.003
|
| 71 |
+
clip_ratio_high: 0.004
|
| 72 |
+
clip_ratio_c: 3.0
|
| 73 |
+
loss_agg_mode: token-mean
|
| 74 |
+
entropy_coeff: 0
|
| 75 |
+
use_kl_loss: true
|
| 76 |
+
use_torch_compile: true
|
| 77 |
+
kl_loss_coef: 0.0
|
| 78 |
+
kl_loss_type: low_var_kl
|
| 79 |
+
ppo_epochs: 1
|
| 80 |
+
shuffle: false
|
| 81 |
+
ulysses_sequence_parallel_size: 1
|
| 82 |
+
checkpoint:
|
| 83 |
+
contents:
|
| 84 |
+
- model
|
| 85 |
+
- optimizer
|
| 86 |
+
- extra
|
| 87 |
+
save_contents:
|
| 88 |
+
- hf_model
|
| 89 |
+
- extra
|
| 90 |
+
optim:
|
| 91 |
+
lr: 1.0e-06
|
| 92 |
+
lr_warmup_steps: -1
|
| 93 |
+
lr_warmup_steps_ratio: 0.0
|
| 94 |
+
min_lr_ratio: 0.0
|
| 95 |
+
num_cycles: 0.5
|
| 96 |
+
warmup_style: constant
|
| 97 |
+
total_training_steps: -1
|
| 98 |
+
weight_decay: 0.01
|
| 99 |
+
fsdp_config:
|
| 100 |
+
wrap_policy:
|
| 101 |
+
min_num_params: 0
|
| 102 |
+
param_offload: false
|
| 103 |
+
optimizer_offload: false
|
| 104 |
+
offload_policy: false
|
| 105 |
+
reshard_after_forward: true
|
| 106 |
+
fsdp_size: -1
|
| 107 |
+
ref:
|
| 108 |
+
strategy: fsdp
|
| 109 |
+
fsdp_config:
|
| 110 |
+
param_offload: true
|
| 111 |
+
reshard_after_forward: true
|
| 112 |
+
wrap_policy:
|
| 113 |
+
min_num_params: 0
|
| 114 |
+
use_torch_compile: ${actor_rollout_ref.actor.use_torch_compile}
|
| 115 |
+
log_prob_micro_batch_size: null
|
| 116 |
+
log_prob_micro_batch_size_per_gpu: null
|
| 117 |
+
log_prob_use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 118 |
+
log_prob_max_token_len_per_gpu: 32768
|
| 119 |
+
ulysses_sequence_parallel_size: ${actor_rollout_ref.actor.ulysses_sequence_parallel_size}
|
| 120 |
+
rollout:
|
| 121 |
+
name: vllm
|
| 122 |
+
mode: sync_with_tool_tree
|
| 123 |
+
chat_scheduler: null
|
| 124 |
+
temperature: 1.0
|
| 125 |
+
top_k: -1
|
| 126 |
+
top_p: 1
|
| 127 |
+
use_fire_sampling: false
|
| 128 |
+
prompt_length: ${data.max_prompt_length}
|
| 129 |
+
response_length: ${data.max_response_length}
|
| 130 |
+
initial_rollouts: 10
|
| 131 |
+
expansion_mode: entropy
|
| 132 |
+
expansion_iterations: 2
|
| 133 |
+
beam_size: 6
|
| 134 |
+
samples_per_tree: 22
|
| 135 |
+
data: ${data}
|
| 136 |
+
use_kl_in_reward: ${algorithm.use_kl_in_reward}
|
| 137 |
+
kl_penalty: ${algorithm.kl_penalty}
|
| 138 |
+
adv_estimator: ${algorithm.adv_estimator}
|
| 139 |
+
gamma: ${algorithm.gamma}
|
| 140 |
+
lam: ${algorithm.lam}
|
| 141 |
+
norm_adv_by_std_in_grpo: ${algorithm.norm_adv_by_std_in_grpo}
|
| 142 |
+
kl_ctrl: ${algorithm.kl_ctrl}
|
| 143 |
+
reward_model: ${reward_model}
|
| 144 |
+
custom_reward_function: ${custom_reward_function}
|
| 145 |
+
leaf_value_norm: true
|
| 146 |
+
node_value_mode: child_softmax
|
| 147 |
+
node_adv_mode: node_value
|
| 148 |
+
entropy_mixing_method: no_entropy
|
| 149 |
+
dtype: bfloat16
|
| 150 |
+
gpu_memory_utilization: 0.6
|
| 151 |
+
ignore_eos: false
|
| 152 |
+
enforce_eager: true
|
| 153 |
+
free_cache_engine: true
|
| 154 |
+
load_format: dummy_dtensor
|
| 155 |
+
layered_summon: false
|
| 156 |
+
tensor_model_parallel_size: 1
|
| 157 |
+
max_num_batched_tokens: 25000
|
| 158 |
+
max_model_len: null
|
| 159 |
+
max_num_seqs: 1024
|
| 160 |
+
log_prob_micro_batch_size: null
|
| 161 |
+
log_prob_micro_batch_size_per_gpu: null
|
| 162 |
+
log_prob_use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 163 |
+
log_prob_max_token_len_per_gpu: 32768
|
| 164 |
+
disable_log_stats: true
|
| 165 |
+
enable_chunked_prefill: true
|
| 166 |
+
do_sample: true
|
| 167 |
+
'n': 22
|
| 168 |
+
branch_probability: 0.5
|
| 169 |
+
entropy_weight: 0.2
|
| 170 |
+
engine_kwargs:
|
| 171 |
+
vllm:
|
| 172 |
+
swap_space: null
|
| 173 |
+
sglang:
|
| 174 |
+
attention_backend: null
|
| 175 |
+
val_kwargs:
|
| 176 |
+
top_k: -1
|
| 177 |
+
top_p: 1.0
|
| 178 |
+
temperature: 0
|
| 179 |
+
'n': 1
|
| 180 |
+
do_sample: false
|
| 181 |
+
multi_turn:
|
| 182 |
+
enable: true
|
| 183 |
+
max_turns: null
|
| 184 |
+
tool_config_path: /scratch/user/sushil22_tamu.edu/projects/ATPO/ATPO/verl_atpo/examples/sglang_multiturn/config/tool_config/search_tool_config.yaml
|
| 185 |
+
format: chatml
|
| 186 |
+
enable_dynamic_rollouts: false
|
| 187 |
+
tools:
|
| 188 |
+
call_limit: 6
|
| 189 |
+
max_workers: 32
|
| 190 |
+
timeout: 120
|
| 191 |
+
retry_count: 4
|
| 192 |
+
verbose_logging: true
|
| 193 |
+
fail_on_error: true
|
| 194 |
+
tool_instances:
|
| 195 |
+
python:
|
| 196 |
+
class_path: verl.workers.rollout.tools.python_tool.PythonTool
|
| 197 |
+
params:
|
| 198 |
+
conda_path: <your_conda_path>
|
| 199 |
+
conda_env: verl
|
| 200 |
+
search:
|
| 201 |
+
class_path: verl.workers.rollout.tools.search_tool.BingSearchTool
|
| 202 |
+
params:
|
| 203 |
+
api_key: unused_local_rag_server
|
| 204 |
+
zone: serp_api_test
|
| 205 |
+
max_results: 10
|
| 206 |
+
result_length: 1000
|
| 207 |
+
location: cn
|
| 208 |
+
cache_file: /scratch/user/sushil22_tamu.edu/projects/ATPO/search_cache/search_cache.json
|
| 209 |
+
async_cache_write: true
|
| 210 |
+
critic:
|
| 211 |
+
rollout_n: ${actor_rollout_ref.rollout.n}
|
| 212 |
+
strategy: fsdp
|
| 213 |
+
optim:
|
| 214 |
+
lr: 1.0e-05
|
| 215 |
+
lr_warmup_steps_ratio: 0.0
|
| 216 |
+
min_lr_ratio: null
|
| 217 |
+
warmup_style: constant
|
| 218 |
+
total_training_steps: -1
|
| 219 |
+
weight_decay: 0.01
|
| 220 |
+
model:
|
| 221 |
+
path: ~/models/deepseek-llm-7b-chat
|
| 222 |
+
use_shm: false
|
| 223 |
+
tokenizer_path: ${actor_rollout_ref.model.path}
|
| 224 |
+
override_config: {}
|
| 225 |
+
external_lib: ${actor_rollout_ref.model.external_lib}
|
| 226 |
+
enable_gradient_checkpointing: true
|
| 227 |
+
enable_activation_offload: false
|
| 228 |
+
use_remove_padding: false
|
| 229 |
+
trust_remote_code: ${actor_rollout_ref.model.trust_remote_code}
|
| 230 |
+
fsdp_config:
|
| 231 |
+
param_offload: false
|
| 232 |
+
optimizer_offload: false
|
| 233 |
+
offload_policy: false
|
| 234 |
+
reshard_after_forward: true
|
| 235 |
+
wrap_policy:
|
| 236 |
+
min_num_params: 0
|
| 237 |
+
fsdp_size: -1
|
| 238 |
+
lora_rank: 0
|
| 239 |
+
lora_alpha: 16
|
| 240 |
+
target_modules: all-linear
|
| 241 |
+
ppo_mini_batch_size: ${actor_rollout_ref.actor.ppo_mini_batch_size}
|
| 242 |
+
ppo_micro_batch_size: null
|
| 243 |
+
ppo_micro_batch_size_per_gpu: null
|
| 244 |
+
forward_micro_batch_size: ${critic.ppo_micro_batch_size}
|
| 245 |
+
forward_micro_batch_size_per_gpu: ${critic.ppo_micro_batch_size_per_gpu}
|
| 246 |
+
use_dynamic_bsz: ${actor_rollout_ref.actor.use_dynamic_bsz}
|
| 247 |
+
ppo_max_token_len_per_gpu: 32768
|
| 248 |
+
forward_max_token_len_per_gpu: ${critic.ppo_max_token_len_per_gpu}
|
| 249 |
+
ulysses_sequence_parallel_size: 1
|
| 250 |
+
ppo_epochs: ${actor_rollout_ref.actor.ppo_epochs}
|
| 251 |
+
shuffle: ${actor_rollout_ref.actor.shuffle}
|
| 252 |
+
grad_clip: 1.0
|
| 253 |
+
cliprange_value: 0.5
|
| 254 |
+
loss_agg_mode: ${actor_rollout_ref.actor.loss_agg_mode}
|
| 255 |
+
checkpoint:
|
| 256 |
+
contents:
|
| 257 |
+
- model
|
| 258 |
+
- optimizer
|
| 259 |
+
- extra
|
| 260 |
+
reward_model:
|
| 261 |
+
enable: false
|
| 262 |
+
strategy: fsdp
|
| 263 |
+
model:
|
| 264 |
+
input_tokenizer: ${actor_rollout_ref.model.path}
|
| 265 |
+
path: ~/models/FsfairX-LLaMA3-RM-v0.1
|
| 266 |
+
use_shm: false
|
| 267 |
+
external_lib: ${actor_rollout_ref.model.external_lib}
|
| 268 |
+
use_remove_padding: false
|
| 269 |
+
use_fused_kernels: ${actor_rollout_ref.model.use_fused_kernels}
|
| 270 |
+
trust_remote_code: false
|
| 271 |
+
fsdp_config:
|
| 272 |
+
wrap_policy:
|
| 273 |
+
min_num_params: 0
|
| 274 |
+
param_offload: false
|
| 275 |
+
reshard_after_forward: true
|
| 276 |
+
fsdp_size: -1
|
| 277 |
+
micro_batch_size: null
|
| 278 |
+
micro_batch_size_per_gpu: null
|
| 279 |
+
max_length: null
|
| 280 |
+
ulysses_sequence_parallel_size: 1
|
| 281 |
+
use_dynamic_bsz: ${critic.use_dynamic_bsz}
|
| 282 |
+
forward_max_token_len_per_gpu: ${critic.forward_max_token_len_per_gpu}
|
| 283 |
+
reward_manager: naive
|
| 284 |
+
launch_reward_fn_async: false
|
| 285 |
+
sandbox_fusion:
|
| 286 |
+
url: null
|
| 287 |
+
max_concurrent: 64
|
| 288 |
+
custom_reward_function:
|
| 289 |
+
path: /scratch/user/sushil22_tamu.edu/projects/ATPO/ATPO/verl_atpo/verl/utils/reward_score/deep_research_em.py
|
| 290 |
+
name: compute_score
|
| 291 |
+
trainer:
|
| 292 |
+
balance_batch: true
|
| 293 |
+
total_epochs: 1
|
| 294 |
+
total_training_steps: null
|
| 295 |
+
project_name: Neurips-final-runs
|
| 296 |
+
experiment_name: atpo_hotpotqa_qwen8b
|
| 297 |
+
logger:
|
| 298 |
+
- console
|
| 299 |
+
- wandb
|
| 300 |
+
log_val_generations: 0
|
| 301 |
+
rollout_data_dir: /scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo_hotpotqa_qwen8b/20260506_124901//rollout
|
| 302 |
+
validation_data_dir: /scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo_hotpotqa_qwen8b/20260506_124901//validation
|
| 303 |
+
nnodes: 1
|
| 304 |
+
n_gpus_per_node: 8
|
| 305 |
+
save_freq: 50
|
| 306 |
+
resume_mode: resume_path
|
| 307 |
+
resume_from_path: /scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo-qwen3-8b-hotpotqa-step200/global_step_200
|
| 308 |
+
val_before_train: true
|
| 309 |
+
test_freq: 10
|
| 310 |
+
critic_warmup: 0
|
| 311 |
+
default_hdfs_dir: null
|
| 312 |
+
del_local_ckpt_after_load: false
|
| 313 |
+
default_local_dir: /scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo_hotpotqa_qwen8b/20260506_124901/
|
| 314 |
+
max_actor_ckpt_to_keep: 2
|
| 315 |
+
max_critic_ckpt_to_keep: null
|
| 316 |
+
ray_wait_register_center_timeout: 300
|
| 317 |
+
device: cuda
|
| 318 |
+
ray_init:
|
| 319 |
+
num_cpus: null
|
atpo_hotpotqa_qwen8b/20260506_124901/outputs/.hydra/hydra.yaml
ADDED
|
@@ -0,0 +1,226 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
hydra:
|
| 2 |
+
run:
|
| 3 |
+
dir: /scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo_hotpotqa_qwen8b/20260506_124901//outputs
|
| 4 |
+
sweep:
|
| 5 |
+
dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
|
| 6 |
+
subdir: ${hydra.job.num}
|
| 7 |
+
launcher:
|
| 8 |
+
_target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
|
| 9 |
+
sweeper:
|
| 10 |
+
_target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
|
| 11 |
+
max_batch_size: null
|
| 12 |
+
params: null
|
| 13 |
+
help:
|
| 14 |
+
app_name: ${hydra.job.name}
|
| 15 |
+
header: '${hydra.help.app_name} is powered by Hydra.
|
| 16 |
+
|
| 17 |
+
'
|
| 18 |
+
footer: 'Powered by Hydra (https://hydra.cc)
|
| 19 |
+
|
| 20 |
+
Use --hydra-help to view Hydra specific help
|
| 21 |
+
|
| 22 |
+
'
|
| 23 |
+
template: '${hydra.help.header}
|
| 24 |
+
|
| 25 |
+
== Configuration groups ==
|
| 26 |
+
|
| 27 |
+
Compose your configuration from those groups (group=option)
|
| 28 |
+
|
| 29 |
+
|
| 30 |
+
$APP_CONFIG_GROUPS
|
| 31 |
+
|
| 32 |
+
|
| 33 |
+
== Config ==
|
| 34 |
+
|
| 35 |
+
Override anything in the config (foo.bar=value)
|
| 36 |
+
|
| 37 |
+
|
| 38 |
+
$CONFIG
|
| 39 |
+
|
| 40 |
+
|
| 41 |
+
${hydra.help.footer}
|
| 42 |
+
|
| 43 |
+
'
|
| 44 |
+
hydra_help:
|
| 45 |
+
template: 'Hydra (${hydra.runtime.version})
|
| 46 |
+
|
| 47 |
+
See https://hydra.cc for more info.
|
| 48 |
+
|
| 49 |
+
|
| 50 |
+
== Flags ==
|
| 51 |
+
|
| 52 |
+
$FLAGS_HELP
|
| 53 |
+
|
| 54 |
+
|
| 55 |
+
== Configuration groups ==
|
| 56 |
+
|
| 57 |
+
Compose your configuration from those groups (For example, append hydra/job_logging=disabled
|
| 58 |
+
to command line)
|
| 59 |
+
|
| 60 |
+
|
| 61 |
+
$HYDRA_CONFIG_GROUPS
|
| 62 |
+
|
| 63 |
+
|
| 64 |
+
Use ''--cfg hydra'' to Show the Hydra config.
|
| 65 |
+
|
| 66 |
+
'
|
| 67 |
+
hydra_help: ???
|
| 68 |
+
hydra_logging:
|
| 69 |
+
version: 1
|
| 70 |
+
formatters:
|
| 71 |
+
simple:
|
| 72 |
+
format: '[%(asctime)s][HYDRA] %(message)s'
|
| 73 |
+
handlers:
|
| 74 |
+
console:
|
| 75 |
+
class: logging.StreamHandler
|
| 76 |
+
formatter: simple
|
| 77 |
+
stream: ext://sys.stdout
|
| 78 |
+
root:
|
| 79 |
+
level: INFO
|
| 80 |
+
handlers:
|
| 81 |
+
- console
|
| 82 |
+
loggers:
|
| 83 |
+
logging_example:
|
| 84 |
+
level: DEBUG
|
| 85 |
+
disable_existing_loggers: false
|
| 86 |
+
job_logging:
|
| 87 |
+
version: 1
|
| 88 |
+
formatters:
|
| 89 |
+
simple:
|
| 90 |
+
format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
|
| 91 |
+
handlers:
|
| 92 |
+
console:
|
| 93 |
+
class: logging.StreamHandler
|
| 94 |
+
formatter: simple
|
| 95 |
+
stream: ext://sys.stdout
|
| 96 |
+
file:
|
| 97 |
+
class: logging.FileHandler
|
| 98 |
+
formatter: simple
|
| 99 |
+
filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
|
| 100 |
+
root:
|
| 101 |
+
level: INFO
|
| 102 |
+
handlers:
|
| 103 |
+
- console
|
| 104 |
+
- file
|
| 105 |
+
disable_existing_loggers: false
|
| 106 |
+
env: {}
|
| 107 |
+
mode: RUN
|
| 108 |
+
searchpath: []
|
| 109 |
+
callbacks: {}
|
| 110 |
+
output_subdir: .hydra
|
| 111 |
+
overrides:
|
| 112 |
+
hydra:
|
| 113 |
+
- hydra.run.dir=/scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo_hotpotqa_qwen8b/20260506_124901//outputs
|
| 114 |
+
- hydra.mode=RUN
|
| 115 |
+
task:
|
| 116 |
+
- algorithm.adv_estimator=grpo
|
| 117 |
+
- algorithm.kl_ctrl.kl_coef=0.0
|
| 118 |
+
- data.train_files=/scratch/user/sushil22_tamu.edu/projects/ATPO/rl_datasets/hotpotqa/train.parquet
|
| 119 |
+
- data.val_files=["/scratch/user/sushil22_tamu.edu/projects/ATPO/rl_datasets/hotpotqa/test_512.parquet"]
|
| 120 |
+
- data.prompt_key=prompt
|
| 121 |
+
- data.train_batch_size=64
|
| 122 |
+
- data.max_prompt_length=2000
|
| 123 |
+
- data.max_response_length=6192
|
| 124 |
+
- actor_rollout_ref.model.path=/scratch/user/sushil22_tamu.edu/projects/ATPO/models/Qwen3-8B
|
| 125 |
+
- actor_rollout_ref.model.enable_gradient_checkpointing=True
|
| 126 |
+
- actor_rollout_ref.model.use_remove_padding=True
|
| 127 |
+
- actor_rollout_ref.actor.policy_loss=gspo_turn
|
| 128 |
+
- actor_rollout_ref.actor.clip_ratio_low=3e-3
|
| 129 |
+
- actor_rollout_ref.actor.clip_ratio_high=4e-3
|
| 130 |
+
- actor_rollout_ref.actor.enable_entropy_balanced_clipping=False
|
| 131 |
+
- actor_rollout_ref.actor.enable_entropy_balanced_advantage=False
|
| 132 |
+
- actor_rollout_ref.actor.optim.lr=1e-6
|
| 133 |
+
- actor_rollout_ref.actor.ppo_mini_batch_size=8
|
| 134 |
+
- actor_rollout_ref.actor.use_dynamic_bsz=True
|
| 135 |
+
- actor_rollout_ref.actor.ppo_max_token_len_per_gpu=16384
|
| 136 |
+
- actor_rollout_ref.actor.use_kl_loss=True
|
| 137 |
+
- actor_rollout_ref.actor.kl_loss_coef=0.0
|
| 138 |
+
- actor_rollout_ref.actor.kl_loss_type=low_var_kl
|
| 139 |
+
- actor_rollout_ref.actor.fsdp_config.param_offload=False
|
| 140 |
+
- actor_rollout_ref.actor.fsdp_config.optimizer_offload=False
|
| 141 |
+
- actor_rollout_ref.rollout.enable_dynamic_rollouts=False
|
| 142 |
+
- actor_rollout_ref.rollout.log_prob_max_token_len_per_gpu=32768
|
| 143 |
+
- actor_rollout_ref.rollout.tensor_model_parallel_size=1
|
| 144 |
+
- actor_rollout_ref.rollout.name=vllm
|
| 145 |
+
- actor_rollout_ref.rollout.mode=sync_with_tool_tree
|
| 146 |
+
- actor_rollout_ref.rollout.gpu_memory_utilization=0.6
|
| 147 |
+
- actor_rollout_ref.rollout.n=22
|
| 148 |
+
- actor_rollout_ref.rollout.initial_rollouts=10
|
| 149 |
+
- actor_rollout_ref.rollout.expansion_mode=entropy
|
| 150 |
+
- actor_rollout_ref.rollout.expansion_iterations=2
|
| 151 |
+
- actor_rollout_ref.rollout.beam_size=6
|
| 152 |
+
- actor_rollout_ref.rollout.samples_per_tree=22
|
| 153 |
+
- actor_rollout_ref.rollout.branch_probability=0.5
|
| 154 |
+
- actor_rollout_ref.rollout.entropy_weight=0.2
|
| 155 |
+
- actor_rollout_ref.rollout.leaf_value_norm=True
|
| 156 |
+
- actor_rollout_ref.rollout.node_value_mode=child_softmax
|
| 157 |
+
- actor_rollout_ref.rollout.node_adv_mode=node_value
|
| 158 |
+
- actor_rollout_ref.rollout.tools.call_limit=6
|
| 159 |
+
- ++actor_rollout_ref.rollout.tools.tool_instances.search.params.cache_file=/scratch/user/sushil22_tamu.edu/projects/ATPO/search_cache/search_cache.json
|
| 160 |
+
- ++actor_rollout_ref.rollout.tools.tool_instances.search.params.api_key=unused_local_rag_server
|
| 161 |
+
- actor_rollout_ref.rollout.multi_turn.enable=True
|
| 162 |
+
- actor_rollout_ref.rollout.multi_turn.tool_config_path=/scratch/user/sushil22_tamu.edu/projects/ATPO/ATPO/verl_atpo/examples/sglang_multiturn/config/tool_config/search_tool_config.yaml
|
| 163 |
+
- actor_rollout_ref.ref.log_prob_max_token_len_per_gpu=32768
|
| 164 |
+
- actor_rollout_ref.ref.fsdp_config.param_offload=True
|
| 165 |
+
- reward_model.reward_manager=naive
|
| 166 |
+
- custom_reward_function.path=/scratch/user/sushil22_tamu.edu/projects/ATPO/ATPO/verl_atpo/verl/utils/reward_score/deep_research_em.py
|
| 167 |
+
- custom_reward_function.name=compute_score
|
| 168 |
+
- trainer.critic_warmup=0
|
| 169 |
+
- trainer.logger=[console, wandb]
|
| 170 |
+
- trainer.project_name=Neurips-final-runs
|
| 171 |
+
- trainer.experiment_name=atpo_hotpotqa_qwen8b
|
| 172 |
+
- trainer.n_gpus_per_node=8
|
| 173 |
+
- trainer.nnodes=1
|
| 174 |
+
- trainer.save_freq=50
|
| 175 |
+
- trainer.test_freq=10
|
| 176 |
+
- trainer.max_actor_ckpt_to_keep=2
|
| 177 |
+
- +actor_rollout_ref.actor.checkpoint.save_contents=[hf_model,extra]
|
| 178 |
+
- trainer.total_epochs=1
|
| 179 |
+
- trainer.default_local_dir=/scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo_hotpotqa_qwen8b/20260506_124901/
|
| 180 |
+
- trainer.val_before_train=True
|
| 181 |
+
- trainer.rollout_data_dir=/scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo_hotpotqa_qwen8b/20260506_124901//rollout
|
| 182 |
+
- trainer.validation_data_dir=/scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo_hotpotqa_qwen8b/20260506_124901//validation
|
| 183 |
+
- ray_init.num_cpus=null
|
| 184 |
+
- trainer.resume_mode=resume_path
|
| 185 |
+
- trainer.resume_from_path=/scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo-qwen3-8b-hotpotqa-step200/global_step_200
|
| 186 |
+
job:
|
| 187 |
+
name: main_ppo
|
| 188 |
+
chdir: null
|
| 189 |
+
override_dirname: ++actor_rollout_ref.rollout.tools.tool_instances.search.params.api_key=unused_local_rag_server,++actor_rollout_ref.rollout.tools.tool_instances.search.params.cache_file=/scratch/user/sushil22_tamu.edu/projects/ATPO/search_cache/search_cache.json,+actor_rollout_ref.actor.checkpoint.save_contents=[hf_model,extra],actor_rollout_ref.actor.clip_ratio_high=4e-3,actor_rollout_ref.actor.clip_ratio_low=3e-3,actor_rollout_ref.actor.enable_entropy_balanced_advantage=False,actor_rollout_ref.actor.enable_entropy_balanced_clipping=False,actor_rollout_ref.actor.fsdp_config.optimizer_offload=False,actor_rollout_ref.actor.fsdp_config.param_offload=False,actor_rollout_ref.actor.kl_loss_coef=0.0,actor_rollout_ref.actor.kl_loss_type=low_var_kl,actor_rollout_ref.actor.optim.lr=1e-6,actor_rollout_ref.actor.policy_loss=gspo_turn,actor_rollout_ref.actor.ppo_max_token_len_per_gpu=16384,actor_rollout_ref.actor.ppo_mini_batch_size=8,actor_rollout_ref.actor.use_dynamic_bsz=True,actor_rollout_ref.actor.use_kl_loss=True,actor_rollout_ref.model.enable_gradient_checkpointing=True,actor_rollout_ref.model.path=/scratch/user/sushil22_tamu.edu/projects/ATPO/models/Qwen3-8B,actor_rollout_ref.model.use_remove_padding=True,actor_rollout_ref.ref.fsdp_config.param_offload=True,actor_rollout_ref.ref.log_prob_max_token_len_per_gpu=32768,actor_rollout_ref.rollout.beam_size=6,actor_rollout_ref.rollout.branch_probability=0.5,actor_rollout_ref.rollout.enable_dynamic_rollouts=False,actor_rollout_ref.rollout.entropy_weight=0.2,actor_rollout_ref.rollout.expansion_iterations=2,actor_rollout_ref.rollout.expansion_mode=entropy,actor_rollout_ref.rollout.gpu_memory_utilization=0.6,actor_rollout_ref.rollout.initial_rollouts=10,actor_rollout_ref.rollout.leaf_value_norm=True,actor_rollout_ref.rollout.log_prob_max_token_len_per_gpu=32768,actor_rollout_ref.rollout.mode=sync_with_tool_tree,actor_rollout_ref.rollout.multi_turn.enable=True,actor_rollout_ref.rollout.multi_turn.tool_config_path=/scratch/user/sushil22_tamu.edu/projects/ATPO/ATPO/verl_atpo/examples/sglang_multiturn/config/tool_config/search_tool_config.yaml,actor_rollout_ref.rollout.n=22,actor_rollout_ref.rollout.name=vllm,actor_rollout_ref.rollout.node_adv_mode=node_value,actor_rollout_ref.rollout.node_value_mode=child_softmax,actor_rollout_ref.rollout.samples_per_tree=22,actor_rollout_ref.rollout.tensor_model_parallel_size=1,actor_rollout_ref.rollout.tools.call_limit=6,algorithm.adv_estimator=grpo,algorithm.kl_ctrl.kl_coef=0.0,custom_reward_function.name=compute_score,custom_reward_function.path=/scratch/user/sushil22_tamu.edu/projects/ATPO/ATPO/verl_atpo/verl/utils/reward_score/deep_research_em.py,data.max_prompt_length=2000,data.max_response_length=6192,data.prompt_key=prompt,data.train_batch_size=64,data.train_files=/scratch/user/sushil22_tamu.edu/projects/ATPO/rl_datasets/hotpotqa/train.parquet,data.val_files=["/scratch/user/sushil22_tamu.edu/projects/ATPO/rl_datasets/hotpotqa/test_512.parquet"],ray_init.num_cpus=null,reward_model.reward_manager=naive,trainer.critic_warmup=0,trainer.default_local_dir=/scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo_hotpotqa_qwen8b/20260506_124901/,trainer.experiment_name=atpo_hotpotqa_qwen8b,trainer.logger=[console,
|
| 190 |
+
wandb],trainer.max_actor_ckpt_to_keep=2,trainer.n_gpus_per_node=8,trainer.nnodes=1,trainer.project_name=Neurips-final-runs,trainer.resume_from_path=/scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo-qwen3-8b-hotpotqa-step200/global_step_200,trainer.resume_mode=resume_path,trainer.rollout_data_dir=/scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo_hotpotqa_qwen8b/20260506_124901//rollout,trainer.save_freq=50,trainer.test_freq=10,trainer.total_epochs=1,trainer.val_before_train=True,trainer.validation_data_dir=/scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo_hotpotqa_qwen8b/20260506_124901//validation
|
| 191 |
+
id: ???
|
| 192 |
+
num: ???
|
| 193 |
+
config_name: ppo_trainer_dr.yaml
|
| 194 |
+
env_set: {}
|
| 195 |
+
env_copy: []
|
| 196 |
+
config:
|
| 197 |
+
override_dirname:
|
| 198 |
+
kv_sep: '='
|
| 199 |
+
item_sep: ','
|
| 200 |
+
exclude_keys: []
|
| 201 |
+
runtime:
|
| 202 |
+
version: 1.3.2
|
| 203 |
+
version_base: '1.3'
|
| 204 |
+
cwd: /scratch/user/sushil22_tamu.edu/projects/ATPO
|
| 205 |
+
config_sources:
|
| 206 |
+
- path: hydra.conf
|
| 207 |
+
schema: pkg
|
| 208 |
+
provider: hydra
|
| 209 |
+
- path: /scratch/user/sushil22_tamu.edu/projects/ATPO/ATPO/scripts/config
|
| 210 |
+
schema: file
|
| 211 |
+
provider: main
|
| 212 |
+
- path: ''
|
| 213 |
+
schema: structured
|
| 214 |
+
provider: schema
|
| 215 |
+
output_dir: /scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo_hotpotqa_qwen8b/20260506_124901/outputs
|
| 216 |
+
choices:
|
| 217 |
+
hydra/env: default
|
| 218 |
+
hydra/callbacks: null
|
| 219 |
+
hydra/job_logging: default
|
| 220 |
+
hydra/hydra_logging: default
|
| 221 |
+
hydra/hydra_help: default
|
| 222 |
+
hydra/help: default
|
| 223 |
+
hydra/sweeper: basic
|
| 224 |
+
hydra/launcher: basic
|
| 225 |
+
hydra/output: default
|
| 226 |
+
verbose: false
|
atpo_hotpotqa_qwen8b/20260506_124901/outputs/.hydra/overrides.yaml
ADDED
|
@@ -0,0 +1,70 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
- algorithm.adv_estimator=grpo
|
| 2 |
+
- algorithm.kl_ctrl.kl_coef=0.0
|
| 3 |
+
- data.train_files=/scratch/user/sushil22_tamu.edu/projects/ATPO/rl_datasets/hotpotqa/train.parquet
|
| 4 |
+
- data.val_files=["/scratch/user/sushil22_tamu.edu/projects/ATPO/rl_datasets/hotpotqa/test_512.parquet"]
|
| 5 |
+
- data.prompt_key=prompt
|
| 6 |
+
- data.train_batch_size=64
|
| 7 |
+
- data.max_prompt_length=2000
|
| 8 |
+
- data.max_response_length=6192
|
| 9 |
+
- actor_rollout_ref.model.path=/scratch/user/sushil22_tamu.edu/projects/ATPO/models/Qwen3-8B
|
| 10 |
+
- actor_rollout_ref.model.enable_gradient_checkpointing=True
|
| 11 |
+
- actor_rollout_ref.model.use_remove_padding=True
|
| 12 |
+
- actor_rollout_ref.actor.policy_loss=gspo_turn
|
| 13 |
+
- actor_rollout_ref.actor.clip_ratio_low=3e-3
|
| 14 |
+
- actor_rollout_ref.actor.clip_ratio_high=4e-3
|
| 15 |
+
- actor_rollout_ref.actor.enable_entropy_balanced_clipping=False
|
| 16 |
+
- actor_rollout_ref.actor.enable_entropy_balanced_advantage=False
|
| 17 |
+
- actor_rollout_ref.actor.optim.lr=1e-6
|
| 18 |
+
- actor_rollout_ref.actor.ppo_mini_batch_size=8
|
| 19 |
+
- actor_rollout_ref.actor.use_dynamic_bsz=True
|
| 20 |
+
- actor_rollout_ref.actor.ppo_max_token_len_per_gpu=16384
|
| 21 |
+
- actor_rollout_ref.actor.use_kl_loss=True
|
| 22 |
+
- actor_rollout_ref.actor.kl_loss_coef=0.0
|
| 23 |
+
- actor_rollout_ref.actor.kl_loss_type=low_var_kl
|
| 24 |
+
- actor_rollout_ref.actor.fsdp_config.param_offload=False
|
| 25 |
+
- actor_rollout_ref.actor.fsdp_config.optimizer_offload=False
|
| 26 |
+
- actor_rollout_ref.rollout.enable_dynamic_rollouts=False
|
| 27 |
+
- actor_rollout_ref.rollout.log_prob_max_token_len_per_gpu=32768
|
| 28 |
+
- actor_rollout_ref.rollout.tensor_model_parallel_size=1
|
| 29 |
+
- actor_rollout_ref.rollout.name=vllm
|
| 30 |
+
- actor_rollout_ref.rollout.mode=sync_with_tool_tree
|
| 31 |
+
- actor_rollout_ref.rollout.gpu_memory_utilization=0.6
|
| 32 |
+
- actor_rollout_ref.rollout.n=22
|
| 33 |
+
- actor_rollout_ref.rollout.initial_rollouts=10
|
| 34 |
+
- actor_rollout_ref.rollout.expansion_mode=entropy
|
| 35 |
+
- actor_rollout_ref.rollout.expansion_iterations=2
|
| 36 |
+
- actor_rollout_ref.rollout.beam_size=6
|
| 37 |
+
- actor_rollout_ref.rollout.samples_per_tree=22
|
| 38 |
+
- actor_rollout_ref.rollout.branch_probability=0.5
|
| 39 |
+
- actor_rollout_ref.rollout.entropy_weight=0.2
|
| 40 |
+
- actor_rollout_ref.rollout.leaf_value_norm=True
|
| 41 |
+
- actor_rollout_ref.rollout.node_value_mode=child_softmax
|
| 42 |
+
- actor_rollout_ref.rollout.node_adv_mode=node_value
|
| 43 |
+
- actor_rollout_ref.rollout.tools.call_limit=6
|
| 44 |
+
- ++actor_rollout_ref.rollout.tools.tool_instances.search.params.cache_file=/scratch/user/sushil22_tamu.edu/projects/ATPO/search_cache/search_cache.json
|
| 45 |
+
- ++actor_rollout_ref.rollout.tools.tool_instances.search.params.api_key=unused_local_rag_server
|
| 46 |
+
- actor_rollout_ref.rollout.multi_turn.enable=True
|
| 47 |
+
- actor_rollout_ref.rollout.multi_turn.tool_config_path=/scratch/user/sushil22_tamu.edu/projects/ATPO/ATPO/verl_atpo/examples/sglang_multiturn/config/tool_config/search_tool_config.yaml
|
| 48 |
+
- actor_rollout_ref.ref.log_prob_max_token_len_per_gpu=32768
|
| 49 |
+
- actor_rollout_ref.ref.fsdp_config.param_offload=True
|
| 50 |
+
- reward_model.reward_manager=naive
|
| 51 |
+
- custom_reward_function.path=/scratch/user/sushil22_tamu.edu/projects/ATPO/ATPO/verl_atpo/verl/utils/reward_score/deep_research_em.py
|
| 52 |
+
- custom_reward_function.name=compute_score
|
| 53 |
+
- trainer.critic_warmup=0
|
| 54 |
+
- trainer.logger=[console, wandb]
|
| 55 |
+
- trainer.project_name=Neurips-final-runs
|
| 56 |
+
- trainer.experiment_name=atpo_hotpotqa_qwen8b
|
| 57 |
+
- trainer.n_gpus_per_node=8
|
| 58 |
+
- trainer.nnodes=1
|
| 59 |
+
- trainer.save_freq=50
|
| 60 |
+
- trainer.test_freq=10
|
| 61 |
+
- trainer.max_actor_ckpt_to_keep=2
|
| 62 |
+
- +actor_rollout_ref.actor.checkpoint.save_contents=[hf_model,extra]
|
| 63 |
+
- trainer.total_epochs=1
|
| 64 |
+
- trainer.default_local_dir=/scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo_hotpotqa_qwen8b/20260506_124901/
|
| 65 |
+
- trainer.val_before_train=True
|
| 66 |
+
- trainer.rollout_data_dir=/scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo_hotpotqa_qwen8b/20260506_124901//rollout
|
| 67 |
+
- trainer.validation_data_dir=/scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo_hotpotqa_qwen8b/20260506_124901//validation
|
| 68 |
+
- ray_init.num_cpus=null
|
| 69 |
+
- trainer.resume_mode=resume_path
|
| 70 |
+
- trainer.resume_from_path=/scratch/user/sushil22_tamu.edu/projects/ATPO/logs/atpo-qwen3-8b-hotpotqa-step200/global_step_200
|
atpo_hotpotqa_qwen8b/20260506_124901/outputs/main_ppo.log
ADDED
|
File without changes
|
atpo_hotpotqa_qwen8b/20260506_124901/validation/200.jsonl
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
atpo_hotpotqa_qwen8b/20260506_124901/validation/210.jsonl
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
atpo_hotpotqa_qwen8b/20260506_124901/validation/220.jsonl
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/added_tokens.json
ADDED
|
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"</think>": 151668,
|
| 3 |
+
"</tool_call>": 151658,
|
| 4 |
+
"</tool_response>": 151666,
|
| 5 |
+
"<think>": 151667,
|
| 6 |
+
"<tool_call>": 151657,
|
| 7 |
+
"<tool_response>": 151665,
|
| 8 |
+
"<|box_end|>": 151649,
|
| 9 |
+
"<|box_start|>": 151648,
|
| 10 |
+
"<|endoftext|>": 151643,
|
| 11 |
+
"<|file_sep|>": 151664,
|
| 12 |
+
"<|fim_middle|>": 151660,
|
| 13 |
+
"<|fim_pad|>": 151662,
|
| 14 |
+
"<|fim_prefix|>": 151659,
|
| 15 |
+
"<|fim_suffix|>": 151661,
|
| 16 |
+
"<|im_end|>": 151645,
|
| 17 |
+
"<|im_start|>": 151644,
|
| 18 |
+
"<|image_pad|>": 151655,
|
| 19 |
+
"<|object_ref_end|>": 151647,
|
| 20 |
+
"<|object_ref_start|>": 151646,
|
| 21 |
+
"<|quad_end|>": 151651,
|
| 22 |
+
"<|quad_start|>": 151650,
|
| 23 |
+
"<|repo_name|>": 151663,
|
| 24 |
+
"<|video_pad|>": 151656,
|
| 25 |
+
"<|vision_end|>": 151653,
|
| 26 |
+
"<|vision_pad|>": 151654,
|
| 27 |
+
"<|vision_start|>": 151652
|
| 28 |
+
}
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/config.json
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_attn_implementation_autoset": true,
|
| 3 |
+
"architectures": [
|
| 4 |
+
"Qwen3ForCausalLM"
|
| 5 |
+
],
|
| 6 |
+
"attention_bias": false,
|
| 7 |
+
"attention_dropout": 0.0,
|
| 8 |
+
"eos_token_id": 151645,
|
| 9 |
+
"head_dim": 128,
|
| 10 |
+
"hidden_act": "silu",
|
| 11 |
+
"hidden_size": 2560,
|
| 12 |
+
"initializer_range": 0.02,
|
| 13 |
+
"intermediate_size": 9728,
|
| 14 |
+
"max_position_embeddings": 40960,
|
| 15 |
+
"max_window_layers": 36,
|
| 16 |
+
"model_type": "qwen3",
|
| 17 |
+
"num_attention_heads": 32,
|
| 18 |
+
"num_hidden_layers": 36,
|
| 19 |
+
"num_key_value_heads": 8,
|
| 20 |
+
"pad_token_id": 151643,
|
| 21 |
+
"rms_norm_eps": 1e-06,
|
| 22 |
+
"rope_scaling": null,
|
| 23 |
+
"rope_theta": 1000000,
|
| 24 |
+
"sliding_window": null,
|
| 25 |
+
"tie_word_embeddings": true,
|
| 26 |
+
"torch_dtype": "float32",
|
| 27 |
+
"transformers_version": "4.51.3",
|
| 28 |
+
"use_cache": true,
|
| 29 |
+
"use_sliding_window": false,
|
| 30 |
+
"vocab_size": 151936
|
| 31 |
+
}
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/generation_config.json
ADDED
|
@@ -0,0 +1,13 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"bos_token_id": 151643,
|
| 3 |
+
"do_sample": true,
|
| 4 |
+
"eos_token_id": [
|
| 5 |
+
151645,
|
| 6 |
+
151643
|
| 7 |
+
],
|
| 8 |
+
"pad_token_id": 151643,
|
| 9 |
+
"temperature": 0.6,
|
| 10 |
+
"top_k": 20,
|
| 11 |
+
"top_p": 0.95,
|
| 12 |
+
"transformers_version": "4.51.3"
|
| 13 |
+
}
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/merges.txt
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/model_world_size_8_rank_0.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3c5ca219f75b95e92f00088168cc9a8b1ecff09d05f122ddca062e9bee78335f
|
| 3 |
+
size 2205945402
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/model_world_size_8_rank_2.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1a2e37f19499645c737402a07765e97b5cac91a413d916bb659af377af7d4dc7
|
| 3 |
+
size 2205945402
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/model_world_size_8_rank_4.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:42e0a4b239b283753df4f08d58ce1c7c662afabc2f335f63972133c2f2ec80b3
|
| 3 |
+
size 2205945402
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/model_world_size_8_rank_6.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:248c550fde02cec22aabedbf35dea04f401ee8b6c238d3b84273d55b7c7f5d6e
|
| 3 |
+
size 2205945402
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/optim_world_size_8_rank_3.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8cf1ae57e0d938f2a1fc6616eb9376f7c8db43f781ab88499ac538856e075db9
|
| 3 |
+
size 4022501290
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/optim_world_size_8_rank_7.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e9652262a8e31a43f2b086d4304d48a90aadbc67dfbd182bd31b871aa3ca53e6
|
| 3 |
+
size 4022501290
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/special_tokens_map.json
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"additional_special_tokens": [
|
| 3 |
+
"<|im_start|>",
|
| 4 |
+
"<|im_end|>",
|
| 5 |
+
"<|object_ref_start|>",
|
| 6 |
+
"<|object_ref_end|>",
|
| 7 |
+
"<|box_start|>",
|
| 8 |
+
"<|box_end|>",
|
| 9 |
+
"<|quad_start|>",
|
| 10 |
+
"<|quad_end|>",
|
| 11 |
+
"<|vision_start|>",
|
| 12 |
+
"<|vision_end|>",
|
| 13 |
+
"<|vision_pad|>",
|
| 14 |
+
"<|image_pad|>",
|
| 15 |
+
"<|video_pad|>"
|
| 16 |
+
],
|
| 17 |
+
"eos_token": {
|
| 18 |
+
"content": "<|im_end|>",
|
| 19 |
+
"lstrip": false,
|
| 20 |
+
"normalized": false,
|
| 21 |
+
"rstrip": false,
|
| 22 |
+
"single_word": false
|
| 23 |
+
},
|
| 24 |
+
"pad_token": {
|
| 25 |
+
"content": "<|endoftext|>",
|
| 26 |
+
"lstrip": false,
|
| 27 |
+
"normalized": false,
|
| 28 |
+
"rstrip": false,
|
| 29 |
+
"single_word": false
|
| 30 |
+
}
|
| 31 |
+
}
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/tokenizer_config.json
ADDED
|
@@ -0,0 +1,240 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"add_bos_token": false,
|
| 3 |
+
"add_prefix_space": false,
|
| 4 |
+
"added_tokens_decoder": {
|
| 5 |
+
"151643": {
|
| 6 |
+
"content": "<|endoftext|>",
|
| 7 |
+
"lstrip": false,
|
| 8 |
+
"normalized": false,
|
| 9 |
+
"rstrip": false,
|
| 10 |
+
"single_word": false,
|
| 11 |
+
"special": true
|
| 12 |
+
},
|
| 13 |
+
"151644": {
|
| 14 |
+
"content": "<|im_start|>",
|
| 15 |
+
"lstrip": false,
|
| 16 |
+
"normalized": false,
|
| 17 |
+
"rstrip": false,
|
| 18 |
+
"single_word": false,
|
| 19 |
+
"special": true
|
| 20 |
+
},
|
| 21 |
+
"151645": {
|
| 22 |
+
"content": "<|im_end|>",
|
| 23 |
+
"lstrip": false,
|
| 24 |
+
"normalized": false,
|
| 25 |
+
"rstrip": false,
|
| 26 |
+
"single_word": false,
|
| 27 |
+
"special": true
|
| 28 |
+
},
|
| 29 |
+
"151646": {
|
| 30 |
+
"content": "<|object_ref_start|>",
|
| 31 |
+
"lstrip": false,
|
| 32 |
+
"normalized": false,
|
| 33 |
+
"rstrip": false,
|
| 34 |
+
"single_word": false,
|
| 35 |
+
"special": true
|
| 36 |
+
},
|
| 37 |
+
"151647": {
|
| 38 |
+
"content": "<|object_ref_end|>",
|
| 39 |
+
"lstrip": false,
|
| 40 |
+
"normalized": false,
|
| 41 |
+
"rstrip": false,
|
| 42 |
+
"single_word": false,
|
| 43 |
+
"special": true
|
| 44 |
+
},
|
| 45 |
+
"151648": {
|
| 46 |
+
"content": "<|box_start|>",
|
| 47 |
+
"lstrip": false,
|
| 48 |
+
"normalized": false,
|
| 49 |
+
"rstrip": false,
|
| 50 |
+
"single_word": false,
|
| 51 |
+
"special": true
|
| 52 |
+
},
|
| 53 |
+
"151649": {
|
| 54 |
+
"content": "<|box_end|>",
|
| 55 |
+
"lstrip": false,
|
| 56 |
+
"normalized": false,
|
| 57 |
+
"rstrip": false,
|
| 58 |
+
"single_word": false,
|
| 59 |
+
"special": true
|
| 60 |
+
},
|
| 61 |
+
"151650": {
|
| 62 |
+
"content": "<|quad_start|>",
|
| 63 |
+
"lstrip": false,
|
| 64 |
+
"normalized": false,
|
| 65 |
+
"rstrip": false,
|
| 66 |
+
"single_word": false,
|
| 67 |
+
"special": true
|
| 68 |
+
},
|
| 69 |
+
"151651": {
|
| 70 |
+
"content": "<|quad_end|>",
|
| 71 |
+
"lstrip": false,
|
| 72 |
+
"normalized": false,
|
| 73 |
+
"rstrip": false,
|
| 74 |
+
"single_word": false,
|
| 75 |
+
"special": true
|
| 76 |
+
},
|
| 77 |
+
"151652": {
|
| 78 |
+
"content": "<|vision_start|>",
|
| 79 |
+
"lstrip": false,
|
| 80 |
+
"normalized": false,
|
| 81 |
+
"rstrip": false,
|
| 82 |
+
"single_word": false,
|
| 83 |
+
"special": true
|
| 84 |
+
},
|
| 85 |
+
"151653": {
|
| 86 |
+
"content": "<|vision_end|>",
|
| 87 |
+
"lstrip": false,
|
| 88 |
+
"normalized": false,
|
| 89 |
+
"rstrip": false,
|
| 90 |
+
"single_word": false,
|
| 91 |
+
"special": true
|
| 92 |
+
},
|
| 93 |
+
"151654": {
|
| 94 |
+
"content": "<|vision_pad|>",
|
| 95 |
+
"lstrip": false,
|
| 96 |
+
"normalized": false,
|
| 97 |
+
"rstrip": false,
|
| 98 |
+
"single_word": false,
|
| 99 |
+
"special": true
|
| 100 |
+
},
|
| 101 |
+
"151655": {
|
| 102 |
+
"content": "<|image_pad|>",
|
| 103 |
+
"lstrip": false,
|
| 104 |
+
"normalized": false,
|
| 105 |
+
"rstrip": false,
|
| 106 |
+
"single_word": false,
|
| 107 |
+
"special": true
|
| 108 |
+
},
|
| 109 |
+
"151656": {
|
| 110 |
+
"content": "<|video_pad|>",
|
| 111 |
+
"lstrip": false,
|
| 112 |
+
"normalized": false,
|
| 113 |
+
"rstrip": false,
|
| 114 |
+
"single_word": false,
|
| 115 |
+
"special": true
|
| 116 |
+
},
|
| 117 |
+
"151657": {
|
| 118 |
+
"content": "<tool_call>",
|
| 119 |
+
"lstrip": false,
|
| 120 |
+
"normalized": false,
|
| 121 |
+
"rstrip": false,
|
| 122 |
+
"single_word": false,
|
| 123 |
+
"special": false
|
| 124 |
+
},
|
| 125 |
+
"151658": {
|
| 126 |
+
"content": "</tool_call>",
|
| 127 |
+
"lstrip": false,
|
| 128 |
+
"normalized": false,
|
| 129 |
+
"rstrip": false,
|
| 130 |
+
"single_word": false,
|
| 131 |
+
"special": false
|
| 132 |
+
},
|
| 133 |
+
"151659": {
|
| 134 |
+
"content": "<|fim_prefix|>",
|
| 135 |
+
"lstrip": false,
|
| 136 |
+
"normalized": false,
|
| 137 |
+
"rstrip": false,
|
| 138 |
+
"single_word": false,
|
| 139 |
+
"special": false
|
| 140 |
+
},
|
| 141 |
+
"151660": {
|
| 142 |
+
"content": "<|fim_middle|>",
|
| 143 |
+
"lstrip": false,
|
| 144 |
+
"normalized": false,
|
| 145 |
+
"rstrip": false,
|
| 146 |
+
"single_word": false,
|
| 147 |
+
"special": false
|
| 148 |
+
},
|
| 149 |
+
"151661": {
|
| 150 |
+
"content": "<|fim_suffix|>",
|
| 151 |
+
"lstrip": false,
|
| 152 |
+
"normalized": false,
|
| 153 |
+
"rstrip": false,
|
| 154 |
+
"single_word": false,
|
| 155 |
+
"special": false
|
| 156 |
+
},
|
| 157 |
+
"151662": {
|
| 158 |
+
"content": "<|fim_pad|>",
|
| 159 |
+
"lstrip": false,
|
| 160 |
+
"normalized": false,
|
| 161 |
+
"rstrip": false,
|
| 162 |
+
"single_word": false,
|
| 163 |
+
"special": false
|
| 164 |
+
},
|
| 165 |
+
"151663": {
|
| 166 |
+
"content": "<|repo_name|>",
|
| 167 |
+
"lstrip": false,
|
| 168 |
+
"normalized": false,
|
| 169 |
+
"rstrip": false,
|
| 170 |
+
"single_word": false,
|
| 171 |
+
"special": false
|
| 172 |
+
},
|
| 173 |
+
"151664": {
|
| 174 |
+
"content": "<|file_sep|>",
|
| 175 |
+
"lstrip": false,
|
| 176 |
+
"normalized": false,
|
| 177 |
+
"rstrip": false,
|
| 178 |
+
"single_word": false,
|
| 179 |
+
"special": false
|
| 180 |
+
},
|
| 181 |
+
"151665": {
|
| 182 |
+
"content": "<tool_response>",
|
| 183 |
+
"lstrip": false,
|
| 184 |
+
"normalized": false,
|
| 185 |
+
"rstrip": false,
|
| 186 |
+
"single_word": false,
|
| 187 |
+
"special": false
|
| 188 |
+
},
|
| 189 |
+
"151666": {
|
| 190 |
+
"content": "</tool_response>",
|
| 191 |
+
"lstrip": false,
|
| 192 |
+
"normalized": false,
|
| 193 |
+
"rstrip": false,
|
| 194 |
+
"single_word": false,
|
| 195 |
+
"special": false
|
| 196 |
+
},
|
| 197 |
+
"151667": {
|
| 198 |
+
"content": "<think>",
|
| 199 |
+
"lstrip": false,
|
| 200 |
+
"normalized": false,
|
| 201 |
+
"rstrip": false,
|
| 202 |
+
"single_word": false,
|
| 203 |
+
"special": false
|
| 204 |
+
},
|
| 205 |
+
"151668": {
|
| 206 |
+
"content": "</think>",
|
| 207 |
+
"lstrip": false,
|
| 208 |
+
"normalized": false,
|
| 209 |
+
"rstrip": false,
|
| 210 |
+
"single_word": false,
|
| 211 |
+
"special": false
|
| 212 |
+
}
|
| 213 |
+
},
|
| 214 |
+
"additional_special_tokens": [
|
| 215 |
+
"<|im_start|>",
|
| 216 |
+
"<|im_end|>",
|
| 217 |
+
"<|object_ref_start|>",
|
| 218 |
+
"<|object_ref_end|>",
|
| 219 |
+
"<|box_start|>",
|
| 220 |
+
"<|box_end|>",
|
| 221 |
+
"<|quad_start|>",
|
| 222 |
+
"<|quad_end|>",
|
| 223 |
+
"<|vision_start|>",
|
| 224 |
+
"<|vision_end|>",
|
| 225 |
+
"<|vision_pad|>",
|
| 226 |
+
"<|image_pad|>",
|
| 227 |
+
"<|video_pad|>"
|
| 228 |
+
],
|
| 229 |
+
"bos_token": null,
|
| 230 |
+
"chat_template": "{%- if tools %}\n {{- '<|im_start|>system\\n' }}\n {%- if messages[0].role == 'system' %}\n {{- messages[0].content + '\\n\\n' }}\n {%- endif %}\n {{- \"# Tools\\n\\nYou may call one or more functions to assist with the user query.\\n\\nYou are provided with function signatures within <tools></tools> XML tags:\\n<tools>\" }}\n {%- for tool in tools %}\n {{- \"\\n\" }}\n {{- tool | tojson }}\n {%- endfor %}\n {{- \"\\n</tools>\\n\\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\\n<tool_call>\\n{\\\"name\\\": <function-name>, \\\"arguments\\\": <args-json-object>}\\n</tool_call><|im_end|>\\n\" }}\n{%- else %}\n {%- if messages[0].role == 'system' %}\n {{- '<|im_start|>system\\n' + messages[0].content + '<|im_end|>\\n' }}\n {%- endif %}\n{%- endif %}\n{%- set ns = namespace(multi_step_tool=true, last_query_index=messages|length - 1) %}\n{%- for message in messages[::-1] %}\n {%- set index = (messages|length - 1) - loop.index0 %}\n {%- if ns.multi_step_tool and message.role == \"user\" and message.content is string and not(message.content.startswith('<tool_response>') and message.content.endswith('</tool_response>')) %}\n {%- set ns.multi_step_tool = false %}\n {%- set ns.last_query_index = index %}\n {%- endif %}\n{%- endfor %}\n{%- for message in messages %}\n {%- if message.content is string %}\n {%- set content = message.content %}\n {%- else %}\n {%- set content = '' %}\n {%- endif %}\n {%- if (message.role == \"user\") or (message.role == \"system\" and not loop.first) %}\n {{- '<|im_start|>' + message.role + '\\n' + content + '<|im_end|>' + '\\n' }}\n {%- elif message.role == \"assistant\" %}\n {%- set reasoning_content = '' %}\n {%- if message.reasoning_content is string %}\n {%- set reasoning_content = message.reasoning_content %}\n {%- else %}\n {%- if '</think>' in content %}\n {%- set reasoning_content = content.split('</think>')[0].rstrip('\\n').split('<think>')[-1].lstrip('\\n') %}\n {%- set content = content.split('</think>')[-1].lstrip('\\n') %}\n {%- endif %}\n {%- endif %}\n {%- if loop.index0 > ns.last_query_index %}\n {%- if loop.last or (not loop.last and reasoning_content) %}\n {{- '<|im_start|>' + message.role + '\\n<think>\\n' + reasoning_content.strip('\\n') + '\\n</think>\\n\\n' + content.lstrip('\\n') }}\n {%- else %}\n {{- '<|im_start|>' + message.role + '\\n' + content }}\n {%- endif %}\n {%- else %}\n {{- '<|im_start|>' + message.role + '\\n' + content }}\n {%- endif %}\n {%- if message.tool_calls %}\n {%- for tool_call in message.tool_calls %}\n {%- if (loop.first and content) or (not loop.first) %}\n {{- '\\n' }}\n {%- endif %}\n {%- if tool_call.function %}\n {%- set tool_call = tool_call.function %}\n {%- endif %}\n {{- '<tool_call>\\n{\"name\": \"' }}\n {{- tool_call.name }}\n {{- '\", \"arguments\": ' }}\n {%- if tool_call.arguments is string %}\n {{- tool_call.arguments }}\n {%- else %}\n {{- tool_call.arguments | tojson }}\n {%- endif %}\n {{- '}\\n</tool_call>' }}\n {%- endfor %}\n {%- endif %}\n {{- '<|im_end|>\\n' }}\n {%- elif message.role == \"tool\" %}\n {%- if loop.first or (messages[loop.index0 - 1].role != \"tool\") %}\n {{- '<|im_start|>user' }}\n {%- endif %}\n {{- '\\n<tool_response>\\n' }}\n {{- content }}\n {{- '\\n</tool_response>' }}\n {%- if loop.last or (messages[loop.index0 + 1].role != \"tool\") %}\n {{- '<|im_end|>\\n' }}\n {%- endif %}\n {%- endif %}\n{%- endfor %}\n{%- if add_generation_prompt %}\n {{- '<|im_start|>assistant\\n' }}\n {%- if enable_thinking is defined and enable_thinking is false %}\n {{- '<think>\\n\\n</think>\\n\\n' }}\n {%- endif %}\n{%- endif %}",
|
| 231 |
+
"clean_up_tokenization_spaces": false,
|
| 232 |
+
"eos_token": "<|im_end|>",
|
| 233 |
+
"errors": "replace",
|
| 234 |
+
"extra_special_tokens": {},
|
| 235 |
+
"model_max_length": 131072,
|
| 236 |
+
"pad_token": "<|endoftext|>",
|
| 237 |
+
"split_special_tokens": false,
|
| 238 |
+
"tokenizer_class": "Qwen2Tokenizer",
|
| 239 |
+
"unk_token": null
|
| 240 |
+
}
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/global_step_200/actor/vocab.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/latest_checkpointed_iteration.txt
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
200
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/1.jsonl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4505874d0421dddd4193dea8d38d4f0e511d577ab39dea751bbec47e17eb0b54
|
| 3 |
+
size 17646341
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/10.jsonl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e2e466b445b49996ea06998a90b1fd255a37cb9eda735d38acf829e8ec797e1c
|
| 3 |
+
size 15670098
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/101.jsonl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b0cc3d6ad3c589cabf1dbec218cb198a0acbd3e493d861cfbadcdd46229780d0
|
| 3 |
+
size 13456182
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/118.jsonl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:99083b5b83e00cc943dd5a7f0f03192be5e78864d97d0559c5ae2a514e58bfe3
|
| 3 |
+
size 14548777
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/122.jsonl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f0957697eb6cc2a6a53513885afc38531573bdd28121a3c86e6cf7a415c1aca1
|
| 3 |
+
size 13743775
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/129.jsonl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a550c99130a9064c8a6cbabc943a96646b731ba5f81058aa1ac1f7013ad0251e
|
| 3 |
+
size 14370467
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/13.jsonl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1ae25147f6ed86bf621fcdb7252274253f2632f0222feb72ad51a73379ad7607
|
| 3 |
+
size 15077323
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/130.jsonl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:fd1f06ed8a8dec0e76ccd7c844cd52da123075ebfe8d74701f39dd74c3bc348d
|
| 3 |
+
size 14168128
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/136.jsonl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ce61183d07f48923a8bcb12de2e3a434b9129fe7d1845c5580746c8ef0881eb0
|
| 3 |
+
size 14347063
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/143.jsonl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6b27889e47f8d82bbd0706a4e86ecfd25134fee0f6b3b9461b7be85a42b594b0
|
| 3 |
+
size 13529441
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/150.jsonl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3f0292eb297b596f0250c2bd7c91ed1bacb732f3e481dc3f87a21777305ccb2e
|
| 3 |
+
size 13971928
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/156.jsonl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:79c049a6800e5807b01cda1d065331cc776cac6d2367ef531f8640d880235bc7
|
| 3 |
+
size 14061460
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/157.jsonl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:27d1fdd87d9e72045d5efb7f460cfbcd2c2880e0de5e758f179fa2136eaac769
|
| 3 |
+
size 14182823
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/160.jsonl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b15e2cc4c5e5a84395833fdb5f76f63e3090465143f5bdadd44a16ee3af79a5f
|
| 3 |
+
size 14120339
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/163.jsonl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f3790ab394e31b6e99d2ad1ddabe9e8d5acfdd8e5b662e2362a97868d45b7840
|
| 3 |
+
size 13695941
|
cache_hotpotqa_qwen4b_ablation_noadvantage/20260504_231531/rollout/176.jsonl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7b910d7159a01c6ab4b6f749412e4b20de38561d2ecd0f18eb1c3b48843157c9
|
| 3 |
+
size 14044003
|