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-32B
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 # 108073MiB / 143771MiB pour 6
46num_epochs: 1
47learning_rate: 8e-6
48lr_scheduler: constant_with_warmup
49
50warmup_ratio: 0.10
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-32B-style-CLM
74
75# Hub
76hub_model_id: philipperen55/Qwen2.5-32B-style-CLM
77push_to_hub: true
78| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| No log | 0.0008 | 1 | 4.7707 |
| 2.2894 | 0.1554 | 200 | 2.2169 |
| 1.9807 | 0.3108 | 400 | 1.9656 |
| 1.9387 | 0.4662 | 600 | 1.9219 |
| 1.9196 | 0.6216 | 800 | 1.9039 |
| 1.9094 | 0.7770 | 1000 | 1.8933 |
| 1.9043 | 0.9324 | 1200 | 1.8861 |