PedramR commited on
Commit
11254af
·
verified ·
1 Parent(s): 0ee7949

checkpoint-500

Browse files
Files changed (30) hide show
  1. qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/README.md +207 -0
  2. qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/adapter_config.json +56 -0
  3. qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/adapter_model.safetensors +3 -0
  4. qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/best_metrics.json +4 -0
  5. qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/optimizer.pt +3 -0
  6. qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/rng_state.pth +3 -0
  7. qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/scheduler.pt +3 -0
  8. qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/trainer_state.json +384 -0
  9. qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/training_args.bin +3 -0
  10. qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/README.md +207 -0
  11. qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/adapter_config.json +56 -0
  12. qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/adapter_model.safetensors +3 -0
  13. qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/optimizer.pt +3 -0
  14. qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/rng_state.pth +3 -0
  15. qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/scheduler.pt +3 -0
  16. qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/trainer_state.json +384 -0
  17. qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/training_args.bin +3 -0
  18. qwen3.5-2b-100pct-masked-weighted-v2/config.yaml +67 -0
  19. qwen3.5-2b-100pct-masked-weighted-v2/resolved_config.yaml +57 -0
  20. qwen3.5-2b-100pct-masked-weighted-v2/tb/events.out.tfevents.1790605476.02b48645612f.4580.0 +3 -0
  21. qwen3.5-2b-100pct-masked-weighted-v2/test_eval_baseline/metrics.json +16 -0
  22. qwen3.5-2b-100pct-masked-weighted-v2/test_eval_baseline/predictions.jsonl +0 -0
  23. qwen3.5-2b-100pct-masked-weighted-v2/test_eval_entity/metrics.json +16 -0
  24. qwen3.5-2b-100pct-masked-weighted-v2/test_eval_entity/predictions.jsonl +0 -0
  25. qwen3.5-2b-100pct-masked-weighted-v2/test_eval_entity_baseline/metrics.json +16 -0
  26. qwen3.5-2b-100pct-masked-weighted-v2/test_eval_entity_baseline/predictions.jsonl +0 -0
  27. qwen3.5-2b-100pct-masked-weighted-v2/test_eval_typo/metrics.json +16 -0
  28. qwen3.5-2b-100pct-masked-weighted-v2/test_eval_typo/predictions.jsonl +0 -0
  29. qwen3.5-2b-100pct-masked-weighted-v2/test_eval_typo_baseline/metrics.json +16 -0
  30. qwen3.5-2b-100pct-masked-weighted-v2/test_eval_typo_baseline/predictions.jsonl +0 -0
qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/README.md ADDED
@@ -0,0 +1,207 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: Qwen/Qwen3.5-2B
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - base_model:adapter:Qwen/Qwen3.5-2B
7
+ - lora
8
+ - transformers
9
+ ---
10
+
11
+ # Model Card for Model ID
12
+
13
+ <!-- Provide a quick summary of what the model is/does. -->
14
+
15
+
16
+
17
+ ## Model Details
18
+
19
+ ### Model Description
20
+
21
+ <!-- Provide a longer summary of what this model is. -->
22
+
23
+
24
+
25
+ - **Developed by:** [More Information Needed]
26
+ - **Funded by [optional]:** [More Information Needed]
27
+ - **Shared by [optional]:** [More Information Needed]
28
+ - **Model type:** [More Information Needed]
29
+ - **Language(s) (NLP):** [More Information Needed]
30
+ - **License:** [More Information Needed]
31
+ - **Finetuned from model [optional]:** [More Information Needed]
32
+
33
+ ### Model Sources [optional]
34
+
35
+ <!-- Provide the basic links for the model. -->
36
+
37
+ - **Repository:** [More Information Needed]
38
+ - **Paper [optional]:** [More Information Needed]
39
+ - **Demo [optional]:** [More Information Needed]
40
+
41
+ ## Uses
42
+
43
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
44
+
45
+ ### Direct Use
46
+
47
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
48
+
49
+ [More Information Needed]
50
+
51
+ ### Downstream Use [optional]
52
+
53
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
54
+
55
+ [More Information Needed]
56
+
57
+ ### Out-of-Scope Use
58
+
59
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
60
+
61
+ [More Information Needed]
62
+
63
+ ## Bias, Risks, and Limitations
64
+
65
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
66
+
67
+ [More Information Needed]
68
+
69
+ ### Recommendations
70
+
71
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
72
+
73
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
74
+
75
+ ## How to Get Started with the Model
76
+
77
+ Use the code below to get started with the model.
78
+
79
+ [More Information Needed]
80
+
81
+ ## Training Details
82
+
83
+ ### Training Data
84
+
85
+ <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
86
+
87
+ [More Information Needed]
88
+
89
+ ### Training Procedure
90
+
91
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
92
+
93
+ #### Preprocessing [optional]
94
+
95
+ [More Information Needed]
96
+
97
+
98
+ #### Training Hyperparameters
99
+
100
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
101
+
102
+ #### Speeds, Sizes, Times [optional]
103
+
104
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
105
+
106
+ [More Information Needed]
107
+
108
+ ## Evaluation
109
+
110
+ <!-- This section describes the evaluation protocols and provides the results. -->
111
+
112
+ ### Testing Data, Factors & Metrics
113
+
114
+ #### Testing Data
115
+
116
+ <!-- This should link to a Dataset Card if possible. -->
117
+
118
+ [More Information Needed]
119
+
120
+ #### Factors
121
+
122
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
123
+
124
+ [More Information Needed]
125
+
126
+ #### Metrics
127
+
128
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
129
+
130
+ [More Information Needed]
131
+
132
+ ### Results
133
+
134
+ [More Information Needed]
135
+
136
+ #### Summary
137
+
138
+
139
+
140
+ ## Model Examination [optional]
141
+
142
+ <!-- Relevant interpretability work for the model goes here -->
143
+
144
+ [More Information Needed]
145
+
146
+ ## Environmental Impact
147
+
148
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
149
+
150
+ Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
151
+
152
+ - **Hardware Type:** [More Information Needed]
153
+ - **Hours used:** [More Information Needed]
154
+ - **Cloud Provider:** [More Information Needed]
155
+ - **Compute Region:** [More Information Needed]
156
+ - **Carbon Emitted:** [More Information Needed]
157
+
158
+ ## Technical Specifications [optional]
159
+
160
+ ### Model Architecture and Objective
161
+
162
+ [More Information Needed]
163
+
164
+ ### Compute Infrastructure
165
+
166
+ [More Information Needed]
167
+
168
+ #### Hardware
169
+
170
+ [More Information Needed]
171
+
172
+ #### Software
173
+
174
+ [More Information Needed]
175
+
176
+ ## Citation [optional]
177
+
178
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
179
+
180
+ **BibTeX:**
181
+
182
+ [More Information Needed]
183
+
184
+ **APA:**
185
+
186
+ [More Information Needed]
187
+
188
+ ## Glossary [optional]
189
+
190
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
191
+
192
+ [More Information Needed]
193
+
194
+ ## More Information [optional]
195
+
196
+ [More Information Needed]
197
+
198
+ ## Model Card Authors [optional]
199
+
200
+ [More Information Needed]
201
+
202
+ ## Model Card Contact
203
+
204
+ [More Information Needed]
205
+ ### Framework versions
206
+
207
+ - PEFT 0.21.0
qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/adapter_config.json ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "Qwen/Qwen3.5-2B",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": false,
13
+ "inference_mode": true,
14
+ "init_lora_weights": true,
15
+ "kasa_config": null,
16
+ "layer_replication": null,
17
+ "layers_pattern": null,
18
+ "layers_to_transform": null,
19
+ "loftq_config": {},
20
+ "lora_alpha": 64,
21
+ "lora_bias": false,
22
+ "lora_dropout": 0.05,
23
+ "lora_ga_config": null,
24
+ "megatron_config": null,
25
+ "megatron_core": "megatron.core",
26
+ "modules_to_save": null,
27
+ "monteclora_config": null,
28
+ "peft_type": "LORA",
29
+ "peft_version": "0.21.0",
30
+ "qalora_group_size": 16,
31
+ "r": 32,
32
+ "rank_pattern": {},
33
+ "revision": null,
34
+ "target_modules": [
35
+ "q_proj",
36
+ "in_proj_b",
37
+ "o_proj",
38
+ "down_proj",
39
+ "v_proj",
40
+ "up_proj",
41
+ "out_proj",
42
+ "in_proj_z",
43
+ "k_proj",
44
+ "gate_proj",
45
+ "in_proj_qkv",
46
+ "in_proj_a"
47
+ ],
48
+ "target_parameters": null,
49
+ "task_type": "CAUSAL_LM",
50
+ "trainable_token_indices": null,
51
+ "use_bdlora": null,
52
+ "use_dora": false,
53
+ "use_qalora": false,
54
+ "use_rslora": false,
55
+ "velora_config": null
56
+ }
qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:90ce164ea4415c1d512535e37647dfad4669f15274cceba22515c0d5f54a6d27
3
+ size 134604152
qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/best_metrics.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "step": 500,
3
+ "wer": 0.21939984789822387
4
+ }
qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:99cacc22e9728e6b599b9fd6b823574377f16b958bc257e8f42e2f83aa921a2e
3
+ size 269426431
qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:90cf8e31f238979a8a77d1f114b86a765e07d4aa3555cdca9685305b07b578b8
3
+ size 14645
qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:490cfc293c61f546e9ca8bea24a6ff2f70f00dafb78159902250efed21573d84
3
+ size 1465
qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/trainer_state.json ADDED
@@ -0,0 +1,384 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.08972331572513262,
6
+ "eval_steps": 500,
7
+ "global_step": 500,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.0017944663145026524,
14
+ "grad_norm": 2.403308153152466,
15
+ "learning_rate": 3.225806451612903e-06,
16
+ "loss": 1.210298442840576,
17
+ "step": 10
18
+ },
19
+ {
20
+ "epoch": 0.003588932629005305,
21
+ "grad_norm": 1.3837767839431763,
22
+ "learning_rate": 6.810035842293908e-06,
23
+ "loss": 1.238302230834961,
24
+ "step": 20
25
+ },
26
+ {
27
+ "epoch": 0.005383398943507958,
28
+ "grad_norm": 1.152842402458191,
29
+ "learning_rate": 1.039426523297491e-05,
30
+ "loss": 1.1493020057678223,
31
+ "step": 30
32
+ },
33
+ {
34
+ "epoch": 0.00717786525801061,
35
+ "grad_norm": 1.2282543182373047,
36
+ "learning_rate": 1.3978494623655914e-05,
37
+ "loss": 1.2698699951171875,
38
+ "step": 40
39
+ },
40
+ {
41
+ "epoch": 0.008972331572513263,
42
+ "grad_norm": 4.5811686515808105,
43
+ "learning_rate": 1.7562724014336916e-05,
44
+ "loss": 1.336978054046631,
45
+ "step": 50
46
+ },
47
+ {
48
+ "epoch": 0.010766797887015915,
49
+ "grad_norm": 0.7280638217926025,
50
+ "learning_rate": 2.1146953405017922e-05,
51
+ "loss": 0.9929692268371582,
52
+ "step": 60
53
+ },
54
+ {
55
+ "epoch": 0.012561264201518567,
56
+ "grad_norm": 0.8359633684158325,
57
+ "learning_rate": 2.4731182795698928e-05,
58
+ "loss": 0.987311840057373,
59
+ "step": 70
60
+ },
61
+ {
62
+ "epoch": 0.01435573051602122,
63
+ "grad_norm": 1.061148762702942,
64
+ "learning_rate": 2.831541218637993e-05,
65
+ "loss": 1.0299980163574218,
66
+ "step": 80
67
+ },
68
+ {
69
+ "epoch": 0.01615019683052387,
70
+ "grad_norm": 1.4723000526428223,
71
+ "learning_rate": 3.1899641577060935e-05,
72
+ "loss": 1.1110344886779786,
73
+ "step": 90
74
+ },
75
+ {
76
+ "epoch": 0.017944663145026525,
77
+ "grad_norm": 5.412300109863281,
78
+ "learning_rate": 3.548387096774194e-05,
79
+ "loss": 1.1978090286254883,
80
+ "step": 100
81
+ },
82
+ {
83
+ "epoch": 0.019739129459529176,
84
+ "grad_norm": 0.7945532202720642,
85
+ "learning_rate": 3.906810035842295e-05,
86
+ "loss": 0.8953464508056641,
87
+ "step": 110
88
+ },
89
+ {
90
+ "epoch": 0.02153359577403183,
91
+ "grad_norm": 0.9241706728935242,
92
+ "learning_rate": 4.265232974910394e-05,
93
+ "loss": 0.8933357238769531,
94
+ "step": 120
95
+ },
96
+ {
97
+ "epoch": 0.023328062088534482,
98
+ "grad_norm": 1.2824897766113281,
99
+ "learning_rate": 4.6236559139784944e-05,
100
+ "loss": 0.9129752159118653,
101
+ "step": 130
102
+ },
103
+ {
104
+ "epoch": 0.025122528403037133,
105
+ "grad_norm": 1.8574696779251099,
106
+ "learning_rate": 4.982078853046595e-05,
107
+ "loss": 1.0019177436828612,
108
+ "step": 140
109
+ },
110
+ {
111
+ "epoch": 0.026916994717539788,
112
+ "grad_norm": 4.865994930267334,
113
+ "learning_rate": 5.340501792114696e-05,
114
+ "loss": 1.1639716148376464,
115
+ "step": 150
116
+ },
117
+ {
118
+ "epoch": 0.02871146103204244,
119
+ "grad_norm": 0.8981343507766724,
120
+ "learning_rate": 5.6989247311827965e-05,
121
+ "loss": 0.8498629570007324,
122
+ "step": 160
123
+ },
124
+ {
125
+ "epoch": 0.03050592734654509,
126
+ "grad_norm": 0.8773536086082458,
127
+ "learning_rate": 6.057347670250897e-05,
128
+ "loss": 0.8652203559875489,
129
+ "step": 170
130
+ },
131
+ {
132
+ "epoch": 0.03230039366104774,
133
+ "grad_norm": 1.2524876594543457,
134
+ "learning_rate": 6.415770609318996e-05,
135
+ "loss": 0.9427430152893066,
136
+ "step": 180
137
+ },
138
+ {
139
+ "epoch": 0.034094859975550396,
140
+ "grad_norm": 1.7148643732070923,
141
+ "learning_rate": 6.774193548387096e-05,
142
+ "loss": 0.9581910133361816,
143
+ "step": 190
144
+ },
145
+ {
146
+ "epoch": 0.03588932629005305,
147
+ "grad_norm": 5.458459854125977,
148
+ "learning_rate": 7.132616487455197e-05,
149
+ "loss": 1.1793695449829102,
150
+ "step": 200
151
+ },
152
+ {
153
+ "epoch": 0.037683792604555705,
154
+ "grad_norm": 0.7586544752120972,
155
+ "learning_rate": 7.491039426523297e-05,
156
+ "loss": 0.8673212051391601,
157
+ "step": 210
158
+ },
159
+ {
160
+ "epoch": 0.03947825891905835,
161
+ "grad_norm": 0.8580747246742249,
162
+ "learning_rate": 7.849462365591398e-05,
163
+ "loss": 0.8158726692199707,
164
+ "step": 220
165
+ },
166
+ {
167
+ "epoch": 0.04127272523356101,
168
+ "grad_norm": 1.0292130708694458,
169
+ "learning_rate": 8.207885304659499e-05,
170
+ "loss": 0.8816629409790039,
171
+ "step": 230
172
+ },
173
+ {
174
+ "epoch": 0.04306719154806366,
175
+ "grad_norm": 1.2191781997680664,
176
+ "learning_rate": 8.566308243727598e-05,
177
+ "loss": 0.9256319999694824,
178
+ "step": 240
179
+ },
180
+ {
181
+ "epoch": 0.04486165786256631,
182
+ "grad_norm": 4.896722316741943,
183
+ "learning_rate": 8.924731182795699e-05,
184
+ "loss": 1.054029369354248,
185
+ "step": 250
186
+ },
187
+ {
188
+ "epoch": 0.046656124177068964,
189
+ "grad_norm": 0.6555370688438416,
190
+ "learning_rate": 9.2831541218638e-05,
191
+ "loss": 0.8240686416625976,
192
+ "step": 260
193
+ },
194
+ {
195
+ "epoch": 0.04845059049157162,
196
+ "grad_norm": 0.8895808458328247,
197
+ "learning_rate": 9.6415770609319e-05,
198
+ "loss": 0.7868500232696534,
199
+ "step": 270
200
+ },
201
+ {
202
+ "epoch": 0.050245056806074266,
203
+ "grad_norm": 0.9253164529800415,
204
+ "learning_rate": 0.0001,
205
+ "loss": 0.821414852142334,
206
+ "step": 280
207
+ },
208
+ {
209
+ "epoch": 0.05203952312057692,
210
+ "grad_norm": 1.134040355682373,
211
+ "learning_rate": 9.981110691348697e-05,
212
+ "loss": 0.9013632774353028,
213
+ "step": 290
214
+ },
215
+ {
216
+ "epoch": 0.053833989435079575,
217
+ "grad_norm": 3.446653366088867,
218
+ "learning_rate": 9.962221382697394e-05,
219
+ "loss": 1.0495364189147949,
220
+ "step": 300
221
+ },
222
+ {
223
+ "epoch": 0.05562845574958222,
224
+ "grad_norm": 0.7303311824798584,
225
+ "learning_rate": 9.94333207404609e-05,
226
+ "loss": 0.7989195346832275,
227
+ "step": 310
228
+ },
229
+ {
230
+ "epoch": 0.05742292206408488,
231
+ "grad_norm": 0.8196483850479126,
232
+ "learning_rate": 9.924442765394787e-05,
233
+ "loss": 0.7937421798706055,
234
+ "step": 320
235
+ },
236
+ {
237
+ "epoch": 0.05921738837858753,
238
+ "grad_norm": 0.9366778135299683,
239
+ "learning_rate": 9.905553456743484e-05,
240
+ "loss": 0.809019660949707,
241
+ "step": 330
242
+ },
243
+ {
244
+ "epoch": 0.06101185469309018,
245
+ "grad_norm": 1.2467617988586426,
246
+ "learning_rate": 9.88666414809218e-05,
247
+ "loss": 0.930391502380371,
248
+ "step": 340
249
+ },
250
+ {
251
+ "epoch": 0.06280632100759284,
252
+ "grad_norm": 3.261760950088501,
253
+ "learning_rate": 9.867774839440876e-05,
254
+ "loss": 1.0620959281921387,
255
+ "step": 350
256
+ },
257
+ {
258
+ "epoch": 0.06460078732209548,
259
+ "grad_norm": 0.6342303156852722,
260
+ "learning_rate": 9.848885530789574e-05,
261
+ "loss": 0.8134325981140137,
262
+ "step": 360
263
+ },
264
+ {
265
+ "epoch": 0.06639525363659814,
266
+ "grad_norm": 0.6608202457427979,
267
+ "learning_rate": 9.82999622213827e-05,
268
+ "loss": 0.789882230758667,
269
+ "step": 370
270
+ },
271
+ {
272
+ "epoch": 0.06818971995110079,
273
+ "grad_norm": 0.8570723533630371,
274
+ "learning_rate": 9.811106913486967e-05,
275
+ "loss": 0.7621271133422851,
276
+ "step": 380
277
+ },
278
+ {
279
+ "epoch": 0.06998418626560345,
280
+ "grad_norm": 1.162699818611145,
281
+ "learning_rate": 9.792217604835664e-05,
282
+ "loss": 0.8825998306274414,
283
+ "step": 390
284
+ },
285
+ {
286
+ "epoch": 0.0717786525801061,
287
+ "grad_norm": 3.485926866531372,
288
+ "learning_rate": 9.77332829618436e-05,
289
+ "loss": 1.0364310264587402,
290
+ "step": 400
291
+ },
292
+ {
293
+ "epoch": 0.07357311889460875,
294
+ "grad_norm": 0.5732496380805969,
295
+ "learning_rate": 9.754438987533056e-05,
296
+ "loss": 0.7694174289703369,
297
+ "step": 410
298
+ },
299
+ {
300
+ "epoch": 0.07536758520911141,
301
+ "grad_norm": 0.683313250541687,
302
+ "learning_rate": 9.735549678881754e-05,
303
+ "loss": 0.7631448745727539,
304
+ "step": 420
305
+ },
306
+ {
307
+ "epoch": 0.07716205152361405,
308
+ "grad_norm": 0.8606892824172974,
309
+ "learning_rate": 9.716660370230451e-05,
310
+ "loss": 0.786726999282837,
311
+ "step": 430
312
+ },
313
+ {
314
+ "epoch": 0.0789565178381167,
315
+ "grad_norm": 1.1524667739868164,
316
+ "learning_rate": 9.697771061579146e-05,
317
+ "loss": 0.8868392944335938,
318
+ "step": 440
319
+ },
320
+ {
321
+ "epoch": 0.08075098415261936,
322
+ "grad_norm": 3.888843297958374,
323
+ "learning_rate": 9.678881752927843e-05,
324
+ "loss": 0.949896240234375,
325
+ "step": 450
326
+ },
327
+ {
328
+ "epoch": 0.08254545046712201,
329
+ "grad_norm": 0.540230393409729,
330
+ "learning_rate": 9.65999244427654e-05,
331
+ "loss": 0.7518480300903321,
332
+ "step": 460
333
+ },
334
+ {
335
+ "epoch": 0.08433991678162467,
336
+ "grad_norm": 0.7087560296058655,
337
+ "learning_rate": 9.641103135625237e-05,
338
+ "loss": 0.7413692474365234,
339
+ "step": 470
340
+ },
341
+ {
342
+ "epoch": 0.08613438309612732,
343
+ "grad_norm": 0.7639017105102539,
344
+ "learning_rate": 9.622213826973934e-05,
345
+ "loss": 0.8329146385192872,
346
+ "step": 480
347
+ },
348
+ {
349
+ "epoch": 0.08792884941062996,
350
+ "grad_norm": 1.0710383653640747,
351
+ "learning_rate": 9.60332451832263e-05,
352
+ "loss": 0.8491949081420899,
353
+ "step": 490
354
+ },
355
+ {
356
+ "epoch": 0.08972331572513262,
357
+ "grad_norm": 3.929804563522339,
358
+ "learning_rate": 9.584435209671326e-05,
359
+ "loss": 1.014235496520996,
360
+ "step": 500
361
+ }
362
+ ],
363
+ "logging_steps": 10,
364
+ "max_steps": 5573,
365
+ "num_input_tokens_seen": 0,
366
+ "num_train_epochs": 1,
367
+ "save_steps": 500,
368
+ "stateful_callbacks": {
369
+ "TrainerControl": {
370
+ "args": {
371
+ "should_epoch_stop": false,
372
+ "should_evaluate": false,
373
+ "should_log": false,
374
+ "should_save": true,
375
+ "should_training_stop": false
376
+ },
377
+ "attributes": {}
378
+ }
379
+ },
380
+ "total_flos": 1.0145643754992077e+17,
381
+ "train_batch_size": 2,
382
+ "trial_name": null,
383
+ "trial_params": null
384
+ }
qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:36324839c555b723546ca05f552d997ae3f0d4b9b817db61c11bbc2e65ec719c
3
+ size 5265
qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/README.md ADDED
@@ -0,0 +1,207 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: Qwen/Qwen3.5-2B
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - base_model:adapter:Qwen/Qwen3.5-2B
7
+ - lora
8
+ - transformers
9
+ ---
10
+
11
+ # Model Card for Model ID
12
+
13
+ <!-- Provide a quick summary of what the model is/does. -->
14
+
15
+
16
+
17
+ ## Model Details
18
+
19
+ ### Model Description
20
+
21
+ <!-- Provide a longer summary of what this model is. -->
22
+
23
+
24
+
25
+ - **Developed by:** [More Information Needed]
26
+ - **Funded by [optional]:** [More Information Needed]
27
+ - **Shared by [optional]:** [More Information Needed]
28
+ - **Model type:** [More Information Needed]
29
+ - **Language(s) (NLP):** [More Information Needed]
30
+ - **License:** [More Information Needed]
31
+ - **Finetuned from model [optional]:** [More Information Needed]
32
+
33
+ ### Model Sources [optional]
34
+
35
+ <!-- Provide the basic links for the model. -->
36
+
37
+ - **Repository:** [More Information Needed]
38
+ - **Paper [optional]:** [More Information Needed]
39
+ - **Demo [optional]:** [More Information Needed]
40
+
41
+ ## Uses
42
+
43
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
44
+
45
+ ### Direct Use
46
+
47
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
48
+
49
+ [More Information Needed]
50
+
51
+ ### Downstream Use [optional]
52
+
53
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
54
+
55
+ [More Information Needed]
56
+
57
+ ### Out-of-Scope Use
58
+
59
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
60
+
61
+ [More Information Needed]
62
+
63
+ ## Bias, Risks, and Limitations
64
+
65
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
66
+
67
+ [More Information Needed]
68
+
69
+ ### Recommendations
70
+
71
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
72
+
73
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
74
+
75
+ ## How to Get Started with the Model
76
+
77
+ Use the code below to get started with the model.
78
+
79
+ [More Information Needed]
80
+
81
+ ## Training Details
82
+
83
+ ### Training Data
84
+
85
+ <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
86
+
87
+ [More Information Needed]
88
+
89
+ ### Training Procedure
90
+
91
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
92
+
93
+ #### Preprocessing [optional]
94
+
95
+ [More Information Needed]
96
+
97
+
98
+ #### Training Hyperparameters
99
+
100
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
101
+
102
+ #### Speeds, Sizes, Times [optional]
103
+
104
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
105
+
106
+ [More Information Needed]
107
+
108
+ ## Evaluation
109
+
110
+ <!-- This section describes the evaluation protocols and provides the results. -->
111
+
112
+ ### Testing Data, Factors & Metrics
113
+
114
+ #### Testing Data
115
+
116
+ <!-- This should link to a Dataset Card if possible. -->
117
+
118
+ [More Information Needed]
119
+
120
+ #### Factors
121
+
122
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
123
+
124
+ [More Information Needed]
125
+
126
+ #### Metrics
127
+
128
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
129
+
130
+ [More Information Needed]
131
+
132
+ ### Results
133
+
134
+ [More Information Needed]
135
+
136
+ #### Summary
137
+
138
+
139
+
140
+ ## Model Examination [optional]
141
+
142
+ <!-- Relevant interpretability work for the model goes here -->
143
+
144
+ [More Information Needed]
145
+
146
+ ## Environmental Impact
147
+
148
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
149
+
150
+ Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
151
+
152
+ - **Hardware Type:** [More Information Needed]
153
+ - **Hours used:** [More Information Needed]
154
+ - **Cloud Provider:** [More Information Needed]
155
+ - **Compute Region:** [More Information Needed]
156
+ - **Carbon Emitted:** [More Information Needed]
157
+
158
+ ## Technical Specifications [optional]
159
+
160
+ ### Model Architecture and Objective
161
+
162
+ [More Information Needed]
163
+
164
+ ### Compute Infrastructure
165
+
166
+ [More Information Needed]
167
+
168
+ #### Hardware
169
+
170
+ [More Information Needed]
171
+
172
+ #### Software
173
+
174
+ [More Information Needed]
175
+
176
+ ## Citation [optional]
177
+
178
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
179
+
180
+ **BibTeX:**
181
+
182
+ [More Information Needed]
183
+
184
+ **APA:**
185
+
186
+ [More Information Needed]
187
+
188
+ ## Glossary [optional]
189
+
190
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
191
+
192
+ [More Information Needed]
193
+
194
+ ## More Information [optional]
195
+
196
+ [More Information Needed]
197
+
198
+ ## Model Card Authors [optional]
199
+
200
+ [More Information Needed]
201
+
202
+ ## Model Card Contact
203
+
204
+ [More Information Needed]
205
+ ### Framework versions
206
+
207
+ - PEFT 0.21.0
qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/adapter_config.json ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": null,
6
+ "base_model_name_or_path": "Qwen/Qwen3.5-2B",
7
+ "bias": "none",
8
+ "corda_config": null,
9
+ "ensure_weight_tying": false,
10
+ "eva_config": null,
11
+ "exclude_modules": null,
12
+ "fan_in_fan_out": false,
13
+ "inference_mode": true,
14
+ "init_lora_weights": true,
15
+ "kasa_config": null,
16
+ "layer_replication": null,
17
+ "layers_pattern": null,
18
+ "layers_to_transform": null,
19
+ "loftq_config": {},
20
+ "lora_alpha": 64,
21
+ "lora_bias": false,
22
+ "lora_dropout": 0.05,
23
+ "lora_ga_config": null,
24
+ "megatron_config": null,
25
+ "megatron_core": "megatron.core",
26
+ "modules_to_save": null,
27
+ "monteclora_config": null,
28
+ "peft_type": "LORA",
29
+ "peft_version": "0.21.0",
30
+ "qalora_group_size": 16,
31
+ "r": 32,
32
+ "rank_pattern": {},
33
+ "revision": null,
34
+ "target_modules": [
35
+ "q_proj",
36
+ "in_proj_b",
37
+ "o_proj",
38
+ "down_proj",
39
+ "v_proj",
40
+ "up_proj",
41
+ "out_proj",
42
+ "in_proj_z",
43
+ "k_proj",
44
+ "gate_proj",
45
+ "in_proj_qkv",
46
+ "in_proj_a"
47
+ ],
48
+ "target_parameters": null,
49
+ "task_type": "CAUSAL_LM",
50
+ "trainable_token_indices": null,
51
+ "use_bdlora": null,
52
+ "use_dora": false,
53
+ "use_qalora": false,
54
+ "use_rslora": false,
55
+ "velora_config": null
56
+ }
qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:90ce164ea4415c1d512535e37647dfad4669f15274cceba22515c0d5f54a6d27
3
+ size 134604152
qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:99cacc22e9728e6b599b9fd6b823574377f16b958bc257e8f42e2f83aa921a2e
3
+ size 269426431
qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:90cf8e31f238979a8a77d1f114b86a765e07d4aa3555cdca9685305b07b578b8
3
+ size 14645
qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:490cfc293c61f546e9ca8bea24a6ff2f70f00dafb78159902250efed21573d84
3
+ size 1465
qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/trainer_state.json ADDED
@@ -0,0 +1,384 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.08972331572513262,
6
+ "eval_steps": 500,
7
+ "global_step": 500,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.0017944663145026524,
14
+ "grad_norm": 2.403308153152466,
15
+ "learning_rate": 3.225806451612903e-06,
16
+ "loss": 1.210298442840576,
17
+ "step": 10
18
+ },
19
+ {
20
+ "epoch": 0.003588932629005305,
21
+ "grad_norm": 1.3837767839431763,
22
+ "learning_rate": 6.810035842293908e-06,
23
+ "loss": 1.238302230834961,
24
+ "step": 20
25
+ },
26
+ {
27
+ "epoch": 0.005383398943507958,
28
+ "grad_norm": 1.152842402458191,
29
+ "learning_rate": 1.039426523297491e-05,
30
+ "loss": 1.1493020057678223,
31
+ "step": 30
32
+ },
33
+ {
34
+ "epoch": 0.00717786525801061,
35
+ "grad_norm": 1.2282543182373047,
36
+ "learning_rate": 1.3978494623655914e-05,
37
+ "loss": 1.2698699951171875,
38
+ "step": 40
39
+ },
40
+ {
41
+ "epoch": 0.008972331572513263,
42
+ "grad_norm": 4.5811686515808105,
43
+ "learning_rate": 1.7562724014336916e-05,
44
+ "loss": 1.336978054046631,
45
+ "step": 50
46
+ },
47
+ {
48
+ "epoch": 0.010766797887015915,
49
+ "grad_norm": 0.7280638217926025,
50
+ "learning_rate": 2.1146953405017922e-05,
51
+ "loss": 0.9929692268371582,
52
+ "step": 60
53
+ },
54
+ {
55
+ "epoch": 0.012561264201518567,
56
+ "grad_norm": 0.8359633684158325,
57
+ "learning_rate": 2.4731182795698928e-05,
58
+ "loss": 0.987311840057373,
59
+ "step": 70
60
+ },
61
+ {
62
+ "epoch": 0.01435573051602122,
63
+ "grad_norm": 1.061148762702942,
64
+ "learning_rate": 2.831541218637993e-05,
65
+ "loss": 1.0299980163574218,
66
+ "step": 80
67
+ },
68
+ {
69
+ "epoch": 0.01615019683052387,
70
+ "grad_norm": 1.4723000526428223,
71
+ "learning_rate": 3.1899641577060935e-05,
72
+ "loss": 1.1110344886779786,
73
+ "step": 90
74
+ },
75
+ {
76
+ "epoch": 0.017944663145026525,
77
+ "grad_norm": 5.412300109863281,
78
+ "learning_rate": 3.548387096774194e-05,
79
+ "loss": 1.1978090286254883,
80
+ "step": 100
81
+ },
82
+ {
83
+ "epoch": 0.019739129459529176,
84
+ "grad_norm": 0.7945532202720642,
85
+ "learning_rate": 3.906810035842295e-05,
86
+ "loss": 0.8953464508056641,
87
+ "step": 110
88
+ },
89
+ {
90
+ "epoch": 0.02153359577403183,
91
+ "grad_norm": 0.9241706728935242,
92
+ "learning_rate": 4.265232974910394e-05,
93
+ "loss": 0.8933357238769531,
94
+ "step": 120
95
+ },
96
+ {
97
+ "epoch": 0.023328062088534482,
98
+ "grad_norm": 1.2824897766113281,
99
+ "learning_rate": 4.6236559139784944e-05,
100
+ "loss": 0.9129752159118653,
101
+ "step": 130
102
+ },
103
+ {
104
+ "epoch": 0.025122528403037133,
105
+ "grad_norm": 1.8574696779251099,
106
+ "learning_rate": 4.982078853046595e-05,
107
+ "loss": 1.0019177436828612,
108
+ "step": 140
109
+ },
110
+ {
111
+ "epoch": 0.026916994717539788,
112
+ "grad_norm": 4.865994930267334,
113
+ "learning_rate": 5.340501792114696e-05,
114
+ "loss": 1.1639716148376464,
115
+ "step": 150
116
+ },
117
+ {
118
+ "epoch": 0.02871146103204244,
119
+ "grad_norm": 0.8981343507766724,
120
+ "learning_rate": 5.6989247311827965e-05,
121
+ "loss": 0.8498629570007324,
122
+ "step": 160
123
+ },
124
+ {
125
+ "epoch": 0.03050592734654509,
126
+ "grad_norm": 0.8773536086082458,
127
+ "learning_rate": 6.057347670250897e-05,
128
+ "loss": 0.8652203559875489,
129
+ "step": 170
130
+ },
131
+ {
132
+ "epoch": 0.03230039366104774,
133
+ "grad_norm": 1.2524876594543457,
134
+ "learning_rate": 6.415770609318996e-05,
135
+ "loss": 0.9427430152893066,
136
+ "step": 180
137
+ },
138
+ {
139
+ "epoch": 0.034094859975550396,
140
+ "grad_norm": 1.7148643732070923,
141
+ "learning_rate": 6.774193548387096e-05,
142
+ "loss": 0.9581910133361816,
143
+ "step": 190
144
+ },
145
+ {
146
+ "epoch": 0.03588932629005305,
147
+ "grad_norm": 5.458459854125977,
148
+ "learning_rate": 7.132616487455197e-05,
149
+ "loss": 1.1793695449829102,
150
+ "step": 200
151
+ },
152
+ {
153
+ "epoch": 0.037683792604555705,
154
+ "grad_norm": 0.7586544752120972,
155
+ "learning_rate": 7.491039426523297e-05,
156
+ "loss": 0.8673212051391601,
157
+ "step": 210
158
+ },
159
+ {
160
+ "epoch": 0.03947825891905835,
161
+ "grad_norm": 0.8580747246742249,
162
+ "learning_rate": 7.849462365591398e-05,
163
+ "loss": 0.8158726692199707,
164
+ "step": 220
165
+ },
166
+ {
167
+ "epoch": 0.04127272523356101,
168
+ "grad_norm": 1.0292130708694458,
169
+ "learning_rate": 8.207885304659499e-05,
170
+ "loss": 0.8816629409790039,
171
+ "step": 230
172
+ },
173
+ {
174
+ "epoch": 0.04306719154806366,
175
+ "grad_norm": 1.2191781997680664,
176
+ "learning_rate": 8.566308243727598e-05,
177
+ "loss": 0.9256319999694824,
178
+ "step": 240
179
+ },
180
+ {
181
+ "epoch": 0.04486165786256631,
182
+ "grad_norm": 4.896722316741943,
183
+ "learning_rate": 8.924731182795699e-05,
184
+ "loss": 1.054029369354248,
185
+ "step": 250
186
+ },
187
+ {
188
+ "epoch": 0.046656124177068964,
189
+ "grad_norm": 0.6555370688438416,
190
+ "learning_rate": 9.2831541218638e-05,
191
+ "loss": 0.8240686416625976,
192
+ "step": 260
193
+ },
194
+ {
195
+ "epoch": 0.04845059049157162,
196
+ "grad_norm": 0.8895808458328247,
197
+ "learning_rate": 9.6415770609319e-05,
198
+ "loss": 0.7868500232696534,
199
+ "step": 270
200
+ },
201
+ {
202
+ "epoch": 0.050245056806074266,
203
+ "grad_norm": 0.9253164529800415,
204
+ "learning_rate": 0.0001,
205
+ "loss": 0.821414852142334,
206
+ "step": 280
207
+ },
208
+ {
209
+ "epoch": 0.05203952312057692,
210
+ "grad_norm": 1.134040355682373,
211
+ "learning_rate": 9.981110691348697e-05,
212
+ "loss": 0.9013632774353028,
213
+ "step": 290
214
+ },
215
+ {
216
+ "epoch": 0.053833989435079575,
217
+ "grad_norm": 3.446653366088867,
218
+ "learning_rate": 9.962221382697394e-05,
219
+ "loss": 1.0495364189147949,
220
+ "step": 300
221
+ },
222
+ {
223
+ "epoch": 0.05562845574958222,
224
+ "grad_norm": 0.7303311824798584,
225
+ "learning_rate": 9.94333207404609e-05,
226
+ "loss": 0.7989195346832275,
227
+ "step": 310
228
+ },
229
+ {
230
+ "epoch": 0.05742292206408488,
231
+ "grad_norm": 0.8196483850479126,
232
+ "learning_rate": 9.924442765394787e-05,
233
+ "loss": 0.7937421798706055,
234
+ "step": 320
235
+ },
236
+ {
237
+ "epoch": 0.05921738837858753,
238
+ "grad_norm": 0.9366778135299683,
239
+ "learning_rate": 9.905553456743484e-05,
240
+ "loss": 0.809019660949707,
241
+ "step": 330
242
+ },
243
+ {
244
+ "epoch": 0.06101185469309018,
245
+ "grad_norm": 1.2467617988586426,
246
+ "learning_rate": 9.88666414809218e-05,
247
+ "loss": 0.930391502380371,
248
+ "step": 340
249
+ },
250
+ {
251
+ "epoch": 0.06280632100759284,
252
+ "grad_norm": 3.261760950088501,
253
+ "learning_rate": 9.867774839440876e-05,
254
+ "loss": 1.0620959281921387,
255
+ "step": 350
256
+ },
257
+ {
258
+ "epoch": 0.06460078732209548,
259
+ "grad_norm": 0.6342303156852722,
260
+ "learning_rate": 9.848885530789574e-05,
261
+ "loss": 0.8134325981140137,
262
+ "step": 360
263
+ },
264
+ {
265
+ "epoch": 0.06639525363659814,
266
+ "grad_norm": 0.6608202457427979,
267
+ "learning_rate": 9.82999622213827e-05,
268
+ "loss": 0.789882230758667,
269
+ "step": 370
270
+ },
271
+ {
272
+ "epoch": 0.06818971995110079,
273
+ "grad_norm": 0.8570723533630371,
274
+ "learning_rate": 9.811106913486967e-05,
275
+ "loss": 0.7621271133422851,
276
+ "step": 380
277
+ },
278
+ {
279
+ "epoch": 0.06998418626560345,
280
+ "grad_norm": 1.162699818611145,
281
+ "learning_rate": 9.792217604835664e-05,
282
+ "loss": 0.8825998306274414,
283
+ "step": 390
284
+ },
285
+ {
286
+ "epoch": 0.0717786525801061,
287
+ "grad_norm": 3.485926866531372,
288
+ "learning_rate": 9.77332829618436e-05,
289
+ "loss": 1.0364310264587402,
290
+ "step": 400
291
+ },
292
+ {
293
+ "epoch": 0.07357311889460875,
294
+ "grad_norm": 0.5732496380805969,
295
+ "learning_rate": 9.754438987533056e-05,
296
+ "loss": 0.7694174289703369,
297
+ "step": 410
298
+ },
299
+ {
300
+ "epoch": 0.07536758520911141,
301
+ "grad_norm": 0.683313250541687,
302
+ "learning_rate": 9.735549678881754e-05,
303
+ "loss": 0.7631448745727539,
304
+ "step": 420
305
+ },
306
+ {
307
+ "epoch": 0.07716205152361405,
308
+ "grad_norm": 0.8606892824172974,
309
+ "learning_rate": 9.716660370230451e-05,
310
+ "loss": 0.786726999282837,
311
+ "step": 430
312
+ },
313
+ {
314
+ "epoch": 0.0789565178381167,
315
+ "grad_norm": 1.1524667739868164,
316
+ "learning_rate": 9.697771061579146e-05,
317
+ "loss": 0.8868392944335938,
318
+ "step": 440
319
+ },
320
+ {
321
+ "epoch": 0.08075098415261936,
322
+ "grad_norm": 3.888843297958374,
323
+ "learning_rate": 9.678881752927843e-05,
324
+ "loss": 0.949896240234375,
325
+ "step": 450
326
+ },
327
+ {
328
+ "epoch": 0.08254545046712201,
329
+ "grad_norm": 0.540230393409729,
330
+ "learning_rate": 9.65999244427654e-05,
331
+ "loss": 0.7518480300903321,
332
+ "step": 460
333
+ },
334
+ {
335
+ "epoch": 0.08433991678162467,
336
+ "grad_norm": 0.7087560296058655,
337
+ "learning_rate": 9.641103135625237e-05,
338
+ "loss": 0.7413692474365234,
339
+ "step": 470
340
+ },
341
+ {
342
+ "epoch": 0.08613438309612732,
343
+ "grad_norm": 0.7639017105102539,
344
+ "learning_rate": 9.622213826973934e-05,
345
+ "loss": 0.8329146385192872,
346
+ "step": 480
347
+ },
348
+ {
349
+ "epoch": 0.08792884941062996,
350
+ "grad_norm": 1.0710383653640747,
351
+ "learning_rate": 9.60332451832263e-05,
352
+ "loss": 0.8491949081420899,
353
+ "step": 490
354
+ },
355
+ {
356
+ "epoch": 0.08972331572513262,
357
+ "grad_norm": 3.929804563522339,
358
+ "learning_rate": 9.584435209671326e-05,
359
+ "loss": 1.014235496520996,
360
+ "step": 500
361
+ }
362
+ ],
363
+ "logging_steps": 10,
364
+ "max_steps": 5573,
365
+ "num_input_tokens_seen": 0,
366
+ "num_train_epochs": 1,
367
+ "save_steps": 500,
368
+ "stateful_callbacks": {
369
+ "TrainerControl": {
370
+ "args": {
371
+ "should_epoch_stop": false,
372
+ "should_evaluate": false,
373
+ "should_log": false,
374
+ "should_save": true,
375
+ "should_training_stop": false
376
+ },
377
+ "attributes": {}
378
+ }
379
+ },
380
+ "total_flos": 1.0145643754992077e+17,
381
+ "train_batch_size": 2,
382
+ "trial_name": null,
383
+ "trial_params": null
384
+ }
qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:36324839c555b723546ca05f552d997ae3f0d4b9b817db61c11bbc2e65ec719c
3
+ size 5265
qwen3.5-2b-100pct-masked-weighted-v2/config.yaml ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model_id: Qwen/Qwen3.5-2B
2
+ dataset_id: PedramR/ASR_Post-processing-dataset
3
+ data_config: configs/data/default.yaml
4
+ train_split: train
5
+ eval_split: null
6
+ eval_fraction: 0.1
7
+ train_fraction: 1.0
8
+ input_column: text_whisper
9
+ target_column: text
10
+ system_prompt: null
11
+ include_punctuation: false
12
+ output_dir: outputs/qwen-mask-weighted/qwen3.5-2b-100pct-masked-weighted-v2
13
+ max_length: 2048
14
+ on_long_example: drop
15
+ mask_low_signal_corrections: true
16
+ mask_min_similarity: 0.75
17
+ mask_max_common_freq: 1
18
+ recoverable_correction_weight: 2.0
19
+ short_target_eos_weight: 0.3
20
+ short_target_max_words: 20
21
+ num_train_epochs: 1.0
22
+ per_device_train_batch_size: 2
23
+ per_device_eval_batch_size: 2
24
+ gradient_accumulation_steps: 16
25
+ learning_rate: 0.0001
26
+ warmup_ratio: 0.05
27
+ lr_scheduler_type: linear
28
+ logging_steps: 10
29
+ eval_steps: 500
30
+ save_steps: 500
31
+ save_total_limit: 3
32
+ seed: 42
33
+ early_stopping_patience: 3
34
+ lora_r: 32
35
+ lora_alpha: 64
36
+ lora_dropout: 0.05
37
+ load_in_4bit: false
38
+ use_unsloth: false
39
+ push_to_hub: true
40
+ hub_repo_id: PedramR/ASR_Post-processing
41
+ hub_repo_folder: qwen3.5-2b-100pct-masked-weighted-v2
42
+ hub_private: false
43
+ tensorboard_logging_dir: outputs/qwen-mask-weighted/qwen3.5-2b-100pct-masked-weighted-v2/tb
44
+ resume_from_checkpoint: false
45
+ lora_init_mode: fresh
46
+ init_lora_from: null
47
+ init_lora_from_subfolder: null
48
+ test_dataset_id: ErfanRou/callcc-test-1k
49
+ test_split: test
50
+ test_input_column: text_whisper
51
+ test_target_column: text
52
+ test_max_new_tokens: 1024
53
+ test_batch_size: 16
54
+ test_max_examples: null
55
+ test_checkpoint_max_examples: null
56
+ test_checkpoint_eval_main: false
57
+ test_hallucination_overlap_floor: 50.0
58
+ test_premature_stop_ratio: 0.3
59
+ test_premature_stop_min_input_words: 30
60
+ test_repetition_penalty: 1.0
61
+ test_no_repeat_ngram_size: 0
62
+ test_baseline: true
63
+ test_entity_dataset_id: PedramR/ASR_Post-processing-eval
64
+ test_typo_dataset_id: PedramR/ASR_Post-processing-eval
65
+ test_entity_row_type: entity
66
+ test_typo_row_type: typo
67
+ test_fix_weight: 0.5
qwen3.5-2b-100pct-masked-weighted-v2/resolved_config.yaml ADDED
@@ -0,0 +1,57 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model_id: Qwen/Qwen3.5-2B
2
+ data_config: configs/data/default.yaml
3
+ train_split: train
4
+ eval_split: null
5
+ eval_fraction: 0.1
6
+ train_fraction: 1.0
7
+ input_column: text_whisper
8
+ target_column: text
9
+ system_prompt: null
10
+ output_dir: outputs/qwen-mask-weighted/qwen3.5-2b-100pct-masked-weighted-v2
11
+ max_length: 2048
12
+ on_long_example: drop
13
+ mask_low_signal_corrections: true
14
+ mask_min_similarity: 0.75
15
+ mask_max_common_freq: 1
16
+ recoverable_correction_weight: 2.0
17
+ short_target_eos_weight: 0.3
18
+ short_target_max_words: 20
19
+ training:
20
+ num_train_epochs: 1.0
21
+ per_device_train_batch_size: 2
22
+ per_device_eval_batch_size: 2
23
+ gradient_accumulation_steps: 16
24
+ learning_rate: 0.0001
25
+ warmup_ratio: 0.05
26
+ logging_steps: 10
27
+ eval_steps: 500
28
+ save_steps: 500
29
+ save_total_limit: 3
30
+ seed: 42
31
+ early_stopping_patience: 3
32
+ lora:
33
+ r: 32
34
+ alpha: 64
35
+ dropout: 0.05
36
+ quantization:
37
+ load_in_4bit: false
38
+ hub:
39
+ push_to_hub: true
40
+ repo_id: PedramR/ASR_Post-processing
41
+ folder: qwen3.5-2b-100pct-masked-weighted-v2
42
+ private: false
43
+ tensorboard:
44
+ logging_dir: outputs/qwen-mask-weighted/qwen3.5-2b-100pct-masked-weighted-v2/tb
45
+ test:
46
+ dataset_id: ErfanRou/callcc-test-1k
47
+ split: test
48
+ input_column: text_whisper
49
+ target_column: text
50
+ max_new_tokens: 1024
51
+ batch_size: 16
52
+ max_examples: null
53
+ checkpoint_max_examples: null
54
+ fix_weight: 0.5
55
+ checkpoint_eval_main: false
56
+ resume_from_checkpoint: false
57
+ use_unsloth: false
qwen3.5-2b-100pct-masked-weighted-v2/tb/events.out.tfevents.1790605476.02b48645612f.4580.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c89e173c0ed135eb5eb1ac01be76e66fd4ba283cddcfbbbadc58112f683dccce
3
+ size 20087
qwen3.5-2b-100pct-masked-weighted-v2/test_eval_baseline/metrics.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "wer": 0.26724399125199005,
3
+ "wer_zwnj_normalized": 0.25653208613594675,
4
+ "exact_match": 0.003033367037411527,
5
+ "fix_rate": 0.028386571209249135,
6
+ "preservation_rate": 0.9043148709482289,
7
+ "targeted_score": 0.46635072107873904,
8
+ "fix_rate_lenient": 0.02983948977751545,
9
+ "preservation_rate_lenient": 0.9093806936843613,
10
+ "targeted_score_lenient": 0.4696100917309384,
11
+ "fix_rate_recoverable": 0.030153097772668967,
12
+ "unrecoverable_targets": 966,
13
+ "hallucination_rate": 0.006066734074823054,
14
+ "premature_stop_rate": 0.006066734074823054,
15
+ "n_examples": 1978
16
+ }
qwen3.5-2b-100pct-masked-weighted-v2/test_eval_baseline/predictions.jsonl ADDED
The diff for this file is too large to render. See raw diff
 
qwen3.5-2b-100pct-masked-weighted-v2/test_eval_entity/metrics.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "wer": 0.25481154129439265,
3
+ "wer_zwnj_normalized": 0.24980518995313886,
4
+ "exact_match": 0.0039603960396039604,
5
+ "fix_rate": 0.2154554278736262,
6
+ "preservation_rate": 0.8851729129805372,
7
+ "targeted_score": 0.5503141704270818,
8
+ "fix_rate_lenient": 0.20545155993431855,
9
+ "preservation_rate_lenient": 0.8867249143604942,
10
+ "targeted_score_lenient": 0.5460882371474064,
11
+ "fix_rate_recoverable": 0.2082457720404283,
12
+ "unrecoverable_targets": 471,
13
+ "hallucination_rate": 0.006930693069306931,
14
+ "premature_stop_rate": 0.04950495049504951,
15
+ "n_examples": 1010
16
+ }
qwen3.5-2b-100pct-masked-weighted-v2/test_eval_entity/predictions.jsonl ADDED
The diff for this file is too large to render. See raw diff
 
qwen3.5-2b-100pct-masked-weighted-v2/test_eval_entity_baseline/metrics.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "wer": 0.2605651092537441,
3
+ "wer_zwnj_normalized": 0.24933551093604894,
4
+ "exact_match": 0.0009900990099009901,
5
+ "fix_rate": 0.029746062560666312,
6
+ "preservation_rate": 0.9081293228245134,
7
+ "targeted_score": 0.4689376926925899,
8
+ "fix_rate_lenient": 0.03126436781609195,
9
+ "preservation_rate_lenient": 0.9134944767329972,
10
+ "targeted_score_lenient": 0.4723794222745446,
11
+ "fix_rate_recoverable": 0.03158877881183495,
12
+ "unrecoverable_targets": 471,
13
+ "hallucination_rate": 0.006930693069306931,
14
+ "premature_stop_rate": 0.0049504950495049506,
15
+ "n_examples": 1010
16
+ }
qwen3.5-2b-100pct-masked-weighted-v2/test_eval_entity_baseline/predictions.jsonl ADDED
The diff for this file is too large to render. See raw diff
 
qwen3.5-2b-100pct-masked-weighted-v2/test_eval_typo/metrics.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "wer": 0.18398815450205508,
3
+ "wer_zwnj_normalized": 0.178116463710398,
4
+ "exact_match": 0.0111731843575419,
5
+ "fix_rate": 0.28247382706475377,
6
+ "preservation_rate": 0.9247640608002823,
7
+ "targeted_score": 0.603618943932518,
8
+ "fix_rate_lenient": 0.2757193127716829,
9
+ "preservation_rate_lenient": 0.9271587213220428,
10
+ "targeted_score_lenient": 0.6014390170468629,
11
+ "fix_rate_recoverable": 0.27923577577157255,
12
+ "unrecoverable_targets": 68,
13
+ "hallucination_rate": 0.008379888268156424,
14
+ "premature_stop_rate": 0.05027932960893855,
15
+ "n_examples": 358
16
+ }
qwen3.5-2b-100pct-masked-weighted-v2/test_eval_typo/predictions.jsonl ADDED
The diff for this file is too large to render. See raw diff
 
qwen3.5-2b-100pct-masked-weighted-v2/test_eval_typo_baseline/metrics.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "wer": 0.19670164152051262,
3
+ "wer_zwnj_normalized": 0.18419238722524317,
4
+ "exact_match": 0.0,
5
+ "fix_rate": 0.042458317177200465,
6
+ "preservation_rate": 0.9313497780260489,
7
+ "targeted_score": 0.4869040476016247,
8
+ "fix_rate_lenient": 0.0461602152763403,
9
+ "preservation_rate_lenient": 0.9384980515104456,
10
+ "targeted_score_lenient": 0.49232913339339296,
11
+ "fix_rate_recoverable": 0.04681923157673735,
12
+ "unrecoverable_targets": 68,
13
+ "hallucination_rate": 0.0111731843575419,
14
+ "premature_stop_rate": 0.002793296089385475,
15
+ "n_examples": 358
16
+ }
qwen3.5-2b-100pct-masked-weighted-v2/test_eval_typo_baseline/predictions.jsonl ADDED
The diff for this file is too large to render. See raw diff