frankenstallm / source /configs /korean_3b_sft.yaml
somebody-to-love's picture
Upload folder using huggingface_hub
da19444 verified
Raw
History Blame
1.39 kB
# Korean 3B SFT Configuration
#
# Base model: checkpoints/korean_3b_fp8_run1/checkpoint-XXXXXX (3B params pretrained)
# SFT λͺ©ν‘œ: instruction following + 반볡 퇴화 μ™„ν™” + 생성 ν’ˆμ§ˆ ν–₯상
# μ•„ν‚€ν…μ²˜: LLaMA-3 3B μ°Έκ³  (d=3072, 28L, 24H, GQA 8:1)
#
# μ‹€ν–‰: bash scripts/launch_3b_sft.sh
#
# [섀계 κ·Όκ±° β€” 2026-03-02]
# - batch: 2 Γ— 8GPU Γ— 4 grad_accum = 64 eff_batch
# - max_steps 33000 β‰ˆ 3 epochs Γ— 700K samples / 64 eff_batch
# - lr=1e-5: pretrain 1.5e-4의 1/15 (catastrophic forgetting λ°©μ§€)
# - NEFTune alpha=5.0: 생성 λ‹€μ–‘μ„± ν–₯상, 반볡 퇴화 μ™„ν™”
# - use_fp8=true: B200 MXFP8 λ„€μ΄ν‹°λΈŒ 가속 μœ μ§€
model:
vocab_size: 64000
d_model: 3072
n_layers: 28
n_heads: 24
n_kv_heads: 8
d_ffn: 8192
max_seq_len: 4096
rope_theta: 500000.0
dropout: 0.0
bias: false
use_flash_attn: true
use_fp8: true
train:
max_steps: 33000 # 3 epochs Γ— 700K / 64 eff_batch
batch_size: 2 # per GPU (3B VRAM μ ˆμ•½)
grad_accum_steps: 4 # eff_batch: 2 Γ— 8GPU Γ— 4 = 64
lr: 1.0e-5 # catastrophic forgetting λ°©μ§€
weight_decay: 0.01
warmup_steps: 500
max_grad_norm: 1.0
log_interval: 10
save_interval: 2000
eval_interval: 500
use_amp: false
compile_model: false
neftune_alpha: 5.0 # NEFTune noise injection
tokenizer:
vocab_size: 64000
type: sentencepiece_unigram