Skip to content

Commit f378951

Browse files
committed
update
1 parent d8e829d commit f378951

5 files changed

Lines changed: 94 additions & 207 deletions

File tree

CapRL++/train/scripts/README.md

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,13 @@ Important reward variables:
6767
- `REWARD_TASK`: defaults to `video`; set to `image` for image caption training.
6868
- `REWARD_QA_NUM`: sampling rounds in `qa` mode. Defaults
6969
to `8`.
70-
- `FORMAT_REWARD_WEIGHT`: format reward weight. Defaults to `0.2`.
70+
- `FORMAT_REWARD_WEIGHT`: video timestamp format reward weight. Defaults to
71+
`0.2`. For video captions, the unweighted format reward is
72+
`0.5 * N_valid / max(N_all, 1) + 0.5 * I_chrono`, where `N_all` is the
73+
number of timestamp-like brackets matched by the regex, `N_valid` is the
74+
number that satisfy logical constraints such as valid seconds and
75+
`t_end >= t_start`, and `I_chrono` is `1` only when valid timestamp start
76+
times are monotonically non-decreasing.
7177

7278
For `REWARD_SCORE_MODE=qa`, the reward model can be a text LLM. For
7379
`REWARD_SCORE_MODE=vl_judge`, use a multimodal VLM.

CapRL++/train/scripts/start_reward_serve_rm.sh

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,6 @@ export REWARD_NUM_WORKERS="${REWARD_NUM_WORKERS:-1}"
4545

4646
# qa-mode reward settings. They are ignored by vl_judge mode.
4747
export FORMAT_REWARD_WEIGHT="${FORMAT_REWARD_WEIGHT:-0.2}"
48-
export FORMAT_MIN_BRACKETS="${FORMAT_MIN_BRACKETS:-3}"
4948
export REWARD_QA_NUM="${REWARD_QA_NUM:-8}"
5049

5150
exec bash "${VERL_ROOT}/recipe/video_captionrl/scripts/start_reward_server.sh"

0 commit comments

Comments
 (0)