Views
No views yet
0.9.01base_model: THUDM/GLM-4-32B-0414
2#base_model_ignore_patterns: "*/*"
3# optionally might have model_type or tokenizer_type
4model_type: AutoModelForCausalLM
5tokenizer_type: AutoTokenizer
6
7load_in_8bit: false
8load_in_4bit: false
9strict: false
10
11datasets:
12 - path: anthracite-core/magnum-v5-sft-proto-glm4-instruct-rev1
13 ds_type: parquet
14 type:
15shuffle_merged_datasets: true
16dataset_prepared_path: ./data/magnum-32b-data
17val_set_size: 0.0025
18output_dir: ./data/32b-lora-out
19
20plugins:
21 - axolotl.integrations.liger.LigerPlugin
22 - axolotl.integrations.cut_cross_entropy.CutCrossEntropyPlugin
23#liger_rope: false
24liger_rms_norm: true
25liger_layer_norm: true
26liger_glu_activation: true
27liger_fused_linear_cross_entropy: true
28cut_cross_entropy: true
29
30sequence_len: 32768
31sample_packing: true
32eval_sample_packing: false
33pad_to_sequence_len: true
34
35adapter: lora
36lora_model_dir:
37lora_r: 128
38lora_alpha: 16
39lora_dropout: 0.05
40lora_target_linear: true
41lora_fan_in_fan_out:
42peft_use_rslora: true
43lora_modules_to_save:
44 - embed_tokens
45 - lm_head
46
47wandb_project: 32b-magnum-lora
48wandb_entity:
49wandb_watch:
50wandb_name: run7-lora-0.02-clip-4.5e-5
51wandb_log_model:
52
53gradient_accumulation_steps: 2
54micro_batch_size: 1
55num_epochs: 2
56optimizer: adamw_bnb_8bit
57lr_scheduler: rex
58learning_rate: 4.5e-5
59max_grad_norm: 0.02
60
61train_on_inputs: false
62group_by_length: false
63bf16: auto
64fp16:
65tf32: false
66
67gradient_checkpointing: true
68early_stopping_patience:
69resume_from_checkpoint:
70local_rank:
71logging_steps: 1
72xformers_attention:
73flash_attention: true
74s2_attention:
75
76warmup_steps: 40
77evals_per_epoch: 4
78eval_table_size:
79eval_max_new_tokens:
80saves_per_epoch: 2
81debug:
82deepspeed: ./deepspeed_configs/zero3_bf16.json
83weight_decay: 0.0025
84fsdp:
85fsdp_config:
86special_tokens:
87| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 1.348 | 0.0024 | 1 | 1.3957 |
| 1.1269 | 0.2506 | 104 | 1.1851 |
| 1.1548 | 0.5012 | 208 | 1.1661 |
| 1.0896 | 0.7518 | 312 | 1.1531 |
| 1.0712 | 1.0024 | 416 | 1.1447 |
| 1.0552 | 1.2530 | 520 | 1.1362 |
| 1.0567 | 1.5036 | 624 | 1.1275 |
| 1.0888 | 1.7542 | 728 | 1.1206 |