Views
No views yet
0.4.11base_model: Qwen/Qwen2.5-14B-Instruct
2
3load_in_8bit: false
4load_in_4bit: false
5strict: false
6
7datasets:
8 - path: output.jsonl
9 type:
10 field_instruction: instruction
11 field_input: input
12 field_output: output
13 format: "<|im_start|>system\n{instruction}<|im_end|>\n<|im_start|>user\n{input}<|im_end|>\n<|im_start|>assistant\n"
14
15special_tokens:
16 bos_token:
17 eos_token: "<|im_end|>"
18 pad_token: "<|endoftext|>"
19
20dataset_prepared_path:
21val_set_size: 0.05
22output_dir: ./outputs/lora-out
23
24sequence_len: 4096
25sample_packing: false
26pad_to_sequence_len: true
27
28adapter: lora
29lora_model_dir:
30lora_r: 8
31lora_alpha: 16
32lora_dropout: 0.05
33lora_target_linear: true
34lora_fan_in_fan_out:
35lora_target_modules:
36 - gate_proj
37 - down_proj
38 - up_proj
39 - q_proj
40 - v_proj
41 - k_proj
42 - o_proj
43
44wandb_project: mssong_axolotl
45wandb_entity: mssong
46wandb_watch:
47wandb_run_id:
48wandb_log_model:
49
50gradient_accumulation_steps: 2
51micro_batch_size: 1
52num_epochs: 3
53optimizer:
54lr_scheduler: cosine
55learning_rate: 0.00005
56train_on_inputs:
57group_by_length: false
58bf16: true
59fp16: false
60tf32: false
61gradient_checkpointing: true
62early_stopping_patience: 3
63local_rank:
64logging_steps: 10
65xformers_attention:
66flash_attention: true
67#warmup_ratio: 0.02
68warmup_steps: 100
69eval_steps: 100
70save_steps: 500
71save_total_limit: 2
72eval_sample_packing:
73debug:
74deepspeed:
75weight_decay: 0.1
76fsdp:
77fsdp_config:
78trust_remote_code: true| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| No log | 0.0035 | 1 | 0.6979 |
| 0.046 | 0.3515 | 100 | 0.0793 |
| 0.0259 | 0.7030 | 200 | 0.0519 |
| 0.0242 | 1.0545 | 300 | 0.0447 |
| 0.0194 | 1.4060 | 400 | 0.0435 |
| 0.016 | 1.7575 | 500 | 0.0427 |
| 0.0097 | 2.1090 | 600 | 0.0392 |
| 0.0179 | 2.4605 | 700 | 0.0410 |
| 0.0081 | 2.8120 | 800 | 0.0405 |