Views
No views yet
0.10.0.dev01base_model: Qwen/Qwen2.5-coder-3B
2model_type: AutoModelForCausalLM
3tokenizer_type: AutoTokenizer
4trust_remote_code: true
5chat_template: qwen_25
6
7adapter: qlora
8lora_r: 8
9lora_alpha: 32
10lora_dropout: 0.05
11lora_target_modules:
12 - c_attn
13 - c_proj
14 - w1
15 - w2
16 - q_proj
17 - v_proj
18 - k_proj
19 - o_proj
20
21load_in_4bit: true
22bnb_4bit_compute_dtype: float16
23bnb_4bit_use_double_quant: true
24bnb_4bit_quant_type: nf4
25
26datasets:
27 - path: ./datasets/generic_formatted_data.jsonl
28 type: alpaca
29
30val_set_size: 0.01
31dataset_prepared_path:
32
33sequence_len: 2048
34pad_to_sequence_len: true
35
36output_dir: ./outputs/qwen2.5-coder-3b-lora
37num_epochs: 3
38micro_batch_size: 2
39gradient_accumulation_steps: 8
40evals_per_epoch: 1
41saves_per_epoch: 1
42optimizer: adamw_bnb_8bit
43learning_rate: 2e-5
44lr_scheduler: cosine
45warmup_steps: 50
46
47gradient_checkpointing: true
48fp16: true
49bf16: false
50tf32: true
51flash_attention: true
52eager_attention: false
53
54logging_steps: 1
55debug: true
56wandb_project: qwen-coder
57wandb_name: qwen2.5-coder-3b-lora
58wandb_log_model: "false"
59wandb_mode: disabled| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 1.0456 | 0.0022 | 1 | 0.9417 |
| 0.3029 | 1.0 | 459 | 0.1403 |
| 0.044 | 2.0 | 918 | 0.0817 |