初始化项目,由ModelHub XC社区提供模型
Model: pathcosmos/frankenstallm Source: Original Platform
This commit is contained in:
55
source/configs/hybrid_3b.yaml
Normal file
55
source/configs/hybrid_3b.yaml
Normal file
@@ -0,0 +1,55 @@
|
||||
# FRANKENSTALLM-H 3B: Hybrid Mamba-2 + Transformer
|
||||
#
|
||||
# [설계 근거 — 2026-03-05]
|
||||
# - 아키텍처: Nemotron-H 8B Dense 참고, 3B 스케일 적용
|
||||
# - 40 layers: 37 Mamba-2 + 3 Attention (layer 13, 26, 39)
|
||||
# - 파라미터: ~2.9B (embedding 포함)
|
||||
# - 데이터: 3b_train.bin (기존 Pure Transformer 동일 데이터)
|
||||
# - lr=2e-4: Mamba-2 논문 참고, Transformer보다 약간 높음
|
||||
# - Attention 3개: 초반(13), 중반(26), 후반(39) 균등 배치
|
||||
# - Mamba 장점: O(n) 시퀀스 처리, 추론 시 constant memory
|
||||
#
|
||||
# 실행: bash scripts/launch_hybrid_3b.sh
|
||||
model:
|
||||
vocab_size: 64000
|
||||
d_model: 3072
|
||||
n_layers: 40
|
||||
n_heads: 24
|
||||
n_kv_heads: 8
|
||||
d_ffn: 9216
|
||||
max_seq_len: 4096
|
||||
rope_theta: 500000.0
|
||||
dropout: 0.0
|
||||
bias: false
|
||||
use_flash_attn: true
|
||||
use_fp8: true
|
||||
# Hybrid settings
|
||||
use_hybrid: true
|
||||
hybrid_pattern: "M M M M M M M M M M M M M A M M M M M M M M M M M M A M M M M M M M M M M M M A"
|
||||
mamba_d_state: 128
|
||||
mamba_head_dim: 64
|
||||
mamba_expand: 2
|
||||
mamba_conv_kernel: 4
|
||||
mamba_n_groups: 1
|
||||
mamba_chunk_size: 256
|
||||
|
||||
train:
|
||||
max_steps: 57000
|
||||
batch_size: 4
|
||||
grad_accum_steps: 8
|
||||
lr: 2e-4
|
||||
weight_decay: 0.1
|
||||
warmup_steps: 2000
|
||||
max_grad_norm: 1.0
|
||||
log_interval: 10
|
||||
save_interval: 2000
|
||||
eval_interval: 500
|
||||
use_amp: false
|
||||
compile_model: false
|
||||
fp8_amax_history_len: 16
|
||||
fp8_amax_compute_algo: "max"
|
||||
fp8_format: "MXFP8"
|
||||
|
||||
tokenizer:
|
||||
vocab_size: 64000
|
||||
type: sentencepiece_unigram
|
||||
Reference in New Issue
Block a user