Video-ORA-9B / code /configs /data_sources.example.yaml
lyhisme's picture
Add OraRL training and evaluation code
53c10a4 verified
Raw
History Blame Contribute Delete
3.16 kB
# Copy this file, then replace every ../local_data placeholder.
#
# OraRL never downloads data or media. You must obtain each dataset under its
# license, keep the required attribution, and point input/media_root at your
# own local, licensed copy. The optional license and url values are audit
# metadata only and are never opened by the builder.
#
# The seven quotas reproduce the final paper mixture: 100,032 train prompts
# (1,563 batches of 64), with 62,656 structured prompts and 37,376 answer-only
# prompts. The canary is
# additional held-out data and is not subtracted from the train target.
version: 1
seed: 42
target: 100032
canary_size: 512
max_prompts_per_media: 2
require_media: true
allow_shortfall: false
# Supply JSON/JSONL benchmark records before a release build. Any candidate
# sharing a normalized prompt identity or media identity is removed.
benchmark_excludes:
- ../local_data/exclusions/public_benchmarks.jsonl
sources:
- name: temporal_grounding_train
input: ../local_data/annotations/temporal_grounding.jsonl
task: temporal grounding
family: temporal
quota: 20096
media_root: ../local_data/media/temporal
license: REPLACE_WITH_DATASET_LICENSE
url: REPLACE_WITH_DATASET_HOME_PAGE
- name: tracking_train
input: ../local_data/annotations/tracking.jsonl
task: tracking
family: tracking
quota: 13952
media_root: ../local_data/media/tracking
license: REPLACE_WITH_DATASET_LICENSE
url: REPLACE_WITH_DATASET_HOME_PAGE
- name: segmentation_train
input: ../local_data/annotations/segmentation.jsonl
task: segmentation
family: segmentation
quota: 12032
media_root: ../local_data/media/segmentation
license: REPLACE_WITH_DATASET_LICENSE
url: REPLACE_WITH_DATASET_HOME_PAGE
- name: spatial_grounding_train
input: ../local_data/annotations/spatial_grounding.jsonl
task: spatial grounding
family: spatial
quota: 7040
media_root: ../local_data/media/spatial
license: REPLACE_WITH_DATASET_LICENSE
url: REPLACE_WITH_DATASET_HOME_PAGE
- name: spatial_temporal_grounding_train
input: ../local_data/annotations/spatial_temporal_grounding.jsonl
task: spatial-temporal grounding
family: stvg
quota: 9536
media_root: ../local_data/media/spatial_temporal
license: REPLACE_WITH_DATASET_LICENSE
url: REPLACE_WITH_DATASET_HOME_PAGE
- name: video_qa_train
input: ../local_data/annotations/video_qa.jsonl
task: video_qa_mc
family: video_qa
quota: 20288
media_root: ../local_data/media/video_qa
license: REPLACE_WITH_DATASET_LICENSE
url: REPLACE_WITH_DATASET_HOME_PAGE
- name: spatial_intelligence_train
input: ../local_data/annotations/spatial_intelligence.jsonl
task: spatial intelligence
family: spatial_intelligence
# Keep subtype labels such as object_counting and object_rel_direction;
# the reward adapter uses them to select the official scoring rule.
preserve_problem_type: true
quota: 17088
media_root: ../local_data/media/spatial_intelligence
license: REPLACE_WITH_DATASET_LICENSE
url: REPLACE_WITH_DATASET_HOME_PAGE