Views
No views yet
0.10.01base_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: false
7strict: false
8
9datasets:
10 - path: tatsu-lab/alpaca
11 type: alpaca
12dataset_prepared_path:
13val_set_size: 0.05
14output_dir: ./outputs/out
15
16sequence_len: 2048
17sample_packing: true
18eval_sample_packing: true
19
20
21# adapter: qlora
22# lora_model_dir:
23# lora_r: 32
24# lora_alpha: 64
25# lora_dropout: 0.05
26# lora_target_linear: true
27
28wandb_project:
29wandb_entity:
30wandb_watch:
31wandb_name:
32wandb_log_model:
33
34gradient_accumulation_steps: 8
35micro_batch_size: 1
36num_epochs: 1
37optimizer: adamw_torch_fused
38lr_scheduler: cosine
39learning_rate: 0.0002
40
41bf16: auto
42tf32: true
43
44gradient_checkpointing: true
45gradient_checkpointing_kwargs:
46 use_reentrant: false
47resume_from_checkpoint:
48logging_steps: 1
49flash_attention: true
50
51warmup_ratio: 0.1
52evals_per_epoch: 4
53saves_per_epoch: 1
54weight_decay: 0.0
55fsdp:
56 - full_shard
57 - auto_wrap
58fsdp_config:
59 fsdp_limit_all_gathers: true
60 fsdp_sync_module_states: true
61 fsdp_offload_params: false
62 fsdp_use_orig_params: false
63 fsdp_cpu_ram_efficient_loading: true
64 fsdp_auto_wrap_policy: TRANSFORMER_BASED_WRAP
65 fsdp_transformer_layer_cls_to_wrap: Qwen3DecoderLayer
66 fsdp_state_dict_type: FULL_STATE_DICT
67 fsdp_sharding_strategy: FULL_SHARD
68special_tokens:
69
70# save_first_step: true # uncomment this to validate checkpoint saving works with your config
71| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| No log | 0 | 0 | 2.0223 |
| 1.5647 | 0.2486 | 39 | 1.5485 |
| 1.4248 | 0.4972 | 78 | 1.5067 |
| 1.4311 | 0.7458 | 117 | 1.4276 |
| 1.4508 | 0.9944 | 156 | 1.4002 |