1base_model: kakaocorp/kanana-1.5-2.1b-instruct-2505
2
3
4load_in_8bit: false
5load_in_4bit: false
6
7datasets:
8 - path: train.jsonl
9 type: chat_template
10
11dataset_prepared_path: preprocess
12val_set_size: 0.01
13output_dir: ./outputs
14dataloader_num_workers: 56
15
16adapter:
17lora_model_dir:
18
19sequence_len: 16384
20sample_packing: false
21eval_sample_packing: false
22pad_to_sequence_len: false
23
24plugins:
25 - axolotl.integrations.liger.LigerPlugin
26liger_rope: true
27liger_rms_norm: true
28liger_swiglu: true
29liger_fused_linear_cross_entropy: true
30
31wandb_project: fastcampus
32wandb_entity: guijinson
33wandb_watch:
34wandb_name: fc-proj2-reasoning-2.1b
35wandb_log_model:
36hub_model_id: amphora/fc-reasoning-2.1b
37
38gradient_accumulation_steps: 64
39micro_batch_size: 2
40num_epochs: 3
41optimizer: adamw_torch_fused
42lr_scheduler: cosine
43learning_rate: 2e-5
44
45bf16: auto
46tf32: false
47
48gradient_checkpointing:
49resume_from_checkpoint:
50logging_steps: 1
51flash_attention: true
52
53warmup_ratio: 0.05
54weight_decay: 0.01
55evals_per_epoch: 0
56saves_per_epoch: 1
57