Views
No views yet
0.8.0.dev01#base_model: ertghiu256/qwen3-4b-code-reasoning
2base_model: Qwen/Qwen3-4B
3model_type: AutoModelForCausalLM
4tokenizer_type: AutoTokenizer
5
6# QLoRA
7adapter: lora
8lora_r: 32
9lora_alpha: 64
10lora_dropout: 0.05
11lora_target_modules: [q_proj, k_proj, v_proj, o_proj, gate_proj, up_proj, down_proj]
12
13datasets:
14 - path: laurent-maille/pcl-test-S17
15 type: chat_template
16 field_messages: messages
17 conversation: chat
18dataset_prepared_path: ./prepared/plc_sharegpt
19output_dir: ./outputs/valuoty-indus-plc-4B
20
21# SFT propre
22train_on_inputs: false
23mask_user_tokens: true
24
25# Séquences
26sequence_len: 2048
27sample_packing: true
28pad_to_sequence_len: false
29group_by_length: true
30flash_attn_impl: fa2 # "none" si FA2 non dispo
31
32# Quantization / dtypes
33load_in_4bit: true
34load_in_8bit: false
35bnb_4bit_quant_type: nf4
36bnb_4bit_use_double_quant: true
37bnb_4bit_compute_dtype: bfloat16
38bf16: true
39fp16: false
40
41# Optim & training
42optimizer: adamw_bnb_8bit
43micro_batch_size: 8 # 8–12 selon VRAM
44gradient_accumulation_steps: 16 # effectif ≈ 192 (12×16)
45gradient_checkpointing: true
46max_grad_norm: 0.3
47learning_rate: 8.0e-5 # 5e-5 si tu veux lisser
48weight_decay: 0.0
49lr_scheduler_type: cosine
50warmup_ratio: 0.03
51num_train_epochs: 1
52seed: 42
53
54# Eval / logs / save
55val_set_size: 2000
56evaluation_strategy: steps
57eval_steps: 500
58logging_steps: 20
59load_best_model_at_end: true
60metric_for_best_model: loss
61save_strategy: steps
62save_steps: 1500
63save_total_limit: 4
64save_safetensors: true
65| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| No log | 0.0667 | 1 | 3.8828 |