Views
No views yet
0.13.0.dev01base_model: Qwen/Qwen3-8B
2# Automatically upload checkpoint and final model to HF
3# hub_model_id: username/custom_model_name
4
5load_in_8bit: false
6load_in_4bit: true
7strict: false
8
9datasets:
10 - path: /workspace/outputs/training_data/
11 ds_type: json
12 data_files:
13 - knowledge.json
14 type: concisechoice
15dataset_prepared_path:
16val_set_size: 0.05
17output_dir: /workspace/outputs/FT_results
18
19sequence_len: 2048
20sample_packing: true
21eval_sample_packing: true
22
23
24adapter: lora
25lora_model_dir:
26lora_r: 32
27lora_alpha: 64
28lora_dropout: 0.05
29lora_target_linear: true
30
31wandb_project:
32wandb_entity:
33wandb_watch:
34wandb_name:
35wandb_log_model:
36
37gradient_accumulation_steps: 4
38micro_batch_size: 1
39num_epochs: 1
40optimizer: adamw_torch_fused
41lr_scheduler: cosine
42learning_rate: 0.0002
43
44bf16: auto
45tf32: true
46
47gradient_checkpointing: true
48gradient_checkpointing_kwargs:
49 use_reentrant: false
50resume_from_checkpoint:
51logging_steps: 1
52flash_attention: true
53
54warmup_ratio: 0.1
55evals_per_epoch: 4
56saves_per_epoch: 1
57weight_decay: 0.0
58#fsdp:
59# - full_shard
60# - auto_wrap
61#fsdp_config:
62# fsdp_limit_all_gathers: true
63# fsdp_sync_module_states: true
64# fsdp_offload_params: true
65# fsdp_use_orig_params: false
66# fsdp_cpu_ram_efficient_loading: true
67# fsdp_auto_wrap_policy: TRANSFORMER_BASED_WRAP
68# fsdp_transformer_layer_cls_to_wrap: Qwen3DecoderLayer
69# fsdp_state_dict_type: FULL_STATE_DICT
70# fsdp_sharding_strategy: FULL_SHARD
71special_tokens:
72
73# save_first_step: true # uncomment this to validate checkpoint saving works with your config| Training Loss | Epoch | Step | Validation Loss | Active (gib) | Allocated (gib) | Reserved (gib) |
|---|---|---|---|---|---|---|
| No log | 0 | 0 | 19.8155 | 9.21 | 9.21 | 9.35 |
| 9.8484 | 0.25 | 2 | 5.2837 | 9.3 | 9.3 | 13.95 |
| 1.4784 | 0.5 | 4 | 0.4333 | 9.3 | 9.3 | 13.95 |
| 0.7341 | 0.75 | 6 | 0.4041 | 9.3 | 9.3 | 13.95 |
| 0.5036 | 1.0 | 8 | 0.4761 | 9.3 | 9.3 | 13.95 |