Views
No views yet
1docker run --rm --runtime nvidia --ipc=host --gpus 'all' \
2 -v /data/huggingface:/root/.cache/huggingface \
3 -v /data:/data \
4 -e "HUGGING_FACE_HUB_TOKEN=$HF_TOKEN" \
5 -p 4000:4000 \
6 vllm/vllm-openai:latest \
7 --model qwen/qwen3-4b \
8 --enforce-eager --port 4000 --served-model-name base \
9 --enable-auto-tool-choice --tool-call-parser hermes \
10 --enable-lora --max-lora-rank 128 --lora-modules tool=minpeter/LoRA-Qwen3-4b-v1-iteration-01-sf-apigen-00🔍 Running test: irrelevance
✅ Test completed: irrelevance. 🎯 Accuracy: 0.875
🔍 Running test: multi_turn_base
✅ Test completed: multi_turn_base. 🎯 Accuracy: 0.085
🔍 Running test: parallel_multiple
✅ Test completed: parallel_multiple. 🎯 Accuracy: 0.89
🔍 Running test: parallel
✅ Test completed: parallel. 🎯 Accuracy: 0.885
🔍 Running test: simple
✅ Test completed: simple. 🎯 Accuracy: 0.9325
🔍 Running test: multiple
✅ Test completed: multiple. 🎯 Accuracy: 0.920.9.21base_model: Qwen/Qwen3-4B
2hub_model_id: minpeter/LoRA-Qwen3-4b-v1-iteration-01-sf-apigen-00
3
4load_in_8bit: false
5load_in_4bit: false
6strict: false
7
8datasets:
9 # 0.5k
10 - path: minpeter/apigen-mt-5k-friendli
11 data_files:
12 - train.jsonl
13 - test.jsonl
14 type: chat_template
15 chat_template: qwen3
16 split_thinking: true
17 field_messages: messages
18 message_field_role: role
19 message_field_content: content
20 shards: 3
21
22chat_template: qwen3
23
24dataset_prepared_path: last_run_prepared
25
26output_dir: ./output
27
28adapter: lora
29lora_model_dir:
30
31sequence_len: 16384
32pad_to_sequence_len: true
33sample_packing: true
34
35val_set_size: 0.05
36eval_sample_packing: true
37evals_per_epoch: 3
38
39lora_r: 8
40lora_alpha: 16
41lora_dropout: 0.05
42lora_fan_in_fan_out:
43lora_target_modules:
44 - gate_proj
45 - down_proj
46 - up_proj
47 - q_proj
48 - v_proj
49 - k_proj
50 - o_proj
51
52wandb_project: "axolotl"
53wandb_entity: "kasfiekfs-e"
54wandb_watch:
55wandb_name:
56wandb_log_model:
57
58gradient_accumulation_steps: 2
59micro_batch_size: 2
60num_epochs: 2
61optimizer: adamw_8bit
62lr_scheduler: cosine
63learning_rate: 0.0002
64
65train_on_inputs: false
66group_by_length: false
67bf16: auto
68tf32: true
69
70gradient_checkpointing: true
71early_stopping_patience:
72resume_from_checkpoint:
73local_rank:
74logging_steps: 1
75xformers_attention:
76flash_attention: true
77
78loss_watchdog_threshold: 5.0
79loss_watchdog_patience: 3
80
81warmup_steps: 10
82saves_per_epoch: 1
83debug:
84deepspeed:
85weight_decay: 0.0
86fsdp:
87fsdp_config:
88| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 0.929 | 0.0045 | 1 | 0.8641 |
| 0.3071 | 0.3341 | 74 | 0.2398 |
| 0.1946 | 0.6682 | 148 | 0.2112 |
| 0.1311 | 1.0 | 222 | 0.1951 |
| 0.1204 | 1.3341 | 296 | 0.1865 |
| 0.1926 | 1.6682 | 370 | 0.1821 |