Views
No views yet
0.5.01# /home/az/Bureau/FINE_TUNING_VASTAI_CLM+SFT_SUR_BASE_MODEL/02-fineTune-CLM/axolotl_config.yaml
2
3base_model: Qwen/Qwen2.5-14B
4model_type: AutoModelForCausalLM
5tokenizer_type: AutoTokenizer
6trust_remote_code: true
7
8seed: 42
9
10load_in_8bit: false
11load_in_4bit: false
12
13# Dataset CLM pur
14datasets:
15 - path: philipperen55/dataset50CFT
16 data_files: dataset50CFT.jsonl
17 type: completion
18 field: text
19
20dataset_prepared_path: /workspace/prepared_data
21val_set_size: 0.003
22output_dir: /workspace/output
23
24# Séquence et packing
25sequence_len: 2048
26sample_packing: false
27pad_to_sequence_len: false
28group_by_length: true
29train_on_eos: true
30
31# cela fonctionne pour Qwen, il met automatiquement le eot # refaire un test pour ministal, et sans doute mettre append_concat_token: true !!!!
32append_concat_token: false
33add_eos_token: false
34
35# LoRA
36adapter: lora
37lora_r: 128
38lora_alpha: 256
39lora_dropout: 0.05
40lora_target_modules:
41 - q_proj
42 - k_proj
43 - v_proj
44 - o_proj
45 - gate_proj
46 - up_proj
47 - down_proj
48
49# Training
50micro_batch_size: 3
51gradient_accumulation_steps: 32
52num_epochs: 1
53learning_rate: 2e-4
54lr_scheduler: constant_with_warmup
55
56warmup_ratio: 0.03
57optimizer: adamw_torch_fused
58
59# Précision
60bf16: true
61fp16: false
62tf32: true
63
64weight_decay: 0.0
65max_grad_norm: 1.0
66
67flash_attention: true
68sdp_attention: false
69
70overrides_of_model_config:
71 attn_implementation: flash_attention_2 # forcer FA2 jusqu’au modèle
72 use_cache: false
73
74
75gradient_checkpointing: false #true
76
77# Logging et sauvegardes
78logging_steps: 10
79save_steps: 500
80save_total_limit: 1
81save_only_model: true
82save_safetensors: true
83
84
85eval_strategy: steps
86eval_steps: 500
87
88
89# WandB
90wandb_project: Qwen2.5-14B-CPT-lora
91
92# Hub
93hub_model_id: philipperen55/Qwen2.5-14B-CPT-lora
94push_to_hub: true
95hub_strategy: every_save
96| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| No log | 0.0003 | 1 | 1.9523 |
| 1.9179 | 0.1352 | 500 | 1.9364 |
| 1.9242 | 0.2704 | 1000 | 1.9267 |
| 1.9127 | 0.4056 | 1500 | 1.9215 |
| 1.9229 | 0.5408 | 2000 | 1.9142 |
| 1.9044 | 0.6761 | 2500 | 1.9114 |
| 1.9005 | 0.8113 | 3000 | 1.9072 |
| 1.9068 | 0.9465 | 3500 | 1.9034 |