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-S27
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: 10 # 8–12 selon VRAM
44gradient_accumulation_steps: 16 # effectif ≈ 192 (12×16)
45gradient_checkpointing: true
46max_grad_norm: 0.3
47#resume_from_checkpoint: ./outputs/valuoty-indus-plc-4B/checkpoint-15/
48
49learning_rate: 8.0e-5 # 5e-5 si tu veux lisser
50weight_decay: 0.0
51lr_scheduler_type: cosine
52warmup_ratio: 0.03
53num_train_epochs: 1
54seed: 42
55
56# Eval / logs / save
57val_set_size: 2000
58evaluation_strategy: steps
59eval_steps: 800
60logging_steps: 20
61load_best_model_at_end: true
62metric_for_best_model: loss
63save_strategy: steps
64save_steps: 1500
65save_total_limit: 4
66save_safetensors: true
67| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| No log | 0.0399 | 1 | 4.1037 |