Views
No views yet
0.12.21# ===== Model =====
2
3base_model: Qwen/Qwen2.5-VL-7B-Instruct
4processor_type: AutoProcessor
5
6chat_template: qwen2_vl
7
8# 멀티모달(비전-챗) 필수 플래그
9skip_prepare_dataset: true
10remove_unused_columns: false
11sample_packing: false
12
13
14dataloader_num_workers: 0
15#shuffle_merged_datasets: false
16#shuffle_before_merging_datasets: false # (기본 false지만 명시 추천)
17
18ddp_find_unused_parameters: true
19
20# ===== Data =====
21datasets:
22 - path: vlm_data_2025101_1/qwen2.5-VL-7B-KoV_0.0.4.jsonl
23 type: chat_template
24 field_messages: messages
25 split: null
26
27val_set_size: 0.0
28dataset_prepared_path:
29
30# ===== Output / Logging =====
31output_dir: ./outputs/qwen2.5-VL-7B-KoV_0.0.4.jsonl
32logging_steps: 1
33
34# wandb 연동(원하면 변경/주석)
35wandb_entity: minkyun1
36wandb_project: kisti_vlm_axo
37wandb_name: qwen2.5-VL-7B-KoV_0.0.4.jsonl
38
39# ===== LoRA / Quantization =====
40#adapter: lora
41# LLaVA에서 언어모델 쪽 프로젝션에만 LoRA(안전 기본값)
42#lora_r: 256
43#lora_alpha: 512
44#lora_dropout: 0.05
45#lora_target_modules: "model.language_model.layers.[\\d]+.(mlp|cross_attn|self_attn).(up|down|gate|q|k|v|o)_proj"
46
47# 메모리 여유 충분하지만, 시작은 4bit 로 안정적으로
48load_in_4bit: false
49load_in_8bit: false
50bf16: true
51tf32: true
52gradient_checkpointing: true
53gradient_checkpointing_kwargs:
54 use_reentrant: false
55flash_attention: true
56eager_attention:
57
58# ===== Optim & Train =====
59optimizer: adamw_torch_fused
60learning_rate: 4e-5
61lr_scheduler: cosine
62warmup_ratio: 0.05
63weight_decay: 0.01
64max_grad_norm: 1.0
65seed: 42
66sequence_len: 8192
67pad_to_sequence_len: false
68excess_length_strategy: drop
69
70# GPU당 마이크로 배치/누적 → 유효 배치 = 1 * 8 * 2GPU = 16
71micro_batch_size: 1
72gradient_accumulation_steps: 16
73
74num_epochs: 5
75evals_per_epoch: 1
76saves_per_epoch: 1
77# save_first_step: true
78
79# ===== Multi-GPU: DeepSpeed (추천) =====
80# deepspeed 프리셋을 받아서 사용:
81# axolotl fetch deepspeed_configs
82# 2×A100 80GB + 7B에는 zero2가 빠르고 안정적
83deepspeed: ds_zero2.json
84
85# ===== 디버그/재현성(선택) =====
86# 데이터 전처리 멀티프로세스가 문제 생기면 1로 낮춰서 원인 파악
87# dataset_processes: 1
88
89# ===== [대안] FSDP2 설정(DeepSpeed 대신 쓰고 싶을 때) =====
90# fsdp_version: 2
91# fsdp_config:
92# offload_params: false
93# cpu_ram_efficient_loading: true
94# auto_wrap_policy: TRANSFORMER_BASED_WRAP
95# transformer_layer_cls_to_wrap: LlamaDecoderLayer
96# state_dict_type: FULL_STATE_DICT
97# reshard_after_forward: true
98