Views
No views yet
0.5.01# /home/az/Bureau/FINE_TUNING_VASTAI_CLM+SFT_SUR_BASE_MODEL/01_DOCKER_finetune/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/dataset41CLM
16 data_files: dataset41CLM.jsonl
17 type: completion
18 field: text
19
20dataset_prepared_path: /workspace/prepared_data
21val_set_size: 0.01
22output_dir: /workspace/output
23
24# Séquence et packing
25sequence_len: 2048
26sample_packing: true
27pad_to_sequence_len: true
28
29# LoRA
30adapter: lora
31lora_r: 64
32lora_alpha: 128
33lora_dropout: 0.05
34lora_target_modules:
35 - q_proj
36 - k_proj
37 - v_proj
38 - o_proj
39 - gate_proj
40 - up_proj
41 - down_proj
42
43# Training
44gradient_accumulation_steps: 2
45micro_batch_size: 8
46num_epochs: 1
47learning_rate: 1e-5
48lr_scheduler: constant_with_warmup
49
50warmup_ratio: 0.03
51optimizer: adamw_torch
52
53# Précision
54bf16: true
55fp16: false
56tf32: true
57
58# Optimisations
59flash_attention: false
60gradient_checkpointing: true
61
62# Logging et sauvegardes
63logging_steps: 10
64save_steps: 200
65save_total_limit: 6
66
67
68eval_strategy: steps
69eval_steps: 200
70
71
72# WandB
73wandb_project: Qwen2.5-14B-CLM-XIXe
74
75# Hub
76hub_model_id: philipperen55/Qwen2.5-14B-style-CLM2
77push_to_hub: true
78| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| No log | 0.0008 | 1 | 5.1648 |
| 2.1702 | 0.1554 | 200 | 2.1176 |
| 1.9953 | 0.3108 | 400 | 1.9794 |
| 1.9666 | 0.4662 | 600 | 1.9497 |
| 1.9544 | 0.6216 | 800 | 1.9369 |
| 1.9449 | 0.7770 | 1000 | 1.9290 |
| 1.9414 | 0.9324 | 1200 | 1.9232 |