lyhisme commited on
Commit
c722cc1
·
verified ·
1 Parent(s): c9ac3c3

Remove accidental run and build artifacts

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. code/orarl.egg-info/PKG-INFO +0 -230
  2. code/orarl.egg-info/SOURCES.txt +0 -196
  3. code/orarl.egg-info/dependency_links.txt +0 -1
  4. code/orarl.egg-info/entry_points.txt +0 -5
  5. code/orarl.egg-info/requires.txt +0 -12
  6. code/orarl.egg-info/top_level.txt +0 -2
  7. code/runs/smoke-training-9b-20260825_174304/smoke.log +0 -467
  8. code/runs/smoke-training-9b-20260825_175213/smoke.log +0 -827
  9. code/runs/smoke-training-9b-20260825_175949/grpo/checkpoint_tracker.json +0 -6
  10. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/extra_state_world_size_8_rank_0.pt +0 -0
  11. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/extra_state_world_size_8_rank_1.pt +0 -0
  12. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/extra_state_world_size_8_rank_2.pt +0 -0
  13. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/extra_state_world_size_8_rank_3.pt +0 -0
  14. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/extra_state_world_size_8_rank_4.pt +0 -0
  15. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/extra_state_world_size_8_rank_5.pt +0 -0
  16. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/extra_state_world_size_8_rank_6.pt +0 -0
  17. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/extra_state_world_size_8_rank_7.pt +0 -0
  18. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/huggingface/chat_template.jinja +0 -154
  19. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/huggingface/config.json +0 -109
  20. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/huggingface/generation_config.json +0 -13
  21. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/huggingface/processor_config.json +0 -60
  22. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/huggingface/tokenizer.json +0 -3
  23. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/huggingface/tokenizer_config.json +0 -32
  24. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/model_world_size_8_rank_0.pt +0 -3
  25. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/model_world_size_8_rank_1.pt +0 -3
  26. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/model_world_size_8_rank_2.pt +0 -3
  27. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/model_world_size_8_rank_3.pt +0 -3
  28. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/model_world_size_8_rank_4.pt +0 -3
  29. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/model_world_size_8_rank_5.pt +0 -3
  30. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/model_world_size_8_rank_6.pt +0 -3
  31. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/model_world_size_8_rank_7.pt +0 -3
  32. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/optim_world_size_8_rank_0.pt +0 -3
  33. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/optim_world_size_8_rank_1.pt +0 -3
  34. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/optim_world_size_8_rank_2.pt +0 -3
  35. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/optim_world_size_8_rank_3.pt +0 -3
  36. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/optim_world_size_8_rank_4.pt +0 -3
  37. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/optim_world_size_8_rank_5.pt +0 -3
  38. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/optim_world_size_8_rank_6.pt +0 -3
  39. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/optim_world_size_8_rank_7.pt +0 -3
  40. code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/dataloader.pt +0 -0
  41. code/runs/smoke-training-9b-20260825_175949/orarl/checkpoint_tracker.json +0 -6
  42. code/runs/smoke-training-9b-20260825_175949/orarl/global_step_1/actor/extra_state_world_size_8_rank_0.pt +0 -0
  43. code/runs/smoke-training-9b-20260825_175949/orarl/global_step_1/actor/extra_state_world_size_8_rank_1.pt +0 -0
  44. code/runs/smoke-training-9b-20260825_175949/orarl/global_step_1/actor/extra_state_world_size_8_rank_2.pt +0 -0
  45. code/runs/smoke-training-9b-20260825_175949/orarl/global_step_1/actor/extra_state_world_size_8_rank_3.pt +0 -0
  46. code/runs/smoke-training-9b-20260825_175949/orarl/global_step_1/actor/extra_state_world_size_8_rank_4.pt +0 -0
  47. code/runs/smoke-training-9b-20260825_175949/orarl/global_step_1/actor/extra_state_world_size_8_rank_5.pt +0 -0
  48. code/runs/smoke-training-9b-20260825_175949/orarl/global_step_1/actor/extra_state_world_size_8_rank_6.pt +0 -0
  49. code/runs/smoke-training-9b-20260825_175949/orarl/global_step_1/actor/extra_state_world_size_8_rank_7.pt +0 -0
  50. code/runs/smoke-training-9b-20260825_175949/orarl/global_step_1/actor/huggingface/chat_template.jinja +0 -154
code/orarl.egg-info/PKG-INFO DELETED
@@ -1,230 +0,0 @@
1
- Metadata-Version: 2.4
2
- Name: orarl
3
- Version: 0.1.0
4
- Summary: Annotations as rollouts for unified video MLLM reinforcement learning
5
- Author: Yunheng Li, Guohong Mu, Hao Li, Shengsheng Qian, Dingwen Zhang, Qibin Hou, Ming-Ming Cheng
6
- License-Expression: Apache-2.0
7
- Project-URL: Homepage, https://orarl.github.io/
8
- Project-URL: Documentation, https://orarl.github.io/
9
- Keywords: reinforcement-learning,video-understanding,multimodal,grpo
10
- Classifier: Development Status :: 3 - Alpha
11
- Classifier: Programming Language :: Python :: 3
12
- Classifier: Programming Language :: Python :: 3.10
13
- Classifier: Programming Language :: Python :: 3.11
14
- Classifier: Programming Language :: Python :: 3.12
15
- Requires-Python: >=3.10
16
- Description-Content-Type: text/markdown
17
- License-File: LICENSE
18
- License-File: NOTICE
19
- Requires-Dist: numpy
20
- Requires-Dist: PyYAML
21
- Requires-Dist: torch
22
- Provides-Extra: test
23
- Requires-Dist: pytest; extra == "test"
24
- Provides-Extra: lint
25
- Requires-Dist: ruff; extra == "lint"
26
- Provides-Extra: hf
27
- Requires-Dist: huggingface_hub; extra == "hf"
28
- Dynamic: license-file
29
-
30
- <p align="right"><a href="README_zh.md">简体中文</a></p>
31
-
32
- <div align="center">
33
-
34
- # OraRL
35
-
36
- ### Annotations as Rollouts
37
-
38
- **Efficient and scalable reinforcement learning for unified video MLLMs**
39
-
40
- Yunheng Li · Guohong Mu · Hao Li · Shengsheng Qian · Dingwen Zhang ·
41
- Qibin Hou · Ming-Ming Cheng
42
-
43
- <p>
44
- <a href="https://arxiv.org/abs/2608.20492">📄 Paper</a>
45
- &nbsp;&nbsp;·&nbsp;&nbsp;
46
- <a href="https://orarl.github.io/">🌐 Project Page</a>
47
- &nbsp;&nbsp;·&nbsp;&nbsp;
48
- <a href="#models">🤗 Models (4B / 9B)</a>
49
- </p>
50
- <p>
51
- <a href="docs/environment.md">⚙️ Environment</a>
52
- &nbsp;&nbsp;·&nbsp;&nbsp;
53
- <a href="docs/training.md">🚀 Training</a>
54
- &nbsp;&nbsp;·&nbsp;&nbsp;
55
- <a href="docs/evaluation.md">📊 Evaluation</a>
56
- &nbsp;&nbsp;·&nbsp;&nbsp;
57
- <a href="LICENSE">⚖️ License</a>
58
- </p>
59
-
60
- <a href="https://orarl.github.io/assets/orarl-teaser.mp4">
61
- <img src="assets/orarl-hero.gif"
62
- alt="Animated OraRL method preview" width="92%">
63
- </a>
64
-
65
- **▶ Click the image to watch the 1:38 project overview.**
66
-
67
- </div>
68
-
69
- ## Why OraRL
70
-
71
- - **Annotation-as-rollout:** annotations become reliable positive rollouts while
72
- policy samples retain an on-policy baseline.
73
- - **Seven task families:** one update rule covers temporal and spatial grounding,
74
- segmentation, tracking, spatial-temporal grounding, video QA, and spatial
75
- intelligence.
76
- - **Efficient training (4B):** sign-balanced pruning delivers **1.48× faster
77
- updates** (**92.5 → 62.4 s/step**) while reducing peak per-GPU memory from
78
- **62.4 to 50.9 GB**.
79
- - **Efficient inference:** on one H20 with vLLM in BF16, weight loading occupies
80
- **8.6 GiB (4B)** and **17.6 GiB (9B)**. On ten-minute, 2-fps videos,
81
- answer-only decoding cuts median post-TTFT latency from **4.78 s to 130 ms**
82
- and total latency from **29.03 to 24.30 s**.
83
- - **Multimodal veRL infrastructure:** a unified video contract carries cached
84
- artifacts, raw paths, or inline frame tensors through vLLM rollouts and FSDP
85
- updates, with decode-once frame reuse, temporal metadata, task-grouped
86
- batching, asynchronous Ray rewards, and safe hybrid-engine cache handling.
87
-
88
- ## OraRL in One Update
89
-
90
- <p align="center">
91
- <img src="assets/orarl-method.gif"
92
- alt="Animated OraRL framework" width="96%">
93
- </p>
94
-
95
- An OraRL update separates reliable annotation guidance from on-policy
96
- normalization:
97
-
98
- 1. **Build the group:** append one serialized annotation rollout to the policy
99
- samples generated for the same prompt.
100
- 2. **Keep the baseline on-policy:** estimate the group baseline from policy
101
- rewards only.
102
- 3. **Guide and select:** convert the annotation-policy reward gap into a
103
- correction, then retain a sign-balanced subset for the update.
104
-
105
- This design uses task-native annotations directly and requires no
106
- chain-of-thought supervision or decoding.
107
-
108
- ## Video-ORA Results
109
-
110
- <p align="center">
111
- <img src="assets/paper-results.png"
112
- alt="Video-ORA-9B results across seven task families" width="100%">
113
- </p>
114
-
115
- ### Dataset-Level Results
116
-
117
- <picture>
118
- <source media="(prefers-color-scheme: dark)"
119
- srcset="assets/video_ora_benchmark_matrix_dark.svg">
120
- <source media="(prefers-color-scheme: light)"
121
- srcset="assets/video_ora_benchmark_matrix_light.svg">
122
- <img src="assets/video_ora_benchmark_matrix_light.svg"
123
- alt="Dataset-level benchmark matrix comparing Video-ORA with multimodal baselines"
124
- width="100%">
125
- </picture>
126
-
127
- Video-ORA-9B leads the matched seven-family comparison without CoT decoding.
128
- Best and second-best values are highlighted per row; `†` denotes an
129
- original-report value whose frame, prompt, split, or decoding settings may
130
- differ. Averages require complete family coverage.
131
-
132
- <!-- <details>
133
- <summary>Benchmark sources</summary>
134
-
135
- Unmarked values come from Tables 1–8 and Appendix Table 20 of the latest
136
- [OraRL paper](https://arxiv.org/abs/2608.20492). External entries follow the original
137
- [LLaVA-OneVision-2](https://arxiv.org/abs/2605.25979),
138
- [VideoChat3](https://github.com/MCG-NJU/VideoChat3), and
139
- [OneThinker](https://arxiv.org/abs/2512.03043) reports. OneThinker is cited only
140
- as the source of public Qwen3-VL scores. ReVSI uses each model's reported frame
141
- setting; the paper's three-benchmark spatial-intelligence average excludes it.
142
-
143
- </details> -->
144
-
145
- ### Model Scaling
146
-
147
- <p align="center">
148
- <img src="assets/orarl-model-scaling.gif"
149
- alt="Animated Video-ORA model scaling from 0.8B to 9B" width="100%">
150
- </p>
151
-
152
- ### Data Scaling
153
-
154
- <p align="center">
155
- <img src="assets/orarl-data-scaling.gif"
156
- alt="Animated OraRL data scaling and reward dynamics" width="100%">
157
- </p>
158
-
159
- ## Models
160
-
161
- | Model | Backbone | Released recipe | Weights |
162
- | --- | --- | --- | --- |
163
- | **Video-ORA-9B** | Qwen3.5-9B | `orarl_9b.yaml` | [Hugging Face](https://huggingface.co/OraRL/Video-ORA-9B) |
164
- | **Video-ORA-4B** | Qwen3.5-4B | `orarl_4b.yaml` | Hugging Face (coming soon) |
165
-
166
- ### vLLM Serving
167
-
168
- Both Video-ORA checkpoints load directly with **vLLM 0.19.1** for
169
- OpenAI-compatible serving:
170
-
171
- ```bash
172
- MODEL=OraRL/Video-ORA-9B
173
-
174
- vllm serve "$MODEL" \
175
- --served-model-name Video-ORA-9B \
176
- --trust-remote-code \
177
- --dtype bfloat16 \
178
- --tensor-parallel-size 1 \
179
- --max-model-len 131072 \
180
- --limit-mm-per-prompt '{"image": 1, "video": 1}'
181
- ```
182
-
183
- Set `--tensor-parallel-size` to the GPU count for multi-GPU deployment and
184
- lower `--max-model-len` on smaller-memory devices. Use
185
- `enable_thinking=false` in the chat template for answer-only inference.
186
-
187
- ## Use OraRL
188
-
189
- The release is organized around three user-facing workflows:
190
-
191
- 1. **[Environment](docs/environment.md):** install the pinned CUDA stack that
192
- covers both the bundled trainer and the evaluators.
193
- 2. **[Training](docs/training.md):** prepare licensed local training data and
194
- launch GRPO or OraRL on one or multiple nodes.
195
- 3. **[Evaluation](docs/evaluation.md):** download Video-ORA and OraRL-Data,
196
- then run a smoke test or the complete paper suite.
197
-
198
- Training and evaluation are dry runs by default; inspect the resolved command
199
- before adding `--run`. Checkpoints and evaluation media are hosted under the
200
- [OraRL Hugging Face organization](https://huggingface.co/OraRL).
201
-
202
- ## Acknowledgements
203
-
204
- OraRL is built on [veRL](https://github.com/volcengine/verl) — a
205
- high-performance RL framework with HybridEngine. We thank its authors and
206
- contributors for open-sourcing the training infrastructure.
207
-
208
- ## License
209
-
210
- OraRL source is released under [Apache-2.0](LICENSE). Datasets, models,
211
- benchmarks, and optional dependencies retain their original licenses; see
212
- [NOTICE](NOTICE).
213
-
214
- ## Citation
215
-
216
- If you find OraRL useful, please consider giving this repository a ⭐ and
217
- citing our [paper](https://arxiv.org/abs/2608.20492).
218
-
219
- ```bibtex
220
- @article{li2026orarl,
221
- title = {Annotations as Rollouts: Efficient and Scalable
222
- Reinforcement Learning for Video MLLMs},
223
- author = {Li, Yunheng and Mu, Guohong and Li, Hao and
224
- Qian, Shengsheng and Zhang, Dingwen and Hou, Qibin
225
- and Cheng, Ming-Ming},
226
- journal = {arXiv preprint arXiv:2608.20492},
227
- year = {2026},
228
- url = {https://arxiv.org/abs/2608.20492}
229
- }
230
- ```
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
code/orarl.egg-info/SOURCES.txt DELETED
@@ -1,196 +0,0 @@
1
- LICENSE
2
- MANIFEST.in
3
- NOTICE
4
- README.md
5
- environment.yml
6
- pyproject.toml
7
- requirements-cu129.txt
8
- configs/data_sources.example.yaml
9
- configs/grpo_4b.yaml
10
- configs/grpo_9b.yaml
11
- configs/orarl_4b.yaml
12
- configs/orarl_9b.yaml
13
- data/eval/README.md
14
- data/eval/datasets.jsonl
15
- docs/environment.md
16
- docs/environment_zh.md
17
- docs/evaluation.md
18
- docs/evaluation_zh.md
19
- docs/training.md
20
- docs/training_zh.md
21
- eval/README.md
22
- eval/task/canonical_data.py
23
- eval/task/eval.sh
24
- eval/task/eval_image_mc_vllm.py
25
- eval/task/eval_prompt.py
26
- eval/task/eval_revsi_multinode.sh
27
- eval/task/eval_vllm.py
28
- eval/task/qwenvl_decord_patch.py
29
- eval/task/mindcube/data_utils.py
30
- eval/task/mmsi/eval_mmsi_transformers.py
31
- eval/task/mmsi/run_eval_transformers.sh
32
- eval/task/revsi/eval_revsi_vllm.py
33
- eval/task/revsi/merge_multinode_shards.py
34
- eval/task/revsi/run_eval_vllm.sh
35
- eval/task/segmentation/eval_seg_vllm.py
36
- eval/task/segmentation/post_sam2.py
37
- eval/task/segmentation/run_eval_vllm.sh
38
- eval/task/spatial_grounding/_grounding_utils.py
39
- eval/task/spatial_grounding/eval_refcoco_vllm.py
40
- eval/task/spatial_temporal_grounding/eval_stvg_vllm.py
41
- eval/task/temporal_grounding/eval_timelens_hf.py
42
- eval/task/temporal_grounding/run_eval.sh
43
- eval/task/tracking/eval_tracking_vllm.py
44
- orarl/__init__.py
45
- orarl/resources.py
46
- orarl.egg-info/PKG-INFO
47
- orarl.egg-info/SOURCES.txt
48
- orarl.egg-info/dependency_links.txt
49
- orarl.egg-info/entry_points.txt
50
- orarl.egg-info/requires.txt
51
- orarl.egg-info/top_level.txt
52
- orarl/algorithm/__init__.py
53
- orarl/algorithm/_utils.py
54
- orarl/algorithm/advantages.py
55
- orarl/algorithm/config.py
56
- orarl/algorithm/correction.py
57
- orarl/algorithm/selection.py
58
- orarl/cli/__init__.py
59
- orarl/cli/eval_data.py
60
- orarl/cli/evaluate.py
61
- orarl/cli/prepare.py
62
- orarl/cli/train.py
63
- orarl/data/__init__.py
64
- orarl/data/build.py
65
- orarl/data/identity.py
66
- orarl/data/schema.py
67
- orarl/evaluation/__init__.py
68
- orarl/evaluation/card.py
69
- orarl/evaluation/converters.py
70
- orarl/evaluation/hub.py
71
- orarl/evaluation/layout.py
72
- orarl/evaluation/manifest.py
73
- orarl/evaluation/schema.py
74
- orarl/evaluation/sources.py
75
- orarl/evaluation/staging.py
76
- orarl/rewards/__init__.py
77
- orarl/rewards/router.py
78
- orarl/rewards/types.py
79
- orarl/rewards/adapters/__init__.py
80
- orarl/rewards/adapters/_common.py
81
- orarl/rewards/adapters/segmentation.py
82
- orarl/rewards/adapters/spatial_grounding.py
83
- orarl/rewards/adapters/spatial_intelligence.py
84
- orarl/rewards/adapters/spatial_temporal_grounding.py
85
- orarl/rewards/adapters/temporal_grounding.py
86
- orarl/rewards/adapters/tracking.py
87
- orarl/rewards/adapters/video_qa.py
88
- scripts/check_environment.py
89
- scripts/check_release.py
90
- scripts/create_conda_env.sh
91
- scripts/create_eval_source_manifest.py
92
- scripts/evaluate.sh
93
- scripts/finalize_eval_index.py
94
- scripts/install_conda_runtime_hook.sh
95
- scripts/launch_multinode.sh
96
- scripts/model_merger.py
97
- scripts/organize_evaluation_outputs.py
98
- scripts/prepare_data.sh
99
- scripts/smoke_evaluation.sh
100
- scripts/smoke_training.sh
101
- scripts/train_grpo.sh
102
- scripts/train_orarl.sh
103
- tests/test_algorithm_advantages.py
104
- tests/test_algorithm_correction.py
105
- tests/test_algorithm_selection.py
106
- tests/test_builtin_reward_adapters.py
107
- tests/test_cli_evaluate.py
108
- tests/test_cli_metadata.py
109
- tests/test_cli_train.py
110
- tests/test_data_build.py
111
- tests/test_data_schema.py
112
- tests/test_environment_spec.py
113
- tests/test_eval_canonical_data.py
114
- tests/test_eval_source_discovery.py
115
- tests/test_evaluation_manifest.py
116
- tests/test_evaluation_release.py
117
- tests/test_evaluation_runtime.py
118
- tests/test_evaluation_schema.py
119
- tests/test_evaluation_staging.py
120
- tests/test_organize_evaluation_outputs.py
121
- tests/test_release_hygiene.py
122
- tests/test_resources.py
123
- tests/test_rewards.py
124
- tests/test_runtime_reward_entrypoint.py
125
- tests/test_training_jsonl_dataset.py
126
- tests/test_training_runtime.py
127
- verl/__init__.py
128
- verl/protocol.py
129
- verl/models/__init__.py
130
- verl/models/monkey_patch.py
131
- verl/models/transformers/__init__.py
132
- verl/models/transformers/flash_attention_utils.py
133
- verl/models/transformers/qwen2_vl.py
134
- verl/models/transformers/qwen3_5.py
135
- verl/models/transformers/qwen3_vl.py
136
- verl/single_controller/__init__.py
137
- verl/single_controller/base/__init__.py
138
- verl/single_controller/base/decorator.py
139
- verl/single_controller/base/worker.py
140
- verl/single_controller/base/worker_group.py
141
- verl/single_controller/base/register_center/__init__.py
142
- verl/single_controller/base/register_center/ray.py
143
- verl/single_controller/ray/__init__.py
144
- verl/single_controller/ray/base.py
145
- verl/trainer/__init__.py
146
- verl/trainer/config.py
147
- verl/trainer/core_algos.py
148
- verl/trainer/data_loader.py
149
- verl/trainer/main.py
150
- verl/trainer/metrics.py
151
- verl/trainer/orarl_config.py
152
- verl/trainer/orarl_post_selection.py
153
- verl/trainer/orarl_selection.py
154
- verl/trainer/ray_trainer.py
155
- verl/utils/__init__.py
156
- verl/utils/dataset.py
157
- verl/utils/flops_counter.py
158
- verl/utils/fsdp_utils.py
159
- verl/utils/model_utils.py
160
- verl/utils/multimodal_contract.py
161
- verl/utils/prompt_template.py
162
- verl/utils/py_functional.py
163
- verl/utils/seqlen_balancing.py
164
- verl/utils/tokenizer.py
165
- verl/utils/torch_dtypes.py
166
- verl/utils/torch_functional.py
167
- verl/utils/ulysses.py
168
- verl/utils/checkpoint/__init__.py
169
- verl/utils/checkpoint/checkpoint_manager.py
170
- verl/utils/checkpoint/fsdp_checkpoint_manager.py
171
- verl/utils/logger/__init__.py
172
- verl/utils/logger/gen_logger.py
173
- verl/utils/logger/logger.py
174
- verl/workers/__init__.py
175
- verl/workers/config.py
176
- verl/workers/fsdp_workers.py
177
- verl/workers/actor/__init__.py
178
- verl/workers/actor/base.py
179
- verl/workers/actor/config.py
180
- verl/workers/actor/dp_actor.py
181
- verl/workers/critic/__init__.py
182
- verl/workers/critic/base.py
183
- verl/workers/critic/config.py
184
- verl/workers/critic/dp_critic.py
185
- verl/workers/reward/__init__.py
186
- verl/workers/reward/config.py
187
- verl/workers/reward/function.py
188
- verl/workers/rollout/__init__.py
189
- verl/workers/rollout/base.py
190
- verl/workers/rollout/config.py
191
- verl/workers/rollout/hf_rollout.py
192
- verl/workers/rollout/vllm_rollout_spmd.py
193
- verl/workers/sharding_manager/__init__.py
194
- verl/workers/sharding_manager/base.py
195
- verl/workers/sharding_manager/fsdp_ulysses.py
196
- verl/workers/sharding_manager/fsdp_vllm.py
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
code/orarl.egg-info/dependency_links.txt DELETED
@@ -1 +0,0 @@
1
-
 
 
code/orarl.egg-info/entry_points.txt DELETED
@@ -1,5 +0,0 @@
1
- [console_scripts]
2
- orarl-eval = orarl.cli.evaluate:main
3
- orarl-eval-data = orarl.cli.eval_data:main
4
- orarl-prepare = orarl.cli.prepare:main
5
- orarl-train = orarl.cli.train:main
 
 
 
 
 
 
code/orarl.egg-info/requires.txt DELETED
@@ -1,12 +0,0 @@
1
- numpy
2
- PyYAML
3
- torch
4
-
5
- [hf]
6
- huggingface_hub
7
-
8
- [lint]
9
- ruff
10
-
11
- [test]
12
- pytest
 
 
 
 
 
 
 
 
 
 
 
 
 
code/orarl.egg-info/top_level.txt DELETED
@@ -1,2 +0,0 @@
1
- orarl
2
- verl
 
 
 
code/runs/smoke-training-9b-20260825_174304/smoke.log DELETED
@@ -1,467 +0,0 @@
1
- OraRL root: /mnt/lyh/TempSamp/EasyVideoR1/OraRL-official
2
- Python: /data/miniconda3/envs/orarl/bin/python
3
- Model: /apdcephfs_gy7/share_305340263/hunyuan/yunhengli/checkpoint/Video-ORA-9B
4
- Train data: /mnt/lyh/TempSamp/EasyVideoR1/data/joint_final/rl_joint_100k_maskaware.jsonl
5
- Val data: /mnt/lyh/TempSamp/EasyVideoR1/data/joint_final/rl_phase3_49152.canary.jsonl
6
- Recipe scale: 9b
7
- World size: 1 x 8
8
- Steps per run: 1
9
- Results: /mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/runs/smoke-training-9b-20260825_174304
10
- Execute: 1
11
- 0, NVIDIA H20, 0 MiB, 97871 MiB
12
- 1, NVIDIA H20, 0 MiB, 97871 MiB
13
- 2, NVIDIA H20, 0 MiB, 97871 MiB
14
- 3, NVIDIA H20, 0 MiB, 97871 MiB
15
- 4, NVIDIA H20, 0 MiB, 97871 MiB
16
- 5, NVIDIA H20, 0 MiB, 97871 MiB
17
- 6, NVIDIA H20, 0 MiB, 97871 MiB
18
- 7, NVIDIA H20, 0 MiB, 97871 MiB
19
- /data/miniconda3/envs/orarl/lib/python3.11/site-packages/transformers/utils/import_utils.py:1006: FutureWarning: `is_flash_attn_greater_or_equal_2_10` is deprecated and will be removed in v5.8. Please use `is_flash_attn_greater_or_equal(library_version='2.1.0')` instead if needed.
20
- warnings.warn(
21
- dependency OK: numpy 2.2.6
22
- dependency OK: torch 2.10.0+cu129
23
- dependency OK: transformers 5.5.4
24
- dependency OK: vllm 0.19.1
25
- dependency OK: ray 2.54.0
26
- dependency OK: tensordict 0.12.2
27
- dependency OK: codetiming 1.4.0
28
- dependency OK: omegaconf 2.3.0
29
- dependency OK: verl.trainer.main /mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/verl/trainer/main.py
30
- dependency OK: orarl.rewards /mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/orarl/rewards/__init__.py
31
-
32
- === grpo smoke (1 update(s)) ===
33
- /data/miniconda3/envs/orarl/lib/python3.11/site-packages/transformers/utils/import_utils.py:1006: FutureWarning: `is_flash_attn_greater_or_equal_2_10` is deprecated and will be removed in v5.8. Please use `is_flash_attn_greater_or_equal(library_version='2.1.0')` instead if needed.
34
- warnings.warn(
35
- 2026-08-25 17:43:40,058 INFO worker.py:2004 -- Started a local Ray instance. View the dashboard at http://127.0.0.1:8265 
36
- /data/miniconda3/envs/orarl/lib/python3.11/site-packages/ray/_private/worker.py:2052: FutureWarning: Tip: In future versions of Ray, Ray will no longer override accelerator visible devices env var if num_gpus=0 or num_gpus=None (default). To enable this behavior and turn off this error message, set RAY_ACCEL_ENV_VAR_OVERRIDE_ON_ZERO=0
37
- warnings.warn(
38
- (pid=1944439) W0825 17:43:56.478000 1944439 site-packages/torch/utils/cpp_extension.py:118] No CUDA runtime is found, using CUDA_HOME='/data/miniconda3/envs/orarl'
39
- (Runner pid=1944439) {
40
- (Runner pid=1944439) "data": {
41
- (Runner pid=1944439) "train_files": "/mnt/lyh/TempSamp/EasyVideoR1/data/joint_final/rl_joint_100k_maskaware.jsonl",
42
- (Runner pid=1944439) "val_files": "/mnt/lyh/TempSamp/EasyVideoR1/data/joint_final/rl_phase3_49152.canary.jsonl",
43
- (Runner pid=1944439) "prompt_key": "problem",
44
- (Runner pid=1944439) "answer_key": "answer",
45
- (Runner pid=1944439) "image_key": "images",
46
- (Runner pid=1944439) "video_key": "videos",
47
- (Runner pid=1944439) "image_dir": "/",
48
- (Runner pid=1944439) "video_fps": 2.0,
49
- (Runner pid=1944439) "video_max_frames": 128,
50
- (Runner pid=1944439) "max_prompt_length": 24576,
51
- (Runner pid=1944439) "max_response_length": 4096,
52
- (Runner pid=1944439) "rollout_batch_size": 8,
53
- (Runner pid=1944439) "mini_rollout_batch_size": null,
54
- (Runner pid=1944439) "val_batch_size": 16,
55
- (Runner pid=1944439) "format_prompt": null,
56
- (Runner pid=1944439) "override_chat_template": null,
57
- (Runner pid=1944439) "enable_thinking": false,
58
- (Runner pid=1944439) "response_prefix": "",
59
- (Runner pid=1944439) "shuffle": true,
60
- (Runner pid=1944439) "seed": 42,
61
- (Runner pid=1944439) "min_pixels": 4096,
62
- (Runner pid=1944439) "max_pixels": 4194304,
63
- (Runner pid=1944439) "image_min_pixels": 4096,
64
- (Runner pid=1944439) "image_max_pixels": 1048576,
65
- (Runner pid=1944439) "video_min_pixels": 4096,
66
- (Runner pid=1944439) "video_max_pixels": 786432,
67
- (Runner pid=1944439) "video_total_pixels": 8388608,
68
- (Runner pid=1944439) "val_video_fps": 2.0,
69
- (Runner pid=1944439) "val_video_max_frames": 128,
70
- (Runner pid=1944439) "val_video_min_pixels": 4096,
71
- (Runner pid=1944439) "val_video_max_pixels": 786432,
72
- (Runner pid=1944439) "val_video_total_pixels": 8388608,
73
- (Runner pid=1944439) "filter_overlong_prompts": false,
74
- (Runner pid=1944439) "filter_overlong_prompts_workers": 16,
75
- (Runner pid=1944439) "use_preprocessed_videos": false,
76
- (Runner pid=1944439) "video_source_mode": "realtime_only",
77
- (Runner pid=1944439) "preprocessed_video_dir": null,
78
- (Runner pid=1944439) "val_preprocessed_video_dir": null,
79
- (Runner pid=1944439) "val_video_source_mode": "realtime_only",
80
- (Runner pid=1944439) "inline_video_tensors": true,
81
- (Runner pid=1944439) "group_by_task": true,
82
- (Runner pid=1944439) "group_by_task_key": "problem_type",
83
- (Runner pid=1944439) "dataloader_num_workers": 1
84
- (Runner pid=1944439) },
85
- (Runner pid=1944439) "worker": {
86
- (Runner pid=1944439) "hybrid_engine": true,
87
- (Runner pid=1944439) "actor": {
88
- (Runner pid=1944439) "strategy": "fsdp",
89
- (Runner pid=1944439) "global_batch_size": 8,
90
- (Runner pid=1944439) "micro_batch_size_per_device_for_update": 1,
91
- (Runner pid=1944439) "micro_batch_size_per_device_for_experience": 1,
92
- (Runner pid=1944439) "max_grad_norm": 1.0,
93
- (Runner pid=1944439) "clip_ratio_low": 0.2,
94
- (Runner pid=1944439) "clip_ratio_high": 0.2,
95
- (Runner pid=1944439) "clip_ratio_dual": 3.0,
96
- (Runner pid=1944439) "loss_avg_mode": "token",
97
- (Runner pid=1944439) "loss_type": "default",
98
- (Runner pid=1944439) "ppo_epochs": 1,
99
- (Runner pid=1944439) "padding_free": true,
100
- (Runner pid=1944439) "dynamic_batching": true,
101
- (Runner pid=1944439) "max_token_len_per_gpu": 32768,
102
- (Runner pid=1944439) "ulysses_size": 1,
103
- (Runner pid=1944439) "use_torch_compile": false,
104
- (Runner pid=1944439) "model": {
105
- (Runner pid=1944439) "model_path": "/apdcephfs_gy7/share_305340263/hunyuan/yunhengli/checkpoint/Video-ORA-9B",
106
- (Runner pid=1944439) "tokenizer_path": "/apdcephfs_gy7/share_305340263/hunyuan/yunhengli/checkpoint/Video-ORA-9B",
107
- (Runner pid=1944439) "override_config": {},
108
- (Runner pid=1944439) "enable_gradient_checkpointing": true,
109
- (Runner pid=1944439) "trust_remote_code": false,
110
- (Runner pid=1944439) "freeze_vision_tower": true,
111
- (Runner pid=1944439) "train_vision_merger": false
112
- (Runner pid=1944439) },
113
- (Runner pid=1944439) "optim": {
114
- (Runner pid=1944439) "lr": 1e-06,
115
- (Runner pid=1944439) "betas": [
116
- (Runner pid=1944439) 0.9,
117
- (Runner pid=1944439) 0.999
118
- (Runner pid=1944439) ],
119
- (Runner pid=1944439) "weight_decay": 0.0,
120
- (Runner pid=1944439) "strategy": "adamw",
121
- (Runner pid=1944439) "lr_warmup_ratio": 0.0,
122
- (Runner pid=1944439) "lr_warmup_steps": null,
123
- (Runner pid=1944439) "min_lr_ratio": null,
124
- (Runner pid=1944439) "lr_scheduler_type": "constant",
125
- (Runner pid=1944439) "training_steps": -1
126
- (Runner pid=1944439) },
127
- (Runner pid=1944439) "fsdp": {
128
- (Runner pid=1944439) "enable_full_shard": true,
129
- (Runner pid=1944439) "enable_cpu_offload": false,
130
- (Runner pid=1944439) "enable_rank0_init": true,
131
- (Runner pid=1944439) "use_orig_params": false,
132
- (Runner pid=1944439) "torch_dtype": "bf16",
133
- (Runner pid=1944439) "fsdp_size": -1,
134
- (Runner pid=1944439) "mp_param_dtype": "bf16",
135
- (Runner pid=1944439) "mp_reduce_dtype": "fp32",
136
- (Runner pid=1944439) "mp_buffer_dtype": "fp32"
137
- (Runner pid=1944439) },
138
- (Runner pid=1944439) "offload": {
139
- (Runner pid=1944439) "offload_params": false,
140
- (Runner pid=1944439) "offload_optimizer": false
141
- (Runner pid=1944439) },
142
- (Runner pid=1944439) "global_batch_size_per_device": -1,
143
- (Runner pid=1944439) "disable_kl": true,
144
- (Runner pid=1944439) "use_kl_loss": false,
145
- (Runner pid=1944439) "kl_penalty": "low_var_kl",
146
- (Runner pid=1944439) "kl_coef": 0.0
147
- (Runner pid=1944439) },
148
- (Runner pid=1944439) "critic": {
149
- (Runner pid=1944439) "strategy": "fsdp",
150
- (Runner pid=1944439) "global_batch_size": 256,
151
- (Runner pid=1944439) "micro_batch_size_per_device_for_update": 4,
152
- (Runner pid=1944439) "micro_batch_size_per_device_for_experience": 16,
153
- (Runner pid=1944439) "max_grad_norm": 1.0,
154
- (Runner pid=1944439) "cliprange_value": 0.5,
155
- (Runner pid=1944439) "loss_avg_mode": "token",
156
- (Runner pid=1944439) "ppo_epochs": 1,
157
- (Runner pid=1944439) "padding_free": false,
158
- (Runner pid=1944439) "dynamic_batching": true,
159
- (Runner pid=1944439) "max_token_len_per_gpu": null,
160
- (Runner pid=1944439) "ulysses_size": 1,
161
- (Runner pid=1944439) "model": {
162
- (Runner pid=1944439) "model_path": null,
163
- (Runner pid=1944439) "tokenizer_path": null,
164
- (Runner pid=1944439) "override_config": {},
165
- (Runner pid=1944439) "enable_gradient_checkpointing": true,
166
- (Runner pid=1944439) "trust_remote_code": true,
167
- (Runner pid=1944439) "freeze_vision_tower": false,
168
- (Runner pid=1944439) "train_vision_merger": false
169
- (Runner pid=1944439) },
170
- (Runner pid=1944439) "optim": {
171
- (Runner pid=1944439) "lr": 1e-06,
172
- (Runner pid=1944439) "betas": [
173
- (Runner pid=1944439) 0.9,(Runner pid=1944439) The `use_fast` parameter is deprecated and will be removed in a future version. Use `backend="torchvision"` instead of `use_fast=True`, or `backend="pil"` instead of `use_fast=False`.
174
- (Runner pid=1944439)
175
- (Runner pid=1944439)
176
- (Runner pid=1944439)
177
- (Runner pid=1944439)
178
- (Runner pid=1944439)
179
- (Runner pid=1944439)
180
- (Runner pid=1944439)
181
- (Runner pid=1944439)
182
- (Runner pid=1944439)
183
- (Runner pid=1944439)
184
- (Runner pid=1944439)
185
- (Runner pid=1944439)
186
- (Runner pid=1944439)
187
- (Runner pid=1944439)
188
- (Runner pid=1944439)
189
- (Runner pid=1944439)
190
- (Runner pid=1944439)
191
- (Runner pid=1944439)
192
- (Runner pid=1944439)
193
- (Runner pid=1944439)
194
- (Runner pid=1944439)
195
- (Runner pid=1944439)
196
- (Runner pid=1944439)
197
- (Runner pid=1944439)
198
- (Runner pid=1944439)
199
- (Runner pid=1944439)
200
- (Runner pid=1944439)
201
- (Runner pid=1944439)
202
- (Runner pid=1944439)
203
- (Runner pid=1944439)
204
- (Runner pid=1944439)
205
- (Runner pid=1944439)
206
- Traceback (most recent call last):
207
- File "<frozen runpy>", line 198, in _run_module_as_main
208
- File "<frozen runpy>", line 88, in _run_code
209
- File "/mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/verl/trainer/main.py", line 137, in <module>
210
- main()
211
- File "/mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/verl/trainer/main.py", line 129, in main
212
- ray.get(runner.run.remote(ppo_config))
213
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/ray/_private/auto_init_hook.py", line 22, in auto_init_wrapper
214
- return fn(*args, **kwargs)
215
- ^^^^^^^^^^^^^^^^^^^
216
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/ray/_private/client_mode_hook.py", line 104, in wrapper
217
- return func(*args, **kwargs)
218
- ^^^^^^^^^^^^^^^^^^^^^
219
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/ray/_private/worker.py", line 2981, in get
220
- values, debugger_breakpoint = worker.get_objects(
221
- ^^^^^^^^^^^^^^^^^^^
222
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/ray/_private/worker.py", line 1012, in get_objects
223
- raise value.as_instanceof_cause()
224
- ray.exceptions.RayTaskError(DatasetGenerationError): ray::Runner.run() (pid=1944439, ip=29.127.69.4, actor_id=e981ae3ef4ecbc78705cfeea01000000, repr=<main.Runner object at 0x7f3a507bda10>)
225
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/datasets/packaged_modules/json/json.py", line 265, in _generate_tables
226
- self._cast_table(pa_table, json_field_paths=json_field_paths),
227
- ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
228
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/datasets/packaged_modules/json/json.py", line 120, in _cast_table
229
- pa_table = table_cast(pa_table, self.info.features.arrow_schema)
230
- ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
231
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/datasets/table.py", line 2272, in table_cast
232
- return cast_table_to_schema(table, schema)
233
- ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
234
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/datasets/table.py", line 2218, in cast_table_to_schema
235
- raise CastError(
236
- datasets.table.CastError: Couldn't cast
237
- problem: string
238
- answer: string
239
- ground_truth: string
240
- problem_type: string
241
- videos: list<item: string>
242
- child 0, item: string
243
- data_type: string
244
- task_source: string
245
- segmentation_output: string
246
- meta: string
247
- images: list<item: string>
248
- child 0, item: string
249
- fps: int64
250
- video_second: double
251
- resolution: struct<width: int64, height: int64>
252
- child 0, width: int64
253
- child 1, height: int64
254
- path: string
255
- data_source: string
256
- messages: list<item: struct<role: string, content: string>>
257
- child 0, item: struct<role: string, content: string>
258
- child 0, role: string
259
- child 1, content: string
260
- solution: string
261
- difficulty: double
262
- pred_iou: double
263
- pred_span: list<item: double>
264
- child 0, item: double
265
- pred_text: string
266
- checkpoint_score: double
267
- hardness_band: string
268
- selection_role: string
269
- selection_checkpoint: string
270
- identity_key: string
271
- media_anchors: list<item: string>
272
- child 0, item: string
273
- selection_temporal_bucket: string
274
- question_type: string
275
- source_file: string
276
- source_id: string
277
- selection_weak_buckets: list<item: string>
278
- child 0, item: string
279
- selection_bucket: string
280
- source_block: string
281
- selection_scores: struct<spatial_relation: int64, object_relation: int64, ego_exo_match: int64, relative_direction: in (... 100 chars omitted)
282
- child 0, spatial_relation: int64
283
- child 1, object_relation: int64
284
- child 2, ego_exo_match: int64
285
- child 3, relative_direction: int64
286
- child 4, multi_step_scene: int64
287
- child 5, cardinal_scene: int64
288
- child 6, temporal_order: int64
289
- child 7, viewpoint_motion: int64
290
- has_offline_trajectory: bool
291
- offline_output: string
292
- bbox_area: double
293
- problem_id: int64
294
- question_type_orig: string
295
- unit_orig: string
296
- unit_is_eval: bool
297
- scene_name: string
298
- preprocessed_video: string
299
- to
300
- {'problem': Value('string'), 'answer': Value('string'), 'ground_truth': Value('string'), 'problem_type': Value('string'), 'data_type': Value('string'), 'task_source': Value('string'), 'videos': List(Value('string')), 'images': List(Value('string')), 'preprocessed_video': Value('string'), 'segmentation_output': Json(decode=True), 'meta': Json(decode=True), 'fps': Value('int64'), 'video_second': Value('float64'), 'resolution': {'width': Value('int64'), 'height': Value('int64')}, 'path': Value('string'), 'data_source': Value('string')}
301
- because column names don't match
302
-
303
- The above exception was the direct cause of the following exception:
304
-
305
- ray::Runner.run() (pid=1944439, ip=29.127.69.4, actor_id=e981ae3ef4ecbc78705cfeea01000000, repr=<main.Runner object at 0x7f3a507bda10>)
306
- ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
307
- ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
308
- File "/mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/verl/trainer/main.py", line 74, in run
309
- train_dataloader, val_dataloader = create_dataloader(
310
- ^^^^^^^^^^^^^^^^^^
311
- File "/mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/verl/trainer/data_loader.py", line 41, in create_dataloader
312
- train_dataset = RLHFDataset(
313
- ^^^^^^^^^^^^
314
- File "/mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/verl/utils/dataset.py", line 444, in __init__
315
- self.dataset = load_dataset(file_type, data_files=data_path, split=data_split)
316
- ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
317
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/datasets/load.py", line 1708, in load_dataset
318
- builder_instance.download_and_prepare(
319
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/datasets/builder.py", line 884, in download_and_prepare
320
- self._download_and_prepare(
321
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/datasets/builder.py", line 947, in _download_and_prepare
322
- self._prepare_split(split_generator, **prepare_split_kwargs)
323
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/datasets/builder.py", line 1739, in _prepare_split
324
- for job_id, done, content in self._prepare_split_single(
325
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/datasets/builder.py", line 1922, in _prepare_split_single
326
- raise DatasetGenerationError("An error occurred while generating the dataset") from e
327
- datasets.exceptions.DatasetGenerationError: An error occurred while generating the dataset
328
-
329
- (Runner pid=1944439) 0.999
330
- (Runner pid=1944439) ],
331
- (Runner pid=1944439) "weight_decay": 0.01,
332
- (Runner pid=1944439) "strategy": "adamw",
333
- (Runner pid=1944439) "lr_warmup_ratio": 0.0,
334
- (Runner pid=1944439) "lr_warmup_steps": null,
335
- (Runner pid=1944439) "min_lr_ratio": null,
336
- (Runner pid=1944439) "lr_scheduler_type": "constant",
337
- (Runner pid=1944439) "training_steps": -1
338
- (Runner pid=1944439) },
339
- (Runner pid=1944439) "fsdp": {
340
- (Runner pid=1944439) "enable_full_shard": true,
341
- (Runner pid=1944439) "enable_cpu_offload": false,
342
- (Runner pid=1944439) "enable_rank0_init": true,
343
- (Runner pid=1944439) "use_orig_params": false,
344
- (Runner pid=1944439) "torch_dtype": null,
345
- (Runner pid=1944439) "fsdp_size": -1,
346
- (Runner pid=1944439) "mp_param_dtype": "bf16",
347
- (Runner pid=1944439) "mp_reduce_dtype": "fp32",
348
- (Runner pid=1944439) "mp_buffer_dtype": "fp32"
349
- (Runner pid=1944439) },
350
- (Runner pid=1944439) "offload": {
351
- (Runner pid=1944439) "offload_params": false,
352
- (Runner pid=1944439) "offload_optimizer": false
353
- (Runner pid=1944439) },
354
- (Runner pid=1944439) "global_batch_size_per_device": -1
355
- (Runner pid=1944439) },
356
- (Runner pid=1944439) "ref": {
357
- (Runner pid=1944439) "strategy": "fsdp",
358
- (Runner pid=1944439) "fsdp": {
359
- (Runner pid=1944439) "enable_full_shard": true,
360
- (Runner pid=1944439) "enable_cpu_offload": false,
361
- (Runner pid=1944439) "enable_rank0_init": true,
362
- (Runner pid=1944439) "use_orig_params": false,
363
- (Runner pid=1944439) "torch_dtype": "bf16",
364
- (Runner pid=1944439) "fsdp_size": -1,
365
- (Runner pid=1944439) "mp_param_dtype": "bf16",
366
- (Runner pid=1944439) "mp_reduce_dtype": "fp32",
367
- (Runner pid=1944439) "mp_buffer_dtype": "fp32"
368
- (Runner pid=1944439) },
369
- (Runner pid=1944439) "offload": {
370
- (Runner pid=1944439) "offload_params": false,
371
- (Runner pid=1944439) "offload_optimizer": false
372
- (Runner pid=1944439) },
373
- (Runner pid=1944439) "micro_batch_size_per_device_for_experience": 1,
374
- (Runner pid=1944439) "padding_free": true,
375
- (Runner pid=1944439) "dynamic_batching": true,
376
- (Runner pid=1944439) "max_token_len_per_gpu": 32768,
377
- (Runner pid=1944439) "ulysses_size": 1,
378
- (Runner pid=1944439) "use_torch_compile": false
379
- (Runner pid=1944439) },
380
- (Runner pid=1944439) "reward": {
381
- (Runner pid=1944439) "reward_function": "orarl.rewards",
382
- (Runner pid=1944439) "reward_function_kwargs": {},
383
- (Runner pid=1944439) "skip_special_tokens": true,
384
- (Runner pid=1944439) "num_cpus": 1,
385
- (Runner pid=1944439) "reward_function_name": "compute_score",
386
- (Runner pid=1944439) "reward_function_is_module": true
387
- (Runner pid=1944439) },
388
- (Runner pid=1944439) "rollout": {
389
- (Runner pid=1944439) "name": "vllm",
390
- (Runner pid=1944439) "n": 8,
391
- (Runner pid=1944439) "temperature": 1.0,
392
- (Runner pid=1944439) "top_p": 0.85,
393
- (Runner pid=1944439) "top_k": -1,
394
- (Runner pid=1944439) "seed": 42,
395
- (Runner pid=1944439) "limit_images": 10,
396
- (Runner pid=1944439) "dtype": "bf16",
397
- (Runner pid=1944439) "gpu_memory_utilization": 0.45,
398
- (Runner pid=1944439) "ignore_eos": false,
399
- (Runner pid=1944439) "enforce_eager": false,
400
- (Runner pid=1944439) "enable_chunked_prefill": true,
401
- (Runner pid=1944439) "tensor_parallel_size": 1,
402
- (Runner pid=1944439) "max_model_len": null,
403
- (Runner pid=1944439) "max_num_batched_tokens": 32768,
404
- (Runner pid=1944439) "disable_log_stats": true,
405
- (Runner pid=1944439) "disable_tqdm": true,
406
- (Runner pid=1944439) "val_override_config": {
407
- (Runner pid=1944439) "temperature": 0.7,
408
- (Runner pid=1944439) "top_p": 0.95,
409
- (Runner pid=1944439) "n": 1
410
- (Runner pid=1944439) },
411
- (Runner pid=1944439) "kv_cache_dtype": "auto",
412
- (Runner pid=1944439) "calculate_log_probs": false,
413
- (Runner pid=1944439) "collect_seq_logprob_for_filter": false,
414
- (Runner pid=1944439) "prompt_length": 24576,
415
- (Runner pid=1944439) "response_length": 4096,
416
- (Runner pid=1944439) "trust_remote_code": false
417
- (Runner pid=1944439) }
418
- (Runner pid=1944439) },
419
- (Runner pid=1944439) "algorithm": {
420
- (Runner pid=1944439) "name": "grpo",
421
- (Runner pid=1944439) "gamma": 1.0,
422
- (Runner pid=1944439) "lam": 1.0,
423
- (Runner pid=1944439) "adv_estimator": "grpo",
424
- (Runner pid=1944439) "scale_rewards": true,
425
- (Runner pid=1944439) "disable_kl": true,
426
- (Runner pid=1944439) "use_kl_loss": false,
427
- (Runner pid=1944439) "kl_penalty": "low_var_kl",
428
- (Runner pid=1944439) "kl_coef": 0.0,
429
- (Runner pid=1944439) "kl_type": "fixed",
430
- (Runner pid=1944439) "kl_horizon": 10000.0,
431
- (Runner pid=1944439) "kl_target": 0.1
432
- (Runner pid=1944439) },
433
- (Runner pid=1944439) "trainer": {
434
- (Runner pid=1944439) "total_epochs": 1,
435
- (Runner pid=1944439) "max_steps": 1,
436
- (Runner pid=1944439) "project_name": "OraRL",
437
- (Runner pid=1944439) "experiment_name": "smoke-grpo-9b",
438
- (Runner pid=1944439) "logger": [
439
- (Runner pid=1944439) "console"
440
- (Runner pid=1944439) ],
441
- (Runner pid=1944439) "nnodes": 1,
442
- (Runner pid=1944439) "n_gpus_per_node": 8,
443
- (Runner pid=1944439) "critic_warmup": 0,
444
- (Runner pid=1944439) "val_freq": -1,
445
- (Runner pid=1944439) "val_before_train": false,
446
- (Runner pid=1944439) "val_only": false,
447
- (Runner pid=1944439) "val_generations_to_log": 0,
448
- (Runner pid=1944439) "save_freq": 1,
449
- (Runner pid=1944439) "save_limit": 3,
450
- (Runner pid=1944439) "save_model_only": false,
451
- (Runner pid=1944439) "keep_optim_only_latest": true,
452
- (Runner pid=1944439) "save_checkpoint_path": "/mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/runs/smoke-training-9b-20260825_174304/grpo",
453
- (Runner pid=1944439) "load_checkpoint_path": null,
454
- (Runner pid=1944439) "ray_timeline": null,
455
- (Runner pid=1944439) "find_last_checkpoint": false,
456
- (Runner pid=1944439) "keep_best_train_ckpt": false,
457
- (Runner pid=1944439) "best_train_metric_key": "reward/overall",
458
- (Runner pid=1944439) "best_train_smooth_window": 5,
459
- (Runner pid=1944439) "best_train_min_step": 10
460
- (Runner pid=1944439) }
461
- (Runner pid=1944439) }
462
- (AutoRewardManager pid=1946036) Using reward function `compute_score` from `orarl.rewards`.
463
- (AutoRewardManager pid=1946036) Reward name: orarl, reward type: batch.
464
- (AutoRewardManager pid=1946038) Using reward function `compute_score` from `orarl.rewards`.
465
- (AutoRewardManager pid=1946038) Reward name: orarl, reward type: batch.
466
- method=grpo mode=run
467
- /data/miniconda3/envs/orarl/bin/python -m verl.trainer.main config=/mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/configs/grpo_9b.yaml trainer.max_steps=1 trainer.save_freq=1 trainer.val_before_train=false trainer.experiment_name=smoke-grpo-9b data.rollout_batch_size=8 worker.actor.global_batch_size=8 worker.actor.model.model_path=/apdcephfs_gy7/share_305340263/hunyuan/yunhengli/checkpoint/Video-ORA-9B worker.actor.model.tokenizer_path=/apdcephfs_gy7/share_305340263/hunyuan/yunhengli/checkpoint/Video-ORA-9B data.train_files=/mnt/lyh/TempSamp/EasyVideoR1/data/joint_final/rl_joint_100k_maskaware.jsonl data.val_files=/mnt/lyh/TempSamp/EasyVideoR1/data/joint_final/rl_phase3_49152.canary.jsonl trainer.save_checkpoint_path=/mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/runs/smoke-training-9b-20260825_174304/grpo trainer.nnodes=1 trainer.n_gpus_per_node=8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
code/runs/smoke-training-9b-20260825_175213/smoke.log DELETED
@@ -1,827 +0,0 @@
1
- OraRL root: /mnt/lyh/TempSamp/EasyVideoR1/OraRL-official
2
- Python: /data/miniconda3/envs/orarl/bin/python
3
- Model: /apdcephfs_gy7/share_305340263/hunyuan/yunhengli/checkpoint/Video-ORA-9B
4
- Train data: /mnt/lyh/TempSamp/EasyVideoR1/data/joint_final/rl_joint_100k_maskaware.jsonl
5
- Val data: /mnt/lyh/TempSamp/EasyVideoR1/data/joint_final/rl_phase3_49152.canary.jsonl
6
- Recipe scale: 9b
7
- World size: 1 x 8
8
- Steps per run: 1
9
- Results: /mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/runs/smoke-training-9b-20260825_175213
10
- Execute: 1
11
- 0, NVIDIA H20, 0 MiB, 97871 MiB
12
- 1, NVIDIA H20, 0 MiB, 97871 MiB
13
- 2, NVIDIA H20, 0 MiB, 97871 MiB
14
- 3, NVIDIA H20, 0 MiB, 97871 MiB
15
- 4, NVIDIA H20, 0 MiB, 97871 MiB
16
- 5, NVIDIA H20, 0 MiB, 97871 MiB
17
- 6, NVIDIA H20, 0 MiB, 97871 MiB
18
- 7, NVIDIA H20, 0 MiB, 97871 MiB
19
- /data/miniconda3/envs/orarl/lib/python3.11/site-packages/transformers/utils/import_utils.py:1006: FutureWarning: `is_flash_attn_greater_or_equal_2_10` is deprecated and will be removed in v5.8. Please use `is_flash_attn_greater_or_equal(library_version='2.1.0')` instead if needed.
20
- warnings.warn(
21
- dependency OK: numpy 2.2.6
22
- dependency OK: torch 2.10.0+cu129
23
- dependency OK: transformers 5.5.4
24
- dependency OK: vllm 0.19.1
25
- dependency OK: ray 2.54.0
26
- dependency OK: tensordict 0.12.2
27
- dependency OK: codetiming 1.4.0
28
- dependency OK: omegaconf 2.3.0
29
- dependency OK: verl.trainer.main /mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/verl/trainer/main.py
30
- dependency OK: orarl.rewards /mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/orarl/rewards/__init__.py
31
-
32
- === grpo smoke (1 update(s)) ===
33
- /data/miniconda3/envs/orarl/lib/python3.11/site-packages/transformers/utils/import_utils.py:1006: FutureWarning: `is_flash_attn_greater_or_equal_2_10` is deprecated and will be removed in v5.8. Please use `is_flash_attn_greater_or_equal(library_version='2.1.0')` instead if needed.
34
- warnings.warn(
35
- 2026-08-25 17:52:41,284 INFO worker.py:2004 -- Started a local Ray instance. View the dashboard at http://127.0.0.1:8265 
36
- /data/miniconda3/envs/orarl/lib/python3.11/site-packages/ray/_private/worker.py:2052: FutureWarning: Tip: In future versions of Ray, Ray will no longer override accelerator visible devices env var if num_gpus=0 or num_gpus=None (default). To enable this behavior and turn off this error message, set RAY_ACCEL_ENV_VAR_OVERRIDE_ON_ZERO=0
37
- warnings.warn(
38
- (pid=1984436) W0825 17:52:57.110000 1984436 site-packages/torch/utils/cpp_extension.py:118] No CUDA runtime is found, using CUDA_HOME='/data/miniconda3/envs/orarl'
39
- (Runner pid=1984436) {
40
- (Runner pid=1984436) "data": {
41
- (Runner pid=1984436) "train_files": "/mnt/lyh/TempSamp/EasyVideoR1/data/joint_final/rl_joint_100k_maskaware.jsonl",
42
- (Runner pid=1984436) "val_files": "/mnt/lyh/TempSamp/EasyVideoR1/data/joint_final/rl_phase3_49152.canary.jsonl",
43
- (Runner pid=1984436) "prompt_key": "problem",
44
- (Runner pid=1984436) "answer_key": "answer",
45
- (Runner pid=1984436) "image_key": "images",
46
- (Runner pid=1984436) "video_key": "videos",
47
- (Runner pid=1984436) "image_dir": "/",
48
- (Runner pid=1984436) "video_fps": 2.0,
49
- (Runner pid=1984436) "video_max_frames": 128,
50
- (Runner pid=1984436) "max_prompt_length": 24576,
51
- (Runner pid=1984436) "max_response_length": 4096,
52
- (Runner pid=1984436) "rollout_batch_size": 8,
53
- (Runner pid=1984436) "mini_rollout_batch_size": null,
54
- (Runner pid=1984436) "val_batch_size": 16,
55
- (Runner pid=1984436) "format_prompt": null,
56
- (Runner pid=1984436) "override_chat_template": null,
57
- (Runner pid=1984436) "enable_thinking": false,
58
- (Runner pid=1984436) "response_prefix": "",
59
- (Runner pid=1984436) "shuffle": true,
60
- (Runner pid=1984436) "seed": 42,
61
- (Runner pid=1984436) "min_pixels": 4096,
62
- (Runner pid=1984436) "max_pixels": 4194304,
63
- (Runner pid=1984436) "image_min_pixels": 4096,
64
- (Runner pid=1984436) "image_max_pixels": 1048576,
65
- (Runner pid=1984436) "video_min_pixels": 4096,
66
- (Runner pid=1984436) "video_max_pixels": 786432,
67
- (Runner pid=1984436) "video_total_pixels": 8388608,
68
- (Runner pid=1984436) "val_video_fps": 2.0,
69
- (Runner pid=1984436) "val_video_max_frames": 128,
70
- (Runner pid=1984436) "val_video_min_pixels": 4096,
71
- (Runner pid=1984436) "val_video_max_pixels": 786432,
72
- (Runner pid=1984436) "val_video_total_pixels": 8388608,
73
- (Runner pid=1984436) "filter_overlong_prompts": false,
74
- (Runner pid=1984436) "filter_overlong_prompts_workers": 16,
75
- (Runner pid=1984436) "use_preprocessed_videos": false,
76
- (Runner pid=1984436) "video_source_mode": "realtime_only",
77
- (Runner pid=1984436) "preprocessed_video_dir": null,
78
- (Runner pid=1984436) "val_preprocessed_video_dir": null,
79
- (Runner pid=1984436) "val_video_source_mode": "realtime_only",
80
- (Runner pid=1984436) "inline_video_tensors": true,
81
- (Runner pid=1984436) "group_by_task": true,
82
- (Runner pid=1984436) "group_by_task_key": "problem_type",
83
- (Runner pid=1984436) "dataloader_num_workers": 1
84
- (Runner pid=1984436) },
85
- (Runner pid=1984436) "worker": {
86
- (Runner pid=1984436) "hybrid_engine": true,
87
- (Runner pid=1984436) "actor": {
88
- (Runner pid=1984436) "strategy": "fsdp",
89
- (Runner pid=1984436) "global_batch_size": 8,
90
- (Runner pid=1984436) "micro_batch_size_per_device_for_update": 1,
91
- (Runner pid=1984436) "micro_batch_size_per_device_for_experience": 1,
92
- (Runner pid=1984436) "max_grad_norm": 1.0,
93
- (Runner pid=1984436) "clip_ratio_low": 0.2,
94
- (Runner pid=1984436) "clip_ratio_high": 0.2,
95
- (Runner pid=1984436) "clip_ratio_dual": 3.0,
96
- (Runner pid=1984436) "loss_avg_mode": "token",
97
- (Runner pid=1984436) "loss_type": "default",
98
- (Runner pid=1984436) "ppo_epochs": 1,
99
- (Runner pid=1984436) "padding_free": true,
100
- (Runner pid=1984436) "dynamic_batching": true,
101
- (Runner pid=1984436) "max_token_len_per_gpu": 32768,
102
- (Runner pid=1984436) "ulysses_size": 1,
103
- (Runner pid=1984436) "use_torch_compile": false,
104
- (Runner pid=1984436) "model": {
105
- (Runner pid=1984436) "model_path": "/apdcephfs_gy7/share_305340263/hunyuan/yunhengli/checkpoint/Video-ORA-9B",
106
- (Runner pid=1984436) "tokenizer_path": "/apdcephfs_gy7/share_305340263/hunyuan/yunhengli/checkpoint/Video-ORA-9B",
107
- (Runner pid=1984436) "override_config": {},
108
- (Runner pid=1984436) "enable_gradient_checkpointing": true,
109
- (Runner pid=1984436) "trust_remote_code": false,
110
- (Runner pid=1984436) "freeze_vision_tower": true,
111
- (Runner pid=1984436) "train_vision_merger": false
112
- (Runner pid=1984436) },
113
- (Runner pid=1984436) "optim": {
114
- (Runner pid=1984436) "lr": 1e-06,
115
- (Runner pid=1984436) "betas": [
116
- (Runner pid=1984436) 0.9,
117
- (Runner pid=1984436) 0.999
118
- (Runner pid=1984436) ],
119
- (Runner pid=1984436) "weight_decay": 0.0,
120
- (Runner pid=1984436) "strategy": "adamw",
121
- (Runner pid=1984436) "lr_warmup_ratio": 0.0,
122
- (Runner pid=1984436) "lr_warmup_steps": null,
123
- (Runner pid=1984436) "min_lr_ratio": null,
124
- (Runner pid=1984436) "lr_scheduler_type": "constant",
125
- (Runner pid=1984436) "training_steps": -1
126
- (Runner pid=1984436) },
127
- (Runner pid=1984436) "fsdp": {
128
- (Runner pid=1984436) "enable_full_shard": true,
129
- (Runner pid=1984436) "enable_cpu_offload": false,
130
- (Runner pid=1984436) "enable_rank0_init": true,
131
- (Runner pid=1984436) "use_orig_params": false,
132
- (Runner pid=1984436) "torch_dtype": "bf16",
133
- (Runner pid=1984436) "fsdp_size": -1,
134
- (Runner pid=1984436) "mp_param_dtype": "bf16",
135
- (Runner pid=1984436) "mp_reduce_dtype": "fp32",
136
- (Runner pid=1984436) "mp_buffer_dtype": "fp32"
137
- (Runner pid=1984436) },
138
- (Runner pid=1984436) "offload": {
139
- (Runner pid=1984436) "offload_params": false,
140
- (Runner pid=1984436) "offload_optimizer": false
141
- (Runner pid=1984436) },
142
- (Runner pid=1984436) "global_batch_size_per_device": -1,
143
- (Runner pid=1984436) "disable_kl": true,
144
- (Runner pid=1984436) "use_kl_loss": false,
145
- (Runner pid=1984436) "kl_penalty": "low_var_kl",
146
- (Runner pid=1984436) "kl_coef": 0.0
147
- (Runner pid=1984436) },
148
- (Runner pid=1984436) "critic": {
149
- (Runner pid=1984436) "strategy": "fsdp",
150
- (Runner pid=1984436) "global_batch_size": 256,
151
- (Runner pid=1984436) "micro_batch_size_per_device_for_update": 4,
152
- (Runner pid=1984436) "micro_batch_size_per_device_for_experience": 16,
153
- (Runner pid=1984436) "max_grad_norm": 1.0,
154
- (Runner pid=1984436) "cliprange_value": 0.5,
155
- (Runner pid=1984436) "loss_avg_mode": "token",
156
- (Runner pid=1984436) "ppo_epochs": 1,
157
- (Runner pid=1984436) "padding_free": false,
158
- (Runner pid=1984436) "dynamic_batching": true,
159
- (Runner pid=1984436) "max_token_len_per_gpu": null,
160
- (Runner pid=1984436) "ulysses_size": 1,
161
- (Runner pid=1984436) "model": {
162
- (Runner pid=1984436) "model_path": null,
163
- (Runner pid=1984436) "tokenizer_path": null,
164
- (Runner pid=1984436) "override_config": {},
165
- (Runner pid=1984436) "enable_gradient_checkpointing": true,
166
- (Runner pid=1984436) "trust_remote_code": true,
167
- (Runner pid=1984436) "freeze_vision_tower": false,
168
- (Runner pid=1984436) "train_vision_merger": false
169
- (Runner pid=1984436) },
170
- (Runner pid=1984436) "optim": {
171
- (Runner pid=1984436) "lr": 1e-06,
172
- (Runner pid=1984436) "betas": [
173
- (Runner pid=1984436) 0.9,(Runner pid=1984436) The `use_fast` parameter is deprecated and will be removed in a future version. Use `backend="torchvision"` instead of `use_fast=True`, or `backend="pil"` instead of `use_fast=False`.
174
-
175
- (Runner pid=1984436) 0.999
176
- (Runner pid=1984436) ],
177
- (Runner pid=1984436) "weight_decay": 0.01,
178
- (Runner pid=1984436) "strategy": "adamw",
179
- (Runner pid=1984436) "lr_warmup_ratio": 0.0,
180
- (Runner pid=1984436) "lr_warmup_steps": null,
181
- (Runner pid=1984436) "min_lr_ratio": null,
182
- (Runner pid=1984436) "lr_scheduler_type": "constant",
183
- (Runner pid=1984436) "training_steps": -1
184
- (Runner pid=1984436) },
185
- (Runner pid=1984436) "fsdp": {
186
- (Runner pid=1984436) "enable_full_shard": true,
187
- (Runner pid=1984436) "enable_cpu_offload": false,
188
- (Runner pid=1984436) "enable_rank0_init": true,
189
- (Runner pid=1984436) "use_orig_params": false,
190
- (Runner pid=1984436) "torch_dtype": null,
191
- (Runner pid=1984436) "fsdp_size": -1,
192
- (Runner pid=1984436) "mp_param_dtype": "bf16",
193
- (Runner pid=1984436) "mp_reduce_dtype": "fp32",
194
- (Runner pid=1984436) "mp_buffer_dtype": "fp32"
195
- (Runner pid=1984436) },
196
- (Runner pid=1984436) "offload": {
197
- (Runner pid=1984436) "offload_params": false,
198
- (Runner pid=1984436) "offload_optimizer": false
199
- (Runner pid=1984436) },
200
- (Runner pid=1984436) "global_batch_size_per_device": -1
201
- (Runner pid=1984436) },
202
- (Runner pid=1984436) "ref": {
203
- (Runner pid=1984436) "strategy": "fsdp",
204
- (Runner pid=1984436) "fsdp": {
205
- (Runner pid=1984436) "enable_full_shard": true,
206
- (Runner pid=1984436) "enable_cpu_offload": false,
207
- (Runner pid=1984436) "enable_rank0_init": true,
208
- (Runner pid=1984436) "use_orig_params": false,
209
- (Runner pid=1984436) "torch_dtype": "bf16",
210
- (Runner pid=1984436) "fsdp_size": -1,
211
- (Runner pid=1984436) "mp_param_dtype": "bf16",
212
- (Runner pid=1984436) "mp_reduce_dtype": "fp32",
213
- (Runner pid=1984436) "mp_buffer_dtype": "fp32"
214
- (Runner pid=1984436) },
215
- (Runner pid=1984436) "offload": {
216
- (Runner pid=1984436) "offload_params": false,
217
- (Runner pid=1984436) "offload_optimizer": false
218
- (Runner pid=1984436) },
219
- (Runner pid=1984436) "micro_batch_size_per_device_for_experience": 1,
220
- (Runner pid=1984436) "padding_free": true,
221
- (Runner pid=1984436) "dynamic_batching": true,
222
- (Runner pid=1984436) "max_token_len_per_gpu": 32768,
223
- (Runner pid=1984436) "ulysses_size": 1,
224
- (Runner pid=1984436) "use_torch_compile": false
225
- (Runner pid=1984436) },
226
- (Runner pid=1984436) "reward": {
227
- (Runner pid=1984436) "reward_function": "orarl.rewards",
228
- (Runner pid=1984436) "reward_function_kwargs": {},
229
- (Runner pid=1984436) "skip_special_tokens": true,
230
- (Runner pid=1984436) "num_cpus": 1,
231
- (Runner pid=1984436) "reward_function_name": "compute_score",
232
- (Runner pid=1984436) "reward_function_is_module": true
233
- (Runner pid=1984436) },
234
- (Runner pid=1984436) "rollout": {
235
- (Runner pid=1984436) "name": "vllm",
236
- (Runner pid=1984436) "n": 8,
237
- (Runner pid=1984436) "temperature": 1.0,
238
- (Runner pid=1984436) "top_p": 0.85,
239
- (Runner pid=1984436) "top_k": -1,
240
- (Runner pid=1984436) "seed": 42,
241
- (Runner pid=1984436) "limit_images": 10,
242
- (Runner pid=1984436) "dtype": "bf16",
243
- (Runner pid=1984436) "gpu_memory_utilization": 0.45,
244
- (Runner pid=1984436) "ignore_eos": false,
245
- (Runner pid=1984436) "enforce_eager": false,
246
- (Runner pid=1984436) "enable_chunked_prefill": true,
247
- (Runner pid=1984436) "tensor_parallel_size": 1,
248
- (Runner pid=1984436) "max_model_len": null,
249
- (Runner pid=1984436) "max_num_batched_tokens": 32768,
250
- (Runner pid=1984436) "disable_log_stats": true,
251
- (Runner pid=1984436) "disable_tqdm": true,
252
- (Runner pid=1984436) "val_override_config": {
253
- (Runner pid=1984436) "temperature": 0.7,
254
- (Runner pid=1984436) "top_p": 0.95,
255
- (Runner pid=1984436) "n": 1
256
- (Runner pid=1984436) },
257
- (Runner pid=1984436) "kv_cache_dtype": "auto",
258
- (Runner pid=1984436) "calculate_log_probs": false,
259
- (Runner pid=1984436) "collect_seq_logprob_for_filter": false,
260
- (Runner pid=1984436) "prompt_length": 24576,
261
- (Runner pid=1984436) "response_length": 4096,
262
- (Runner pid=1984436) "trust_remote_code": false
263
- (Runner pid=1984436) }
264
- (Runner pid=1984436) },
265
- (Runner pid=1984436) "algorithm": {
266
- (Runner pid=1984436) "name": "grpo",
267
- (Runner pid=1984436) "gamma": 1.0,
268
- (Runner pid=1984436) "lam": 1.0,
269
- (Runner pid=1984436) "adv_estimator": "grpo",
270
- (Runner pid=1984436) "scale_rewards": true,
271
- (Runner pid=1984436) "disable_kl": true,
272
- (Runner pid=1984436) "use_kl_loss": false,
273
- (Runner pid=1984436) "kl_penalty": "low_var_kl",
274
- (Runner pid=1984436) "kl_coef": 0.0,
275
- (Runner pid=1984436) "kl_type": "fixed",
276
- (Runner pid=1984436) "kl_horizon": 10000.0,
277
- (Runner pid=1984436) "kl_target": 0.1
278
- (Runner pid=1984436) },
279
- (Runner pid=1984436) "trainer": {
280
- (Runner pid=1984436) "total_epochs": 1,
281
- (Runner pid=1984436) "max_steps": 1,
282
- (Runner pid=1984436) "project_name": "OraRL",
283
- (Runner pid=1984436) "experiment_name": "smoke-grpo-9b",
284
- (Runner pid=1984436) "logger": [
285
- (Runner pid=1984436) "console"
286
- (Runner pid=1984436) ],
287
- (Runner pid=1984436) "nnodes": 1,
288
- (Runner pid=1984436) "n_gpus_per_node": 8,
289
- (Runner pid=1984436) "critic_warmup": 0,
290
- (Runner pid=1984436) "val_freq": -1,
291
- (Runner pid=1984436) "val_before_train": false,
292
- (Runner pid=1984436) "val_only": false,
293
- (Runner pid=1984436) "val_generations_to_log": 0,
294
- (Runner pid=1984436) "save_freq": 1,
295
- (Runner pid=1984436) "save_limit": 3,
296
- (Runner pid=1984436) "save_model_only": false,
297
- (Runner pid=1984436) "keep_optim_only_latest": true,
298
- (Runner pid=1984436) "save_checkpoint_path": "/mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/runs/smoke-training-9b-20260825_175213/grpo",
299
- (Runner pid=1984436) "load_checkpoint_path": null,
300
- (Runner pid=1984436) "ray_timeline": null,
301
- (Runner pid=1984436) "find_last_checkpoint": false,
302
- (Runner pid=1984436) "keep_best_train_ckpt": false,
303
- (Runner pid=1984436) "best_train_metric_key": "reward/overall",
304
- (Runner pid=1984436) "best_train_smooth_window": 5,
305
- (Runner pid=1984436) "best_train_min_step": 10
306
- (Runner pid=1984436) }
307
- (Runner pid=1984436) }
308
- (Runner pid=1984436) [TaskGroupedBatchSampler] 15 groups: image_sequence_mc_answer_only(10304), obj_appearance_order(64), object_abs_distance(1344), object_counting(960), object_rel_direction(448), object_rel_distance(640), object_size_estimation(1408), room_size_estimation(960), route_planning(960), segmentation(12032), spatial grounding(7040), spatial-temporal grounding(9536), temporal grounding(20096), tracking(13952), video_qa_mc(20288) | batch_size=8 | total_batches=12504
309
- (Runner pid=1984436) Size of train dataloader: 12504
310
- (Runner pid=1984436) Size of val dataloader: 32
311
- (Runner pid=1984436) KL is disabled, no KL metrics will be logged. Please set `kl_coef=0` to log KL metrics.(pid=1986239) /data/miniconda3/envs/orarl/lib/python3.11/site-packages/transformers/utils/import_utils.py:1006: FutureWarning: `is_flash_attn_greater_or_equal_2_10` is deprecated and will be removed in v5.8. Please use `is_flash_attn_greater_or_equal(library_version='2.1.0')` instead if needed.
312
- (pid=1986239) warnings.warn(
313
- (pid=1986747) /data/miniconda3/envs/orarl/lib/python3.11/site-packages/transformers/utils/import_utils.py:1006: FutureWarning: `is_flash_attn_greater_or_equal_2_10` is deprecated and will be removed in v5.8. Please use `is_flash_attn_greater_or_equal(library_version='2.1.0')` instead if needed.
314
- (pid=1986747) warnings.warn(
315
- (pid=1986750) /data/miniconda3/envs/orarl/lib/python3.11/site-packages/transformers/utils/import_utils.py:1006: FutureWarning: `is_flash_attn_greater_or_equal_2_10` is deprecated and will be removed in v5.8. Please use `is_flash_attn_greater_or_equal(library_version='2.1.0')` instead if needed.
316
- (pid=1986750) warnings.warn(
317
- (WorkerDict pid=1986750) [W825 17:53:20.598420573 ProcessGroupNCCL.cpp:929] Warning: TORCH_NCCL_AVOID_RECORD_STREAMS is the default now, this environment variable is thus deprecated. (function operator())
318
- (pid=1986745) /data/miniconda3/envs/orarl/lib/python3.11/site-packages/transformers/utils/import_utils.py:1006: FutureWarning: `is_flash_attn_greater_or_equal_2_10` is deprecated and will be removed in v5.8. Please use `is_flash_attn_greater_or_equal(library_version='2.1.0')` instead if needed. [repeated 5x across cluster] (Ray deduplicates logs by default. Set RAY_DEDUP_LOGS=0 to disable log deduplication, or see https://docs.ray.io/en/master/ray-observability/user-guides/configure-logging.html#log-deduplication for more options.)
319
- (pid=1986745) warnings.warn( [repeated 5x across cluster]
320
- (WorkerDict pid=1986750) The `use_fast` parameter is deprecated and will be removed in a future version. Use `backend="torchvision"` instead of `use_fast=True`, or `backend="pil"` instead of `use_fast=False`.
321
- (WorkerDict pid=1986750) `torch_dtype` is deprecated! Use `dtype` instead!
322
- (WorkerDict pid=1986745) [W825 17:53:22.611308342 ProcessGroupNCCL.cpp:929] Warning: TORCH_NCCL_AVOID_RECORD_STREAMS is the default now, this environment variable is thus deprecated. (function operator()) [repeated 7x across cluster]
323
- (WorkerDict pid=1986750) The fast path is not available because one of the required library is not installed. Falling back to torch implementation. To install follow https://github.com/fla-org/flash-linear-attention#installation and https://github.com/Dao-AILab/causal-conv1d
324
- (WorkerDict pid=1986750) [rank7]:[W825 17:53:25.252368567 ProcessGroupNCCL.cpp:5138] Guessing device ID based on global rank. This can cause a hang if rank to GPU mapping is heterogeneous. You can specify device_id in init_process_group()
325
- (WorkerDict pid=1986239) The `use_fast` parameter is deprecated and will be removed in a future version. Use `backend="torchvision"` instead of `use_fast=True`, or `backend="pil"` instead of `use_fast=False`. [repeated 7x across cluster]
326
- (WorkerDict pid=1986239)
327
- (WorkerDict pid=1986239)
328
- (WorkerDict pid=1986239) /data/miniconda3/envs/orarl/lib/python3.11/site-packages/torch/distributed/c10d_logger.py:83: UserWarning: barrier(): using the device under current context. You can specify `device_id` in `init_process_group` to mute this warning.
329
- (WorkerDict pid=1986239) return func(*args, **kwargs)
330
- (WorkerDict pid=1986750) `Qwen2VLImageProcessorFast` is deprecated. The `Fast` suffix for image processors has been removed; use `Qwen2VLImageProcessor` instead.
331
- (WorkerDict pid=1986748) `torch_dtype` is deprecated! Use `dtype` instead! [repeated 6x across cluster]
332
- (WorkerDict pid=1986748) The fast path is not available because one of the required library is not installed. Falling back to torch implementation. To install follow https://github.com/fla-org/flash-linear-attention#installation and https://github.com/Dao-AILab/causal-conv1d [repeated 7x across cluster]
333
- (WorkerDict pid=1986239) [rank0]:[W825 17:53:28.954233606 ProcessGroupNCCL.cpp:5138] Guessing device ID based on global rank. This can cause a hang if rank to GPU mapping is heterogeneous. You can specify device_id in init_process_group() [repeated 7x across cluster]
334
- (WorkerDict pid=1986747) <frozen importlib._bootstrap_external>:1241: FutureWarning: The cuda.cudart module is deprecated and will be removed in a future release, please switch to use the cuda.bindings.runtime module instead.
335
- (WorkerDict pid=1986747) <frozen importlib._bootstrap_external>:1241: FutureWarning: The cuda.nvrtc module is deprecated and will be removed in a future release, please switch to use the cuda.bindings.nvrtc module instead.
336
- (WorkerDict pid=1986749) `Qwen2VLImageProcessorFast` is deprecated. The `Fast` suffix for image processors has been removed; use `Qwen2VLImageProcessor` instead. [repeated 7x across cluster]
337
- (WorkerDict pid=1986239) 2026-08-25 17:54:33,940 - INFO - autotuner.py:262 - flashinfer.jit: [Autotuner]: Autotuning process starts ...
338
- (WorkerDict pid=1986239) 2026-08-25 17:54:33,958 - INFO - autotuner.py:268 - flashinfer.jit: [Autotuner]: Autotuning process ends
339
- (WorkerDict pid=1986748) <frozen importlib._bootstrap_external>:1241: FutureWarning: The cuda.cudart module is deprecated and will be removed in a future release, please switch to use the cuda.bindings.runtime module instead. [repeated 7x across cluster]
340
- (WorkerDict pid=1986748) <frozen importlib._bootstrap_external>:1241: FutureWarning: The cuda.nvrtc module is deprecated and will be removed in a future release, please switch to use the cuda.bindings.nvrtc module instead. [repeated 7x across cluster]
341
- (WorkerDict pid=1986239)
342
- (WorkerDict pid=1986239)
343
- (WorkerDict pid=1986239)
344
- (WorkerDict pid=1986239)
345
- (WorkerDict pid=1986239)
346
- (WorkerDict pid=1986239)
347
- (WorkerDict pid=1986239)
348
- (WorkerDict pid=1986239)
349
- (WorkerDict pid=1986239)
350
- (WorkerDict pid=1986239)
351
- (WorkerDict pid=1986239)
352
- (WorkerDict pid=1986239)
353
- (WorkerDict pid=1986239)
354
- (WorkerDict pid=1986239)
355
- (WorkerDict pid=1986239)
356
- (WorkerDict pid=1986750) 2026-08-25 17:54:33,938 - INFO - autotuner.py:262 - flashinfer.jit: [Autotuner]: Autotuning process starts ... [repeated 7x across cluster]
357
- (WorkerDict pid=1986750) 2026-08-25 17:54:33,956 - INFO - autotuner.py:268 - flashinfer.jit: [Autotuner]: Autotuning process ends [repeated 7x across cluster]
358
- (WorkerDict pid=1986239)
359
- (WorkerDict pid=1986239)
360
- (WorkerDict pid=1986239)
361
- (WorkerDict pid=1986239)
362
- (WorkerDict pid=1986239)
363
- (WorkerDict pid=1986239)
364
- (WorkerDict pid=1986239)
365
- (WorkerDict pid=1986239)
366
- (WorkerDict pid=1986239)
367
- (WorkerDict pid=1986239)
368
- (WorkerDict pid=1986239)
369
- (WorkerDict pid=1986239)
370
- (WorkerDict pid=1986239)
371
- (WorkerDict pid=1986239)
372
- (WorkerDict pid=1986239)
373
- (WorkerDict pid=1986239)
374
- (WorkerDict pid=1986239)
375
- (WorkerDict pid=1986239)
376
- (WorkerDict pid=1986239)
377
- (WorkerDict pid=1986239)
378
- (WorkerDict pid=1986239)
379
- (WorkerDict pid=1986239)
380
- (WorkerDict pid=1986239)
381
- (WorkerDict pid=1986239)
382
- (WorkerDict pid=1986239)
383
- (WorkerDict pid=1986239)
384
- (WorkerDict pid=1986239)
385
- (WorkerDict pid=1986239)
386
-
387
- (Runner pid=1984436) [GRPO] Scaled advantages ON: A=(r-mean)/(std+eps).
388
- (Runner pid=1984436) Total training steps: 1
389
- (AutoRewardManager pid=1986029) Using reward function `compute_score` from `orarl.rewards`.
390
- (AutoRewardManager pid=1986029) Reward name: orarl, reward type: batch.
391
- (WorkerDict pid=1986239) actor will use global batch size 64.
392
- (AutoRewardManager pid=1986031) Using reward function `compute_score` from `orarl.rewards`.
393
- (AutoRewardManager pid=1986031) Reward name: orarl, reward type: batch.
394
- (WorkerDict pid=1986239) Model config: Qwen3_5Config {
395
- (WorkerDict pid=1986239) "architectures": [
396
- (WorkerDict pid=1986239) "Qwen3_5ForConditionalGeneration"
397
- (WorkerDict pid=1986239) ],
398
- (WorkerDict pid=1986239) "dtype": "bfloat16",
399
- (WorkerDict pid=1986239) "image_token_id": 248056,
400
- (WorkerDict pid=1986239) "model_type": "qwen3_5",
401
- (WorkerDict pid=1986239) "text_config": {
402
- (WorkerDict pid=1986239) "attention_bias": false,
403
- (WorkerDict pid=1986239) "attention_dropout": 0.0,
404
- (WorkerDict pid=1986239) "attn_output_gate": true,
405
- (WorkerDict pid=1986239) "bos_token_id": null,
406
- (WorkerDict pid=1986239) "dtype": "bfloat16",
407
- (WorkerDict pid=1986239) "eos_token_id": 248044,
408
- (WorkerDict pid=1986239) "full_attention_interval": 4,
409
- (WorkerDict pid=1986239) "head_dim": 256,
410
- (WorkerDict pid=1986239) "hidden_act": "silu",
411
- (WorkerDict pid=1986239) "hidden_size": 4096,
412
- (WorkerDict pid=1986239) "initializer_range": 0.02,
413
- (WorkerDict pid=1986239) "intermediate_size": 12288,
414
- (WorkerDict pid=1986239) "layer_types": [
415
- (WorkerDict pid=1986239) "linear_attention",
416
- (WorkerDict pid=1986239) "linear_attention",
417
- (WorkerDict pid=1986239) "linear_attention",
418
- (WorkerDict pid=1986239) "full_attention",
419
- (WorkerDict pid=1986239) "linear_attention",
420
- (WorkerDict pid=1986239) "linear_attention",
421
- (WorkerDict pid=1986239) "linear_attention",
422
- (WorkerDict pid=1986239) "full_attention",
423
- (WorkerDict pid=1986239) "linear_attention",
424
- (WorkerDict pid=1986239) "linear_attention",
425
- (WorkerDict pid=1986239) "linear_attention",
426
- (WorkerDict pid=1986239) "full_attention",
427
- (WorkerDict pid=1986239) "linear_attention",
428
- (WorkerDict pid=1986239) "linear_attention",
429
- (WorkerDict pid=1986239) "linear_attention",
430
- (WorkerDict pid=1986239) "full_attention",
431
- (WorkerDict pid=1986239) "linear_attention",
432
- (WorkerDict pid=1986239) "linear_attention",
433
- (WorkerDict pid=1986239) "linear_attention",
434
- (WorkerDict pid=1986239) "full_attention",
435
- (WorkerDict pid=1986239) "linear_attention",
436
- (WorkerDict pid=1986239) "linear_attention",
437
- (WorkerDict pid=1986239) "linear_attention",
438
- (WorkerDict pid=1986239) "full_attention",
439
- (WorkerDict pid=1986239) "linear_attention",
440
- (WorkerDict pid=1986239) "linear_attention",
441
- (WorkerDict pid=1986239) "linear_attention",
442
- (WorkerDict pid=1986239) "full_attention",
443
- (WorkerDict pid=1986239) "linear_attention",
444
- (WorkerDict pid=1986239) "linear_attention",
445
- (WorkerDict pid=1986239) "linear_attention",
446
- (WorkerDict pid=1986239) "full_attention"
447
- (WorkerDict pid=1986239) ],
448
- (WorkerDict pid=1986239) "linear_conv_kernel_dim": 4,
449
- (WorkerDict pid=1986239) "linear_key_head_dim": 128,
450
- (WorkerDict pid=1986239) "linear_num_key_heads": 16,
451
- (WorkerDict pid=1986239) "linear_num_value_heads": 32,
452
- (WorkerDict pid=1986239) "linear_value_head_dim": 128,
453
- (WorkerDict pid=1986239) "mamba_ssm_dtype": "float32",
454
- (WorkerDict pid=1986239) "max_position_embeddings": 262144,
455
- (WorkerDict pid=1986239) "mlp_only_layers": [],
456
- (WorkerDict pid=1986239) "model_type": "qwen3_5_text",
457
- (WorkerDict pid=1986239) "mtp_num_hidden_layers": 0,
458
- (WorkerDict pid=1986239) "mtp_use_dedicated_embeddings": false,
459
- (WorkerDict pid=1986239) "num_attention_heads": 16,
460
- (WorkerDict pid=1986239) "num_hidden_layers": 32,
461
- (WorkerDict pid=1986239) "num_key_value_heads": 4,
462
- (WorkerDict pid=1986239) "pad_token_id": null,
463
- (WorkerDict pid=1986239) "partial_rotary_factor": 0.25,
464
- (WorkerDict pid=1986239) "rms_norm_eps": 1e-06,
465
- (WorkerDict pid=1986239) "rope_parameters": {
466
- (WorkerDict pid=1986239) "mrope_interleaved": true,
467
- (WorkerDict pid=1986239) "mrope_section": [
468
- (WorkerDict pid=1986239) 11,
469
- (WorkerDict pid=1986239) 11,
470
- (WorkerDict pid=1986239) 10
471
- (WorkerDict pid=1986239) ],
472
- (WorkerDict pid=1986239) "partial_rotary_factor": 0.25,
473
- (WorkerDict pid=1986239) "rope_theta": 10000000,
474
- (WorkerDict pid=1986239) "rope_type": "default"
475
- (WorkerDict pid=1986239) },
476
- (WorkerDict pid=1986239) "tie_word_embeddings": false,
477
- (WorkerDict pid=1986239) "use_cache": true,
478
- (WorkerDict pid=1986239) "vocab_size": 248320
479
- (WorkerDict pid=1986239) },
480
- (WorkerDict pid=1986239) "tie_word_embeddings": false,
481
- (WorkerDict pid=1986239) "transformers_version": "5.5.4",
482
- (WorkerDict pid=1986239) "video_token_id": 248057,
483
- (WorkerDict pid=1986239) "vision_config": {
484
- (WorkerDict pid=1986239) "deepstack_visual_indexes": [],
485
- (WorkerDict pid=1986239) "depth": 27,
486
- (WorkerDict pid=1986239) "dtype": "bfloat16",
487
- (WorkerDict pid=1986239) "hidden_act": "gelu_pytorch_tanh",
488
- (WorkerDict pid=1986239) "hidden_size": 1152,
489
- (WorkerDict pid=1986239) "in_channels": 3,
490
- (WorkerDict pid=1986239) "initializer_range": 0.02,
491
- (WorkerDict pid=1986239) "intermediate_size": 4304,
492
- (WorkerDict pid=1986239) "model_type": "qwen3_5",
493
- (WorkerDict pid=1986239) "num_heads": 16,
494
- (WorkerDict pid=1986239) "num_position_embeddings": 2304,
495
- (WorkerDict pid=1986239) "out_hidden_size": 4096,
496
- (WorkerDict pid=1986239) "patch_size": 16,
497
- (WorkerDict pid=1986239) "spatial_merge_size": 2,
498
- (WorkerDict pid=1986239) "temporal_patch_size": 2
499
- (WorkerDict pid=1986239) },
500
- (WorkerDict pid=1986239) "vision_end_token_id": 248054,
501
- (WorkerDict pid=1986239) "vision_start_token_id": 248053
502
- (WorkerDict pid=1986239) }
503
- (WorkerDict pid=1986239)
504
- (WorkerDict pid=1986239) Ulysses patch applied!
505
- (WorkerDict pid=1986239) Vision backbone is set to not trainable.
506
- (WorkerDict pid=1986239) Qwen3_5ForConditionalGeneration contains 9.41B parameters.
507
- (WorkerDict pid=1986239) After huggingface model init: 1.17 GB / 95.00 GB.
508
- (WorkerDict pid=1986239) FSDP wrap policy: functools.partial(<function transformer_auto_wrap_policy at 0x7f3e365eb600>, transformer_layer_cls={<class 'transformers.models.qwen3_5.modeling_qwen3_5.Qwen3_5VisionBlock'>, <class 'transformers.models.qwen3_5.modeling_qwen3_5.Qwen3_5DecoderLayer'>}).
509
- (WorkerDict pid=1986239) After FSDP module init: 15.42 GB / 95.00 GB.
510
- (WorkerDict pid=1986239) After optimizer init: 15.42 GB / 95.00 GB.
511
- (WorkerDict pid=1986744) Sampling params: {'max_tokens': 4096, 'detokenize': False, 'logit_bias': {248056: -100, 248057: -100}, 'n': 8, 'temperature': 1.0, 'top_p': 0.85, 'top_k': -1, 'ignore_eos': False}.
512
- (WorkerDict pid=1986239) After vllm init: 7.76 GB / 95.00 GB.
513
- (Runner pid=1984436) Config
514
- (Runner pid=1984436) algorithm:
515
- (Runner pid=1984436) adv_estimator: grpo
516
- (Runner pid=1984436) disable_kl: true
517
- (Runner pid=1984436) gamma: 1.0
518
- (Runner pid=1984436) kl_coef: 0.0
519
- (Runner pid=1984436) kl_horizon: 10000.0
520
- (Runner pid=1984436) kl_penalty: low_var_kl
521
- (Runner pid=1984436) kl_target: 0.1
522
- (Runner pid=1984436) kl_type: fixed
523
- (Runner pid=1984436) lam: 1.0
524
- (Runner pid=1984436) name: grpo
525
- (Runner pid=1984436) scale_rewards: true
526
- (Runner pid=1984436) use_kl_loss: false
527
- (Runner pid=1984436) data:
528
- (Runner pid=1984436) answer_key: answer
529
- (Runner pid=1984436) dataloader_num_workers: 1
530
- (Runner pid=1984436) enable_thinking: false
531
- (Runner pid=1984436) filter_overlong_prompts: false
532
- (Runner pid=1984436) filter_overlong_prompts_workers: 16
533
- (Runner pid=1984436) format_prompt: null
534
- (Runner pid=1984436) group_by_task: true
535
- (Runner pid=1984436) group_by_task_key: problem_type
536
- (Runner pid=1984436) image_dir: /
537
- (Runner pid=1984436) image_key: images
538
- (Runner pid=1984436) image_max_pixels: 1048576
539
- (Runner pid=1984436) image_min_pixels: 4096
540
- (Runner pid=1984436) inline_video_tensors: true
541
- (Runner pid=1984436) max_pixels: 4194304
542
- (Runner pid=1984436) max_prompt_length: 24576
543
- (Runner pid=1984436) max_response_length: 4096
544
- (Runner pid=1984436) min_pixels: 4096
545
- (Runner pid=1984436) mini_rollout_batch_size: null
546
- (Runner pid=1984436) override_chat_template: null
547
- (Runner pid=1984436) preprocessed_video_dir: null
548
- (Runner pid=1984436) prompt_key: problem
549
- (Runner pid=1984436) response_prefix: ''
550
- (Runner pid=1984436) rollout_batch_size: 8
551
- (Runner pid=1984436) seed: 42
552
- (Runner pid=1984436) shuffle: true
553
- (Runner pid=1984436) train_files: /mnt/lyh/TempSamp/EasyVideoR1/data/joint_final/rl_joint_100k_maskaware.jsonl
554
- (Runner pid=1984436) use_preprocessed_videos: false
555
- (Runner pid=1984436) val_batch_size: 16
556
- (Runner pid=1984436) val_files: /mnt/lyh/TempSamp/EasyVideoR1/data/joint_final/rl_phase3_49152.canary.jsonl
557
- (Runner pid=1984436) val_preprocessed_video_dir: null
558
- (Runner pid=1984436) val_video_fps: 2.0
559
- (Runner pid=1984436) val_video_max_frames: 128
560
- (Runner pid=1984436) val_video_max_pixels: 786432
561
- (Runner pid=1984436) val_video_min_pixels: 4096
562
- (Runner pid=1984436) val_video_source_mode: realtime_only
563
- (Runner pid=1984436) val_video_total_pixels: 8388608
564
- (Runner pid=1984436) video_fps: 2.0
565
- (Runner pid=1984436) video_key: videos
566
- (Runner pid=1984436) video_max_frames: 128
567
- (Runner pid=1984436) video_max_pixels: 786432
568
- (Runner pid=1984436) video_min_pixels: 4096
569
- (Runner pid=1984436) video_source_mode: realtime_only
570
- (Runner pid=1984436) video_total_pixels: 8388608
571
- (Runner pid=1984436) trainer:
572
- (Runner pid=1984436) best_train_metric_key: reward/overall
573
- (Runner pid=1984436) best_train_min_step: 10
574
- (Runner pid=1984436) best_train_smooth_window: 5
575
- (Runner pid=1984436) critic_warmup: 0
576
- (Runner pid=1984436) experiment_name: smoke-grpo-9b
577
- (Runner pid=1984436) find_last_checkpoint: false
578
- (Runner pid=1984436) keep_best_train_ckpt: false
579
- (Runner pid=1984436) keep_optim_only_latest: true
580
- (Runner pid=1984436) load_checkpoint_path: null
581
- (Runner pid=1984436) logger:
582
- (Runner pid=1984436) - console
583
- (Runner pid=1984436) max_steps: 1
584
- (Runner pid=1984436) n_gpus_per_node: 8
585
- (Runner pid=1984436) nnodes: 1
586
- (Runner pid=1984436) project_name: OraRL
587
- (Runner pid=1984436) ray_timeline: null
588
- (Runner pid=1984436) save_checkpoint_path: /mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/runs/smoke-training-9b-20260825_175213/grpo
589
- (Runner pid=1984436) save_freq: 1
590
- (Runner pid=1984436) save_limit: 3
591
- (Runner pid=1984436) save_model_only: false
592
- (Runner pid=1984436) total_epochs: 1
593
- (Runner pid=1984436) val_before_train: false
594
- (Runner pid=1984436) val_freq: -1
595
- (Runner pid=1984436) val_generations_to_log: 0
596
- (Runner pid=1984436) val_only: false
597
- (Runner pid=1984436) worker:
598
- (Runner pid=1984436) actor:
599
- (Runner pid=1984436) clip_ratio_dual: 3.0
600
- (Runner pid=1984436) clip_ratio_high: 0.2
601
- (Runner pid=1984436) clip_ratio_low: 0.2
602
- (Runner pid=1984436) disable_kl: true
603
- (Runner pid=1984436) dynamic_batching: true
604
- (Runner pid=1984436) fsdp:
605
- (Runner pid=1984436) enable_cpu_offload: false
606
- (Runner pid=1984436) enable_full_shard: true
607
- (Runner pid=1984436) enable_rank0_init: true
608
- (Runner pid=1984436) fsdp_size: -1
609
- (Runner pid=1984436) mp_buffer_dtype: fp32
610
- (Runner pid=1984436) mp_param_dtype: bf16
611
- (Runner pid=1984436) mp_reduce_dtype: fp32
612
- (Runner pid=1984436) torch_dtype: bf16
613
- (Runner pid=1984436) use_orig_params: false
614
- (Runner pid=1984436) global_batch_size: 8
615
- (Runner pid=1984436) global_batch_size_per_device: -1
616
- (Runner pid=1984436) kl_coef: 0.0
617
- (Runner pid=1984436) kl_penalty: low_var_kl
618
- (Runner pid=1984436) loss_avg_mode: token
619
- (Runner pid=1984436) loss_type: default
620
- (Runner pid=1984436) max_grad_norm: 1.0
621
- (Runner pid=1984436) max_token_len_per_gpu: 32768
622
- (Runner pid=1984436) micro_batch_size_per_device_for_experience: 1
623
- (Runner pid=1984436) micro_batch_size_per_device_for_update: 1
624
- (Runner pid=1984436) model:
625
- (Runner pid=1984436) enable_gradient_checkpointing: true
626
- (Runner pid=1984436) freeze_vision_tower: true
627
- (Runner pid=1984436) model_path: /apdcephfs_gy7/share_305340263/hunyuan/yunhengli/checkpoint/Video-ORA-9B
628
- (Runner pid=1984436) override_config: {}
629
- (Runner pid=1984436) tokenizer_path: /apdcephfs_gy7/share_305340263/hunyuan/yunhengli/checkpoint/Video-ORA-9B
630
- (Runner pid=1984436) train_vision_merger: false
631
- (Runner pid=1984436) trust_remote_code: false
632
- (Runner pid=1984436) offload:
633
- (Runner pid=1984436) offload_optimizer: false
634
- (Runner pid=1984436) offload_params: false
635
- (Runner pid=1984436) optim:
636
- (Runner pid=1984436) betas:
637
- (Runner pid=1984436) - 0.9
638
- (Runner pid=1984436) - 0.999
639
- (Runner pid=1984436) lr: 1.0e-06
640
- (Runner pid=1984436) lr_scheduler_type: constant
641
- (Runner pid=1984436) lr_warmup_ratio: 0.0
642
- (Runner pid=1984436) lr_warmup_steps: null
643
- (Runner pid=1984436) min_lr_ratio: null
644
- (Runner pid=1984436) strategy: adamw
645
- (Runner pid=1984436) training_steps: 1
646
- (Runner pid=1984436) weight_decay: 0.0
647
- (Runner pid=1984436) padding_free: true
648
- (Runner pid=1984436) ppo_epochs: 1
649
- (Runner pid=1984436) strategy: fsdp
650
- (Runner pid=1984436) ulysses_size: 1
651
- (Runner pid=1984436) use_kl_loss: false
652
- (Runner pid=1984436) use_torch_compile: false
653
- (Runner pid=1984436) critic:
654
- (Runner pid=1984436) cliprange_value: 0.5
655
- (Runner pid=1984436) dynamic_batching: true
656
- (Runner pid=1984436) fsdp:
657
- (Runner pid=1984436) enable_cpu_offload: false
658
- (Runner pid=1984436) enable_full_shard: true
659
- (Runner pid=1984436) enable_rank0_init: true
660
- (Runner pid=1984436) fsdp_size: -1
661
- (Runner pid=1984436) mp_buffer_dtype: fp32
662
- (Runner pid=1984436) mp_param_dtype: bf16
663
- (Runner pid=1984436) mp_reduce_dtype: fp32
664
- (Runner pid=1984436) torch_dtype: null
665
- (Runner pid=1984436) use_orig_params: false
666
- (Runner pid=1984436) global_batch_size: 256
667
- (Runner pid=1984436) global_batch_size_per_device: -1
668
- (Runner pid=1984436) loss_avg_mode: token
669
- (Runner pid=1984436) max_grad_norm: 1.0
670
- (Runner pid=1984436) max_token_len_per_gpu: null
671
- (Runner pid=1984436) micro_batch_size_per_device_for_experience: 16
672
- (Runner pid=1984436) micro_batch_size_per_device_for_update: 4
673
- (Runner pid=1984436) model:
674
- (Runner pid=1984436) enable_gradient_checkpointing: true
675
- (Runner pid=1984436) freeze_vision_tower: false
676
- (Runner pid=1984436) model_path: null
677
- (Runner pid=1984436) override_config: {}
678
- (Runner pid=1984436) tokenizer_path: null
679
- (Runner pid=1984436) train_vision_merger: false
680
- (Runner pid=1984436) trust_remote_code: true
681
- (Runner pid=1984436) offload:
682
- (Runner pid=1984436) offload_optimizer: false
683
- (Runner pid=1984436) offload_params: false
684
- (Runner pid=1984436) optim:
685
- (Runner pid=1984436) betas:
686
- (Runner pid=1984436) - 0.9
687
- (Runner pid=1984436) - 0.999
688
- (Runner pid=1984436) lr: 1.0e-06
689
- (Runner pid=1984436) lr_scheduler_type: constant
690
- (Runner pid=1984436) lr_warmup_ratio: 0.0
691
- (Runner pid=1984436) lr_warmup_steps: null
692
- (Runner pid=1984436) min_lr_ratio: null
693
- (Runner pid=1984436) strategy: adamw
694
- (Runner pid=1984436) training_steps: 1
695
- (Runner pid=1984436) weight_decay: 0.01
696
- (Runner pid=1984436) padding_free: false
697
- (Runner pid=1984436) ppo_epochs: 1
698
- (Runner pid=1984436) strategy: fsdp
699
- (Runner pid=1984436) ulysses_size: 1
700
- (Runner pid=1984436) hybrid_engine: true
701
- (Runner pid=1984436) ref:
702
- (Runner pid=1984436) dynamic_batching: true
703
- (Runner pid=1984436) fsdp:
704
- (Runner pid=1984436) enable_cpu_offload: false
705
- (Runner pid=1984436) enable_full_shard: true
706
- (Runner pid=1984436) enable_rank0_init: true
707
- (Runner pid=1984436) fsdp_size: -1
708
- (Runner pid=1984436) mp_buffer_dtype: fp32
709
- (Runner pid=1984436) mp_param_dtype: bf16
710
- (Runner pid=1984436) mp_reduce_dtype: fp32
711
- (Runner pid=1984436) torch_dtype: bf16
712
- (Runner pid=1984436) use_orig_params: false
713
- (Runner pid=1984436) max_token_len_per_gpu: 32768
714
- (Runner pid=1984436) micro_batch_size_per_device_for_experience: 1
715
- (Runner pid=1984436) offload:
716
- (Runner pid=1984436) offload_optimizer: false
717
- (Runner pid=1984436) offload_params: false
718
- (Runner pid=1984436) padding_free: true
719
- (Runner pid=1984436) strategy: fsdp
720
- (Runner pid=1984436) ulysses_size: 1
721
- (Runner pid=1984436) use_torch_compile: false
722
- (Runner pid=1984436) reward:
723
- (Runner pid=1984436) num_cpus: 1
724
- (Runner pid=1984436) reward_function: orarl.rewards
725
- (Runner pid=1984436) reward_function_is_module: true
726
- (Runner pid=1984436) reward_function_kwargs: {}
727
- (Runner pid=1984436) reward_function_name: compute_score
728
- (Runner pid=1984436) skip_special_tokens: true
729
- (Runner pid=1984436) rollout:
730
- (Runner pid=1984436) calculate_log_probs: false
731
- (Runner pid=1984436) collect_seq_logprob_for_filter: false
732
- (Runner pid=1984436) disable_log_stats: true
733
- (Runner pid=1984436) disable_tqdm: true
734
- (Runner pid=1984436) dtype: bf16
735
- (Runner pid=1984436) enable_chunked_prefill: true
736
- (Runner pid=1984436) enforce_eager: false
737
- (Runner pid=1984436) gpu_memory_utilization: 0.45
738
- (Runner pid=1984436) ignore_eos: false
739
- (Runner pid=1984436) kv_cache_dtype: auto
740
- (Runner pid=1984436) limit_images: 10
741
- (Runner pid=1984436) max_model_len: null
742
- (Runner pid=1984436) max_num_batched_tokens: 32768
743
- (Runner pid=1984436) n: 8
744
- (Runner pid=1984436) name: vllm
745
- (Runner pid=1984436) prompt_length: 24576
746
- (Runner pid=1984436) response_length: 4096
747
- (Runner pid=1984436) seed: 42
748
- (Runner pid=1984436) temperature: 1.0
749
- (Runner pid=1984436) tensor_parallel_size: 1
750
- (Runner pid=1984436) top_k: -1
751
- (Runner pid=1984436) top_p: 0.85
752
- (Runner pid=1984436) trust_remote_code: false
753
- (Runner pid=1984436) val_override_config:
754
- (Runner pid=1984436) n: 1
755
- (Runner pid=1984436) temperature: 0.7
756
- (Runner pid=1984436) top_p: 0.95
757
- (Runner pid=1984436)
758
-
759
 
760
 
761
 
762
 
763
 
764
- File "<frozen runpy>", line 198, in _run_module_as_main
765
- File "<frozen runpy>", line 88, in _run_code
766
- File "/mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/verl/trainer/main.py", line 137, in <module>
767
- main()
768
- File "/mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/verl/trainer/main.py", line 129, in main
769
- ray.get(runner.run.remote(ppo_config))
770
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/ray/_private/auto_init_hook.py", line 22, in auto_init_wrapper
771
- return fn(*args, **kwargs)
772
- ^^^^^^^^^^^^^^^^^^^
773
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/ray/_private/client_mode_hook.py", line 104, in wrapper
774
- return func(*args, **kwargs)
775
- ^^^^^^^^^^^^^^^^^^^^^
776
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/ray/_private/worker.py", line 2981, in get
777
- values, debugger_breakpoint = worker.get_objects(
778
- ^^^^^^^^^^^^^^^^^^^
779
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/ray/_private/worker.py", line 1012, in get_objects
780
- raise value.as_instanceof_cause()
781
- ray.exceptions.RayTaskError(IndexError): ray::Runner.run() (pid=1984436, ip=29.127.69.4, actor_id=2700fe7dc5d33e51d56c429c01000000, repr=<main.Runner object at 0x7ff5f0413990>)
782
- ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
783
- ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
784
- File "/mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/verl/trainer/main.py", line 92, in run
785
- trainer.fit()
786
- File "/mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/verl/trainer/ray_trainer.py", line 1870, in fit
787
- batch = self._make_batch_data(metrics=metrics)
788
- ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
789
- File "/mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/verl/trainer/ray_trainer.py", line 1662, in _make_batch_data
790
- batch_dict = next(self.data_iterator)
791
- ^^^^^^^^^^^^^^^^^^^^^^^^
792
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/torchdata/stateful_dataloader/stateful_dataloader.py", line 450, in __next__
793
- return super().__next__()
794
- ^^^^^^^^^^^^^^^^^^
795
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/torch/utils/data/dataloader.py", line 741, in __next__
796
- data = self._next_data()
797
- ^^^^^^^^^^^^^^^^^
798
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/torchdata/stateful_dataloader/stateful_dataloader.py", line 1456, in _next_data
799
- return self._process_data(data, worker_id, state_dict)
800
- ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
801
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/torchdata/stateful_dataloader/stateful_dataloader.py", line 1543, in _process_data
802
- data.reraise()
803
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/torch/_utils.py", line 775, in reraise
804
- raise exception
805
- IndexError: Caught IndexError in DataLoader worker process 0.
806
- Original Traceback (most recent call last):
807
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/torchdata/stateful_dataloader/worker.py", line 242, in _worker_loop
808
- data = fetcher.fetch(index) # type: ignore[union-attr]
809
- ^^^^^^^^^^^^^^^^^^^^
810
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/torch/utils/data/_utils/fetch.py", line 54, in fetch
811
- data = [self.dataset[idx] for idx in possibly_batched_index]
812
- ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
813
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/torch/utils/data/_utils/fetch.py", line 54, in <listcomp>
814
- data = [self.dataset[idx] for idx in possibly_batched_index]
815
- ~~~~~~~~~~~~^^^^^
816
- File "/mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/verl/utils/dataset.py", line 826, in __getitem__
817
- model_inputs = self.processor(processed_images, [prompt], add_special_tokens=False, return_tensors="pt")
818
- ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
819
- File "/data/miniconda3/envs/orarl/lib/python3.11/site-packages/transformers/models/qwen3_vl/processing_qwen3_vl.py", line 132, in __call__
820
- num_image_tokens = image_grid_thw[index].prod() // merge_length
821
- ~~~~~~~~~~~~~~^^^^^^^
822
- IndexError: index 8 is out of bounds for dimension 0 with size 8
823
-
824
 
825
-
826
 
827
- (WorkerDict pid=1986239) After gather model weights in sharding manager: 31.26 GB / 95.00 GB.
828
- (Runner pid=1984436) Start generating batch...
829
- (WorkerDict pid=1986239) After sync model weights in sharding manager: 25.44 GB / 95.00 GB.
830
- (WorkerDict pid=1986239) After vllm wake up in sharding manager: 45.79 GB / 95.00 GB.
831
- (WorkerDict pid=1986746) Sampling params: {'max_tokens': 4096, 'detokenize': False, 'logit_bias': {248056: -100, 248057: -100}, 'n': 8, 'temperature': 1.0, 'top_p': 0.85, 'top_k': -1, 'ignore_eos': False}. [repeated 7x across cluster]
832
-
833
- method=grpo mode=run
834
- /data/miniconda3/envs/orarl/bin/python -m verl.trainer.main config=/mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/configs/grpo_9b.yaml trainer.max_steps=1 trainer.save_freq=1 trainer.val_before_train=false trainer.experiment_name=smoke-grpo-9b data.rollout_batch_size=8 worker.actor.global_batch_size=8 worker.actor.model.model_path=/apdcephfs_gy7/share_305340263/hunyuan/yunhengli/checkpoint/Video-ORA-9B worker.actor.model.tokenizer_path=/apdcephfs_gy7/share_305340263/hunyuan/yunhengli/checkpoint/Video-ORA-9B data.train_files=/mnt/lyh/TempSamp/EasyVideoR1/data/joint_final/rl_joint_100k_maskaware.jsonl data.val_files=/mnt/lyh/TempSamp/EasyVideoR1/data/joint_final/rl_phase3_49152.canary.jsonl trainer.save_checkpoint_path=/mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/runs/smoke-training-9b-20260825_175213/grpo trainer.nnodes=1 trainer.n_gpus_per_node=8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
0
 
1
 
2
 
3
 
4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
 
 
6
 
 
 
 
 
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/checkpoint_tracker.json DELETED
@@ -1,6 +0,0 @@
1
- {
2
- "best_global_step": null,
3
- "best_val_reward_score": -1.0,
4
- "last_global_step": 1,
5
- "last_actor_path": "/mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor"
6
- }
 
 
 
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/extra_state_world_size_8_rank_0.pt DELETED
Binary file (15.2 kB)
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/extra_state_world_size_8_rank_1.pt DELETED
Binary file (15.2 kB)
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/extra_state_world_size_8_rank_2.pt DELETED
Binary file (15.2 kB)
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/extra_state_world_size_8_rank_3.pt DELETED
Binary file (15.2 kB)
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/extra_state_world_size_8_rank_4.pt DELETED
Binary file (15.2 kB)
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/extra_state_world_size_8_rank_5.pt DELETED
Binary file (15.2 kB)
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/extra_state_world_size_8_rank_6.pt DELETED
Binary file (15.2 kB)
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/extra_state_world_size_8_rank_7.pt DELETED
Binary file (15.2 kB)
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/huggingface/chat_template.jinja DELETED
@@ -1,154 +0,0 @@
1
- {%- set image_count = namespace(value=0) %}
2
- {%- set video_count = namespace(value=0) %}
3
- {%- macro render_content(content, do_vision_count, is_system_content=false) %}
4
- {%- if content is string %}
5
- {{- content }}
6
- {%- elif content is iterable and content is not mapping %}
7
- {%- for item in content %}
8
- {%- if 'image' in item or 'image_url' in item or item.type == 'image' %}
9
- {%- if is_system_content %}
10
- {{- raise_exception('System message cannot contain images.') }}
11
- {%- endif %}
12
- {%- if do_vision_count %}
13
- {%- set image_count.value = image_count.value + 1 %}
14
- {%- endif %}
15
- {%- if add_vision_id %}
16
- {{- 'Picture ' ~ image_count.value ~ ': ' }}
17
- {%- endif %}
18
- {{- '<|vision_start|><|image_pad|><|vision_end|>' }}
19
- {%- elif 'video' in item or item.type == 'video' %}
20
- {%- if is_system_content %}
21
- {{- raise_exception('System message cannot contain videos.') }}
22
- {%- endif %}
23
- {%- if do_vision_count %}
24
- {%- set video_count.value = video_count.value + 1 %}
25
- {%- endif %}
26
- {%- if add_vision_id %}
27
- {{- 'Video ' ~ video_count.value ~ ': ' }}
28
- {%- endif %}
29
- {{- '<|vision_start|><|video_pad|><|vision_end|>' }}
30
- {%- elif 'text' in item %}
31
- {{- item.text }}
32
- {%- else %}
33
- {{- raise_exception('Unexpected item type in content.') }}
34
- {%- endif %}
35
- {%- endfor %}
36
- {%- elif content is none or content is undefined %}
37
- {{- '' }}
38
- {%- else %}
39
- {{- raise_exception('Unexpected content type.') }}
40
- {%- endif %}
41
- {%- endmacro %}
42
- {%- if not messages %}
43
- {{- raise_exception('No messages provided.') }}
44
- {%- endif %}
45
- {%- if tools and tools is iterable and tools is not mapping %}
46
- {{- '<|im_start|>system\n' }}
47
- {{- "# Tools\n\nYou have access to the following functions:\n\n<tools>" }}
48
- {%- for tool in tools %}
49
- {{- "\n" }}
50
- {{- tool | tojson }}
51
- {%- endfor %}
52
- {{- "\n</tools>" }}
53
- {{- '\n\nIf you choose to call a function ONLY reply in the following format with NO suffix:\n\n<tool_call>\n<function=example_function_name>\n<parameter=example_parameter_1>\nvalue_1\n</parameter>\n<parameter=example_parameter_2>\nThis is the value for the second parameter\nthat can span\nmultiple lines\n</parameter>\n</function>\n</tool_call>\n\n<IMPORTANT>\nReminder:\n- Function calls MUST follow the specified format: an inner <function=...></function> block must be nested within <tool_call></tool_call> XML tags\n- Required parameters MUST be specified\n- You may provide optional reasoning for your function call in natural language BEFORE the function call, but NOT after\n- If there is no function call available, answer the question like normal with your current knowledge and do not tell the user about function calls\n</IMPORTANT>' }}
54
- {%- if messages[0].role == 'system' %}
55
- {%- set content = render_content(messages[0].content, false, true)|trim %}
56
- {%- if content %}
57
- {{- '\n\n' + content }}
58
- {%- endif %}
59
- {%- endif %}
60
- {{- '<|im_end|>\n' }}
61
- {%- else %}
62
- {%- if messages[0].role == 'system' %}
63
- {%- set content = render_content(messages[0].content, false, true)|trim %}
64
- {{- '<|im_start|>system\n' + content + '<|im_end|>\n' }}
65
- {%- endif %}
66
- {%- endif %}
67
- {%- set ns = namespace(multi_step_tool=true, last_query_index=messages|length - 1) %}
68
- {%- for message in messages[::-1] %}
69
- {%- set index = (messages|length - 1) - loop.index0 %}
70
- {%- if ns.multi_step_tool and message.role == "user" %}
71
- {%- set content = render_content(message.content, false)|trim %}
72
- {%- if not(content.startswith('<tool_response>') and content.endswith('</tool_response>')) %}
73
- {%- set ns.multi_step_tool = false %}
74
- {%- set ns.last_query_index = index %}
75
- {%- endif %}
76
- {%- endif %}
77
- {%- endfor %}
78
- {%- if ns.multi_step_tool %}
79
- {{- raise_exception('No user query found in messages.') }}
80
- {%- endif %}
81
- {%- for message in messages %}
82
- {%- set content = render_content(message.content, true)|trim %}
83
- {%- if message.role == "system" %}
84
- {%- if not loop.first %}
85
- {{- raise_exception('System message must be at the beginning.') }}
86
- {%- endif %}
87
- {%- elif message.role == "user" %}
88
- {{- '<|im_start|>' + message.role + '\n' + content + '<|im_end|>' + '\n' }}
89
- {%- elif message.role == "assistant" %}
90
- {%- set reasoning_content = '' %}
91
- {%- if message.reasoning_content is string %}
92
- {%- set reasoning_content = message.reasoning_content %}
93
- {%- else %}
94
- {%- if '</think>' in content %}
95
- {%- set reasoning_content = content.split('</think>')[0].rstrip('\n').split('<think>')[-1].lstrip('\n') %}
96
- {%- set content = content.split('</think>')[-1].lstrip('\n') %}
97
- {%- endif %}
98
- {%- endif %}
99
- {%- set reasoning_content = reasoning_content|trim %}
100
- {%- if loop.index0 > ns.last_query_index %}
101
- {{- '<|im_start|>' + message.role + '\n<think>\n' + reasoning_content + '\n</think>\n\n' + content }}
102
- {%- else %}
103
- {{- '<|im_start|>' + message.role + '\n' + content }}
104
- {%- endif %}
105
- {%- if message.tool_calls and message.tool_calls is iterable and message.tool_calls is not mapping %}
106
- {%- for tool_call in message.tool_calls %}
107
- {%- if tool_call.function is defined %}
108
- {%- set tool_call = tool_call.function %}
109
- {%- endif %}
110
- {%- if loop.first %}
111
- {%- if content|trim %}
112
- {{- '\n\n<tool_call>\n<function=' + tool_call.name + '>\n' }}
113
- {%- else %}
114
- {{- '<tool_call>\n<function=' + tool_call.name + '>\n' }}
115
- {%- endif %}
116
- {%- else %}
117
- {{- '\n<tool_call>\n<function=' + tool_call.name + '>\n' }}
118
- {%- endif %}
119
- {%- if tool_call.arguments is defined %}
120
- {%- for args_name, args_value in tool_call.arguments|items %}
121
- {{- '<parameter=' + args_name + '>\n' }}
122
- {%- set args_value = args_value | tojson | safe if args_value is mapping or (args_value is sequence and args_value is not string) else args_value | string %}
123
- {{- args_value }}
124
- {{- '\n</parameter>\n' }}
125
- {%- endfor %}
126
- {%- endif %}
127
- {{- '</function>\n</tool_call>' }}
128
- {%- endfor %}
129
- {%- endif %}
130
- {{- '<|im_end|>\n' }}
131
- {%- elif message.role == "tool" %}
132
- {%- if loop.previtem and loop.previtem.role != "tool" %}
133
- {{- '<|im_start|>user' }}
134
- {%- endif %}
135
- {{- '\n<tool_response>\n' }}
136
- {{- content }}
137
- {{- '\n</tool_response>' }}
138
- {%- if not loop.last and loop.nextitem.role != "tool" %}
139
- {{- '<|im_end|>\n' }}
140
- {%- elif loop.last %}
141
- {{- '<|im_end|>\n' }}
142
- {%- endif %}
143
- {%- else %}
144
- {{- raise_exception('Unexpected message role.') }}
145
- {%- endif %}
146
- {%- endfor %}
147
- {%- if add_generation_prompt %}
148
- {{- '<|im_start|>assistant\n' }}
149
- {%- if enable_thinking is defined and enable_thinking is false %}
150
- {{- '<think>\n\n</think>\n\n' }}
151
- {%- else %}
152
- {{- '<think>\n' }}
153
- {%- endif %}
154
- {%- endif %}
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/huggingface/config.json DELETED
@@ -1,109 +0,0 @@
1
- {
2
- "architectures": [
3
- "Qwen3_5ForConditionalGeneration"
4
- ],
5
- "dtype": "bfloat16",
6
- "image_token_id": 248056,
7
- "model_type": "qwen3_5",
8
- "text_config": {
9
- "attention_bias": false,
10
- "attention_dropout": 0.0,
11
- "attn_output_gate": true,
12
- "bos_token_id": null,
13
- "dtype": "bfloat16",
14
- "eos_token_id": 248044,
15
- "full_attention_interval": 4,
16
- "head_dim": 256,
17
- "hidden_act": "silu",
18
- "hidden_size": 4096,
19
- "initializer_range": 0.02,
20
- "intermediate_size": 12288,
21
- "layer_types": [
22
- "linear_attention",
23
- "linear_attention",
24
- "linear_attention",
25
- "full_attention",
26
- "linear_attention",
27
- "linear_attention",
28
- "linear_attention",
29
- "full_attention",
30
- "linear_attention",
31
- "linear_attention",
32
- "linear_attention",
33
- "full_attention",
34
- "linear_attention",
35
- "linear_attention",
36
- "linear_attention",
37
- "full_attention",
38
- "linear_attention",
39
- "linear_attention",
40
- "linear_attention",
41
- "full_attention",
42
- "linear_attention",
43
- "linear_attention",
44
- "linear_attention",
45
- "full_attention",
46
- "linear_attention",
47
- "linear_attention",
48
- "linear_attention",
49
- "full_attention",
50
- "linear_attention",
51
- "linear_attention",
52
- "linear_attention",
53
- "full_attention"
54
- ],
55
- "linear_conv_kernel_dim": 4,
56
- "linear_key_head_dim": 128,
57
- "linear_num_key_heads": 16,
58
- "linear_num_value_heads": 32,
59
- "linear_value_head_dim": 128,
60
- "mamba_ssm_dtype": "float32",
61
- "max_position_embeddings": 262144,
62
- "mlp_only_layers": [],
63
- "model_type": "qwen3_5_text",
64
- "mtp_num_hidden_layers": 0,
65
- "mtp_use_dedicated_embeddings": false,
66
- "num_attention_heads": 16,
67
- "num_hidden_layers": 32,
68
- "num_key_value_heads": 4,
69
- "pad_token_id": null,
70
- "partial_rotary_factor": 0.25,
71
- "rms_norm_eps": 1e-06,
72
- "rope_parameters": {
73
- "mrope_interleaved": true,
74
- "mrope_section": [
75
- 11,
76
- 11,
77
- 10
78
- ],
79
- "partial_rotary_factor": 0.25,
80
- "rope_theta": 10000000,
81
- "rope_type": "default"
82
- },
83
- "tie_word_embeddings": false,
84
- "use_cache": true,
85
- "vocab_size": 248320
86
- },
87
- "tie_word_embeddings": false,
88
- "transformers_version": "5.5.4",
89
- "video_token_id": 248057,
90
- "vision_config": {
91
- "deepstack_visual_indexes": [],
92
- "depth": 27,
93
- "dtype": "bfloat16",
94
- "hidden_act": "gelu_pytorch_tanh",
95
- "hidden_size": 1152,
96
- "in_channels": 3,
97
- "initializer_range": 0.02,
98
- "intermediate_size": 4304,
99
- "model_type": "qwen3_5",
100
- "num_heads": 16,
101
- "num_position_embeddings": 2304,
102
- "out_hidden_size": 4096,
103
- "patch_size": 16,
104
- "spatial_merge_size": 2,
105
- "temporal_patch_size": 2
106
- },
107
- "vision_end_token_id": 248054,
108
- "vision_start_token_id": 248053
109
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/huggingface/generation_config.json DELETED
@@ -1,13 +0,0 @@
1
- {
2
- "bos_token_id": 248044,
3
- "do_sample": true,
4
- "eos_token_id": [
5
- 248046,
6
- 248044
7
- ],
8
- "pad_token_id": 248044,
9
- "temperature": 1.0,
10
- "top_k": 20,
11
- "top_p": 0.95,
12
- "transformers_version": "5.5.4"
13
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/huggingface/processor_config.json DELETED
@@ -1,60 +0,0 @@
1
- {
2
- "image_processor": {
3
- "do_convert_rgb": true,
4
- "do_normalize": true,
5
- "do_rescale": true,
6
- "do_resize": true,
7
- "image_mean": [
8
- 0.5,
9
- 0.5,
10
- 0.5
11
- ],
12
- "image_processor_type": "Qwen2VLImageProcessor",
13
- "image_std": [
14
- 0.5,
15
- 0.5,
16
- 0.5
17
- ],
18
- "merge_size": 2,
19
- "patch_size": 16,
20
- "resample": 3,
21
- "rescale_factor": 0.00392156862745098,
22
- "size": {
23
- "longest_edge": 16777216,
24
- "shortest_edge": 65536
25
- },
26
- "temporal_patch_size": 2
27
- },
28
- "processor_class": "Qwen3VLProcessor",
29
- "video_processor": {
30
- "do_convert_rgb": true,
31
- "do_normalize": true,
32
- "do_rescale": true,
33
- "do_resize": true,
34
- "do_sample_frames": true,
35
- "fps": 2,
36
- "image_mean": [
37
- 0.5,
38
- 0.5,
39
- 0.5
40
- ],
41
- "image_std": [
42
- 0.5,
43
- 0.5,
44
- 0.5
45
- ],
46
- "max_frames": 768,
47
- "merge_size": 2,
48
- "min_frames": 4,
49
- "patch_size": 16,
50
- "resample": 3,
51
- "rescale_factor": 0.00392156862745098,
52
- "return_metadata": false,
53
- "size": {
54
- "longest_edge": 25165824,
55
- "shortest_edge": 4096
56
- },
57
- "temporal_patch_size": 2,
58
- "video_processor_type": "Qwen3VLVideoProcessor"
59
- }
60
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/huggingface/tokenizer.json DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:87a7830d63fcf43bf241c3c5242e96e62dd3fdc29224ca26fed8ea333db72de4
3
- size 19989343
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/huggingface/tokenizer_config.json DELETED
@@ -1,32 +0,0 @@
1
- {
2
- "add_prefix_space": false,
3
- "audio_bos_token": "<|audio_start|>",
4
- "audio_eos_token": "<|audio_end|>",
5
- "audio_token": "<|audio_pad|>",
6
- "backend": "tokenizers",
7
- "bos_token": null,
8
- "clean_up_tokenization_spaces": false,
9
- "eos_token": "<|im_end|>",
10
- "errors": "replace",
11
- "image_token": "<|image_pad|>",
12
- "is_local": true,
13
- "model_max_length": 262144,
14
- "model_specific_special_tokens": {
15
- "audio_bos_token": "<|audio_start|>",
16
- "audio_eos_token": "<|audio_end|>",
17
- "audio_token": "<|audio_pad|>",
18
- "image_token": "<|image_pad|>",
19
- "video_token": "<|video_pad|>",
20
- "vision_bos_token": "<|vision_start|>",
21
- "vision_eos_token": "<|vision_end|>"
22
- },
23
- "pad_token": "<|endoftext|>",
24
- "pretokenize_regex": "(?i:'s|'t|'re|'ve|'m|'ll|'d)|[^\\r\\n\\p{L}\\p{N}]?[\\p{L}\\p{M}]+|\\p{N}| ?[^\\s\\p{L}\\p{M}\\p{N}]+[\\r\\n]*|\\s*[\\r\\n]+|\\s+(?!\\S)|\\s+",
25
- "processor_class": "Qwen3VLProcessor",
26
- "split_special_tokens": false,
27
- "tokenizer_class": "TokenizersBackend",
28
- "unk_token": null,
29
- "video_token": "<|video_pad|>",
30
- "vision_bos_token": "<|vision_start|>",
31
- "vision_eos_token": "<|vision_end|>"
32
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/model_world_size_8_rank_0.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:e7a112d3ef8e9efd0c7404343835e8d1cacb68f1bf47ec9e70a6035d628bfeae
3
- size 2352797236
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/model_world_size_8_rank_1.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:929bde074b4648a5e40ea9d59f1b37f587f605044b3013c2d9663496ff270997
3
- size 2352797236
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/model_world_size_8_rank_2.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:5f191a9b777c259be21b12b8ed02585b790915f74c547f0ac2600a45ce3a598a
3
- size 2352797236
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/model_world_size_8_rank_3.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:ff8e73e975bbe1ea882acbebe7533b050a9c127beb1c39b8a5d3cfd72ffbc17c
3
- size 2352797236
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/model_world_size_8_rank_4.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:49a0e15c74448d5debb04be1f773463910f08df82e32e33fc164e1e674a1d3e2
3
- size 2352797236
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/model_world_size_8_rank_5.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:84a14b9ff89d4c3769d06de5e98af7f71b0b2c7d82226c3e9761d37af480a7db
3
- size 2352797236
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/model_world_size_8_rank_6.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:53206b46e825c183bee092c93d115c7863e368745f0b4a9340f477845fff6521
3
- size 2352797236
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/model_world_size_8_rank_7.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:5f69907e184a8e128ebd0087e9e47ebd62f2cfa47dd93b356b2b03df20317cda
3
- size 2352797236
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/optim_world_size_8_rank_0.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:dc1018349737fd19af9fe2ae532ee8aed53a1cd2a6289bd51efc0e893092b5e7
3
- size 4477429805
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/optim_world_size_8_rank_1.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:72b7df9ff763ae286ab239e33b2d6711bff9f19a16f9043418b8c68accc08cb2
3
- size 4477429805
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/optim_world_size_8_rank_2.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:fad4ce51bcfbc75a0bd746f41af31c8c44cd57974acc1a36adcc4153bf303a6b
3
- size 4477429805
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/optim_world_size_8_rank_3.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:11ed1bcf8aa7199792a3f094d8844b6840cdd53efb3fcdfd7f1ea6931b07ea75
3
- size 4477429805
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/optim_world_size_8_rank_4.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:ccb3231b0406316170091c991dd49652d43e89bcc3f8544cbe436742baf2e960
3
- size 4477429805
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/optim_world_size_8_rank_5.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:8b1e07b9349724c6f93fd95d24146c217483921a201c48db3cd1742e22527b70
3
- size 4477429805
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/optim_world_size_8_rank_6.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:92c3605f87040e3ee654db866052d94620101854bc2ee697c10a0536bc5b3dbb
3
- size 4477429805
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/actor/optim_world_size_8_rank_7.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:c2f8b77f8faef58fce850c407f800d2cd0e7c9e4388e7c7a08f6465357fb9891
3
- size 4477429805
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/grpo/global_step_1/dataloader.pt DELETED
Binary file (1.47 kB)
 
code/runs/smoke-training-9b-20260825_175949/orarl/checkpoint_tracker.json DELETED
@@ -1,6 +0,0 @@
1
- {
2
- "best_global_step": null,
3
- "best_val_reward_score": -1.0,
4
- "last_global_step": 1,
5
- "last_actor_path": "/mnt/lyh/TempSamp/EasyVideoR1/OraRL-official/runs/smoke-training-9b-20260825_175949/orarl/global_step_1/actor"
6
- }
 
 
 
 
 
 
 
code/runs/smoke-training-9b-20260825_175949/orarl/global_step_1/actor/extra_state_world_size_8_rank_0.pt DELETED
Binary file (15.2 kB)
 
code/runs/smoke-training-9b-20260825_175949/orarl/global_step_1/actor/extra_state_world_size_8_rank_1.pt DELETED
Binary file (15.2 kB)
 
code/runs/smoke-training-9b-20260825_175949/orarl/global_step_1/actor/extra_state_world_size_8_rank_2.pt DELETED
Binary file (15.2 kB)
 
code/runs/smoke-training-9b-20260825_175949/orarl/global_step_1/actor/extra_state_world_size_8_rank_3.pt DELETED
Binary file (15.2 kB)
 
code/runs/smoke-training-9b-20260825_175949/orarl/global_step_1/actor/extra_state_world_size_8_rank_4.pt DELETED
Binary file (15.2 kB)
 
code/runs/smoke-training-9b-20260825_175949/orarl/global_step_1/actor/extra_state_world_size_8_rank_5.pt DELETED
Binary file (15.2 kB)
 
code/runs/smoke-training-9b-20260825_175949/orarl/global_step_1/actor/extra_state_world_size_8_rank_6.pt DELETED
Binary file (15.2 kB)
 
code/runs/smoke-training-9b-20260825_175949/orarl/global_step_1/actor/extra_state_world_size_8_rank_7.pt DELETED
Binary file (15.2 kB)
 
code/runs/smoke-training-9b-20260825_175949/orarl/global_step_1/actor/huggingface/chat_template.jinja DELETED
@@ -1,154 +0,0 @@
1
- {%- set image_count = namespace(value=0) %}
2
- {%- set video_count = namespace(value=0) %}
3
- {%- macro render_content(content, do_vision_count, is_system_content=false) %}
4
- {%- if content is string %}
5
- {{- content }}
6
- {%- elif content is iterable and content is not mapping %}
7
- {%- for item in content %}
8
- {%- if 'image' in item or 'image_url' in item or item.type == 'image' %}
9
- {%- if is_system_content %}
10
- {{- raise_exception('System message cannot contain images.') }}
11
- {%- endif %}
12
- {%- if do_vision_count %}
13
- {%- set image_count.value = image_count.value + 1 %}
14
- {%- endif %}
15
- {%- if add_vision_id %}
16
- {{- 'Picture ' ~ image_count.value ~ ': ' }}
17
- {%- endif %}
18
- {{- '<|vision_start|><|image_pad|><|vision_end|>' }}
19
- {%- elif 'video' in item or item.type == 'video' %}
20
- {%- if is_system_content %}
21
- {{- raise_exception('System message cannot contain videos.') }}
22
- {%- endif %}
23
- {%- if do_vision_count %}
24
- {%- set video_count.value = video_count.value + 1 %}
25
- {%- endif %}
26
- {%- if add_vision_id %}
27
- {{- 'Video ' ~ video_count.value ~ ': ' }}
28
- {%- endif %}
29
- {{- '<|vision_start|><|video_pad|><|vision_end|>' }}
30
- {%- elif 'text' in item %}
31
- {{- item.text }}
32
- {%- else %}
33
- {{- raise_exception('Unexpected item type in content.') }}
34
- {%- endif %}
35
- {%- endfor %}
36
- {%- elif content is none or content is undefined %}
37
- {{- '' }}
38
- {%- else %}
39
- {{- raise_exception('Unexpected content type.') }}
40
- {%- endif %}
41
- {%- endmacro %}
42
- {%- if not messages %}
43
- {{- raise_exception('No messages provided.') }}
44
- {%- endif %}
45
- {%- if tools and tools is iterable and tools is not mapping %}
46
- {{- '<|im_start|>system\n' }}
47
- {{- "# Tools\n\nYou have access to the following functions:\n\n<tools>" }}
48
- {%- for tool in tools %}
49
- {{- "\n" }}
50
- {{- tool | tojson }}
51
- {%- endfor %}
52
- {{- "\n</tools>" }}
53
- {{- '\n\nIf you choose to call a function ONLY reply in the following format with NO suffix:\n\n<tool_call>\n<function=example_function_name>\n<parameter=example_parameter_1>\nvalue_1\n</parameter>\n<parameter=example_parameter_2>\nThis is the value for the second parameter\nthat can span\nmultiple lines\n</parameter>\n</function>\n</tool_call>\n\n<IMPORTANT>\nReminder:\n- Function calls MUST follow the specified format: an inner <function=...></function> block must be nested within <tool_call></tool_call> XML tags\n- Required parameters MUST be specified\n- You may provide optional reasoning for your function call in natural language BEFORE the function call, but NOT after\n- If there is no function call available, answer the question like normal with your current knowledge and do not tell the user about function calls\n</IMPORTANT>' }}
54
- {%- if messages[0].role == 'system' %}
55
- {%- set content = render_content(messages[0].content, false, true)|trim %}
56
- {%- if content %}
57
- {{- '\n\n' + content }}
58
- {%- endif %}
59
- {%- endif %}
60
- {{- '<|im_end|>\n' }}
61
- {%- else %}
62
- {%- if messages[0].role == 'system' %}
63
- {%- set content = render_content(messages[0].content, false, true)|trim %}
64
- {{- '<|im_start|>system\n' + content + '<|im_end|>\n' }}
65
- {%- endif %}
66
- {%- endif %}
67
- {%- set ns = namespace(multi_step_tool=true, last_query_index=messages|length - 1) %}
68
- {%- for message in messages[::-1] %}
69
- {%- set index = (messages|length - 1) - loop.index0 %}
70
- {%- if ns.multi_step_tool and message.role == "user" %}
71
- {%- set content = render_content(message.content, false)|trim %}
72
- {%- if not(content.startswith('<tool_response>') and content.endswith('</tool_response>')) %}
73
- {%- set ns.multi_step_tool = false %}
74
- {%- set ns.last_query_index = index %}
75
- {%- endif %}
76
- {%- endif %}
77
- {%- endfor %}
78
- {%- if ns.multi_step_tool %}
79
- {{- raise_exception('No user query found in messages.') }}
80
- {%- endif %}
81
- {%- for message in messages %}
82
- {%- set content = render_content(message.content, true)|trim %}
83
- {%- if message.role == "system" %}
84
- {%- if not loop.first %}
85
- {{- raise_exception('System message must be at the beginning.') }}
86
- {%- endif %}
87
- {%- elif message.role == "user" %}
88
- {{- '<|im_start|>' + message.role + '\n' + content + '<|im_end|>' + '\n' }}
89
- {%- elif message.role == "assistant" %}
90
- {%- set reasoning_content = '' %}
91
- {%- if message.reasoning_content is string %}
92
- {%- set reasoning_content = message.reasoning_content %}
93
- {%- else %}
94
- {%- if '</think>' in content %}
95
- {%- set reasoning_content = content.split('</think>')[0].rstrip('\n').split('<think>')[-1].lstrip('\n') %}
96
- {%- set content = content.split('</think>')[-1].lstrip('\n') %}
97
- {%- endif %}
98
- {%- endif %}
99
- {%- set reasoning_content = reasoning_content|trim %}
100
- {%- if loop.index0 > ns.last_query_index %}
101
- {{- '<|im_start|>' + message.role + '\n<think>\n' + reasoning_content + '\n</think>\n\n' + content }}
102
- {%- else %}
103
- {{- '<|im_start|>' + message.role + '\n' + content }}
104
- {%- endif %}
105
- {%- if message.tool_calls and message.tool_calls is iterable and message.tool_calls is not mapping %}
106
- {%- for tool_call in message.tool_calls %}
107
- {%- if tool_call.function is defined %}
108
- {%- set tool_call = tool_call.function %}
109
- {%- endif %}
110
- {%- if loop.first %}
111
- {%- if content|trim %}
112
- {{- '\n\n<tool_call>\n<function=' + tool_call.name + '>\n' }}
113
- {%- else %}
114
- {{- '<tool_call>\n<function=' + tool_call.name + '>\n' }}
115
- {%- endif %}
116
- {%- else %}
117
- {{- '\n<tool_call>\n<function=' + tool_call.name + '>\n' }}
118
- {%- endif %}
119
- {%- if tool_call.arguments is defined %}
120
- {%- for args_name, args_value in tool_call.arguments|items %}
121
- {{- '<parameter=' + args_name + '>\n' }}
122
- {%- set args_value = args_value | tojson | safe if args_value is mapping or (args_value is sequence and args_value is not string) else args_value | string %}
123
- {{- args_value }}
124
- {{- '\n</parameter>\n' }}
125
- {%- endfor %}
126
- {%- endif %}
127
- {{- '</function>\n</tool_call>' }}
128
- {%- endfor %}
129
- {%- endif %}
130
- {{- '<|im_end|>\n' }}
131
- {%- elif message.role == "tool" %}
132
- {%- if loop.previtem and loop.previtem.role != "tool" %}
133
- {{- '<|im_start|>user' }}
134
- {%- endif %}
135
- {{- '\n<tool_response>\n' }}
136
- {{- content }}
137
- {{- '\n</tool_response>' }}
138
- {%- if not loop.last and loop.nextitem.role != "tool" %}
139
- {{- '<|im_end|>\n' }}
140
- {%- elif loop.last %}
141
- {{- '<|im_end|>\n' }}
142
- {%- endif %}
143
- {%- else %}
144
- {{- raise_exception('Unexpected message role.') }}
145
- {%- endif %}
146
- {%- endfor %}
147
- {%- if add_generation_prompt %}
148
- {{- '<|im_start|>assistant\n' }}
149
- {%- if enable_thinking is defined and enable_thinking is false %}
150
- {{- '<think>\n\n</think>\n\n' }}
151
- {%- else %}
152
- {{- '<think>\n' }}
153
- {%- endif %}
154
- {%- endif %}