dpo_tiv2t_10k_baseline / arguments.yaml
htlou's picture
Upload folder using huggingface_hub
5f3fc63 verified
raw
history blame
1.27 kB
data_cfgs:
eval_data_files: null
eval_datasets: null
eval_optional_args: []
eval_size: null
eval_split: null
eval_subset: null
eval_template: null
train_data_files: extracted_preference_10k_washed.json
train_datasets: /aifs4su/yaodong/datasets/aaa_dataset/TV2T-preference/extracted
train_optional_args: []
train_size: null
train_split: train
train_subset: null
train_template: NExTQA_preference
logger_cfgs:
cache_dir: null
log_project: align-anything
log_run_name: dpo
log_type: wandb
output_dir: ../outputs/dpo_tiv2t_10k_baseline
save_interval: 100000
model_cfgs:
model_max_length: 4096
model_name_or_path: /aifs4su/yaodong/models/Qwen2-VL-7B-Instruct
trust_remote_code: true
special_tokens: null
train_cfgs:
adam_betas:
- 0.9
- 0.95
bf16: true
ds_cfgs: ds_z3_config.json
epochs: 3
eval_interval: 10
eval_strategy: epoch
fp16: false
freeze_language_model: false
freeze_mm_proj: false
freeze_vision_tower: true
gradient_accumulation_steps: 1.0
gradient_checkpointing: true
learning_rate: 1.0e-06
lr_scheduler_type: cosine
lr_warmup_ratio: 0.01
per_device_eval_batch_size: 1.0
per_device_train_batch_size: 1.0
regularization: 0.001
scale_coeff: 0.1
seed: 42
weight_decay: 0.0