hardlyworking commited on
Commit
3e51f2f
·
verified ·
1 Parent(s): d21e26d

End of training

Browse files
Files changed (1) hide show
  1. README.md +176 -0
README.md ADDED
@@ -0,0 +1,176 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: Qwen/Qwen3-4B-Base
5
+ tags:
6
+ - axolotl
7
+ - generated_from_trainer
8
+ datasets:
9
+ - GreenerPastures/All-Your-Base-Full
10
+ model-index:
11
+ - name: Sugma4B
12
+ results: []
13
+ ---
14
+
15
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
16
+ should probably proofread and complete it, then remove this comment. -->
17
+
18
+ [<img src="https://raw.githubusercontent.com/axolotl-ai-cloud/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/axolotl-ai-cloud/axolotl)
19
+ <details><summary>See axolotl config</summary>
20
+
21
+ axolotl version: `0.10.0.dev0`
22
+ ```yaml
23
+ base_model: Qwen/Qwen3-4B-Base
24
+
25
+ load_in_8bit: false
26
+ load_in_4bit: false
27
+ strict: false
28
+
29
+ chat_template: qwen3
30
+ datasets:
31
+ - path: GreenerPastures/All-Your-Base-Full
32
+ type: chat_template
33
+ split: train
34
+ field_messages: conversations
35
+ message_property_mappings:
36
+ role: from
37
+ content: value
38
+ val_set_size: 0.01
39
+ output_dir: ./outputs/out
40
+ dataset_prepared_path: last_run_prepared
41
+ shuffle_merged_datasets: true
42
+
43
+ hub_model_id: hardlyworking/Sugma4B
44
+ hub_strategy: "all_checkpoints"
45
+ push_dataset_to_hub:
46
+ hf_use_auth_token: true
47
+
48
+ plugins:
49
+ - axolotl.integrations.liger.LigerPlugin
50
+ - axolotl.integrations.cut_cross_entropy.CutCrossEntropyPlugin
51
+ liger_rope: true
52
+ liger_rms_norm: true
53
+ liger_layer_norm: true
54
+ liger_glu_activation: true
55
+ liger_fused_linear_cross_entropy: false
56
+ cut_cross_entropy: true
57
+
58
+ sequence_len: 8192
59
+ sample_packing: true
60
+ eval_sample_packing: true
61
+ pad_to_sequence_len: true
62
+
63
+ wandb_project: Qwen4B
64
+ wandb_entity:
65
+ wandb_watch:
66
+ wandb_name: Qwen4B
67
+ wandb_log_model:
68
+
69
+ evals_per_epoch: 8
70
+ eval_table_size:
71
+ eval_max_new_tokens: 128
72
+
73
+ gradient_accumulation_steps: 8
74
+ micro_batch_size: 2
75
+ num_epochs: 2
76
+ optimizer: adamw_bnb_8bit
77
+ lr_scheduler: cosine
78
+ learning_rate: 1e-5
79
+
80
+ train_on_inputs: false
81
+ group_by_length: false
82
+ bf16: auto
83
+ fp16:
84
+ tf32: false
85
+
86
+ gradient_checkpointing: offload
87
+ gradient_checkpointing_kwargs:
88
+ use_reentrant: false
89
+ early_stopping_patience:
90
+ resume_from_checkpoint:
91
+ local_rank:
92
+ logging_steps: 1
93
+ xformers_attention:
94
+ flash_attention: true
95
+ s2_attention:
96
+
97
+ deepspeed:
98
+
99
+ warmup_ratio: 0.05
100
+ saves_per_epoch: 1
101
+ debug:
102
+ weight_decay: 0.01
103
+ fsdp:
104
+ fsdp_config:
105
+ special_tokens:
106
+ pad_token:
107
+ ```
108
+
109
+ </details><br>
110
+
111
+ # Sugma4B
112
+
113
+ This model is a fine-tuned version of [Qwen/Qwen3-4B-Base](https://huggingface.co/Qwen/Qwen3-4B-Base) on the GreenerPastures/All-Your-Base-Full dataset.
114
+ It achieves the following results on the evaluation set:
115
+ - Loss: 0.9300
116
+
117
+ ## Model description
118
+
119
+ More information needed
120
+
121
+ ## Intended uses & limitations
122
+
123
+ More information needed
124
+
125
+ ## Training and evaluation data
126
+
127
+ More information needed
128
+
129
+ ## Training procedure
130
+
131
+ ### Training hyperparameters
132
+
133
+ The following hyperparameters were used during training:
134
+ - learning_rate: 1e-05
135
+ - train_batch_size: 2
136
+ - eval_batch_size: 2
137
+ - seed: 42
138
+ - distributed_type: multi-GPU
139
+ - num_devices: 2
140
+ - gradient_accumulation_steps: 8
141
+ - total_train_batch_size: 32
142
+ - total_eval_batch_size: 4
143
+ - optimizer: Use OptimizerNames.ADAMW_BNB with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
144
+ - lr_scheduler_type: cosine
145
+ - lr_scheduler_warmup_steps: 52
146
+ - num_epochs: 2.0
147
+
148
+ ### Training results
149
+
150
+ | Training Loss | Epoch | Step | Validation Loss |
151
+ |:-------------:|:------:|:----:|:---------------:|
152
+ | 1.1154 | 0.0019 | 1 | 1.1372 |
153
+ | 0.9351 | 0.125 | 65 | 1.0074 |
154
+ | 0.8884 | 0.25 | 130 | 0.9758 |
155
+ | 0.9853 | 0.375 | 195 | 0.9608 |
156
+ | 0.8998 | 0.5 | 260 | 0.9490 |
157
+ | 0.8919 | 0.625 | 325 | 0.9420 |
158
+ | 0.914 | 0.75 | 390 | 0.9376 |
159
+ | 0.8873 | 0.875 | 455 | 0.9346 |
160
+ | 0.8854 | 1.0 | 520 | 0.9326 |
161
+ | 0.9365 | 1.125 | 585 | 0.9316 |
162
+ | 0.8865 | 1.25 | 650 | 0.9308 |
163
+ | 0.9696 | 1.375 | 715 | 0.9304 |
164
+ | 0.9119 | 1.5 | 780 | 0.9302 |
165
+ | 0.8793 | 1.625 | 845 | 0.9301 |
166
+ | 0.9265 | 1.75 | 910 | 0.9301 |
167
+ | 0.9375 | 1.875 | 975 | 0.9301 |
168
+ | 0.8473 | 2.0 | 1040 | 0.9300 |
169
+
170
+
171
+ ### Framework versions
172
+
173
+ - Transformers 4.51.3
174
+ - Pytorch 2.6.0+cu124
175
+ - Datasets 3.5.0
176
+ - Tokenizers 0.21.1