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