PedramR commited on
Commit
9199b86
·
verified ·
1 Parent(s): f62e8fa

checkpoint-100

Browse files
qwen3.5-2b/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/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
+ "v_proj",
36
+ "up_proj",
37
+ "in_proj_a",
38
+ "gate_proj",
39
+ "out_proj",
40
+ "q_proj",
41
+ "k_proj",
42
+ "in_proj_b",
43
+ "o_proj",
44
+ "in_proj_qkv",
45
+ "in_proj_z",
46
+ "down_proj"
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/best_checkpoint_wer/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b2e249ff273f8df09605bca89a1571361cd7bfc966c1a24f7672d253f488458f
3
+ size 134604152
qwen3.5-2b/best_checkpoint_wer/best_metrics.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "step": 100,
3
+ "wer": 0.46168322007318346,
4
+ "exact_match": 0.00390625,
5
+ "hallucination_rate": 0.09375,
6
+ "n_examples": 256
7
+ }
qwen3.5-2b/best_checkpoint_wer/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:216255d2009f78676e6a71be2913f7485c161494a3b1b2ad6b608621d991684e
3
+ size 269426431
qwen3.5-2b/best_checkpoint_wer/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0bd3c954d401b65bf65ab4946a048a3a709cb2bbbe22c8401248fde3d9108ce2
3
+ size 14645
qwen3.5-2b/best_checkpoint_wer/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5161ab58165f1ba32df42f8d11c180d6e456f2660b1a85bbaf6c2be2e076253a
3
+ size 1465
qwen3.5-2b/best_checkpoint_wer/trainer_state.json ADDED
@@ -0,0 +1,112 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.07914523149980214,
6
+ "eval_steps": 100,
7
+ "global_step": 100,
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.007914523149980214,
14
+ "grad_norm": 1.404574990272522,
15
+ "learning_rate": 7.086614173228347e-06,
16
+ "loss": 0.9118253707885742,
17
+ "step": 10
18
+ },
19
+ {
20
+ "epoch": 0.015829046299960427,
21
+ "grad_norm": 0.868704617023468,
22
+ "learning_rate": 1.4960629921259845e-05,
23
+ "loss": 0.8003530502319336,
24
+ "step": 20
25
+ },
26
+ {
27
+ "epoch": 0.02374356944994064,
28
+ "grad_norm": 0.8633992075920105,
29
+ "learning_rate": 2.283464566929134e-05,
30
+ "loss": 0.7220969676971436,
31
+ "step": 30
32
+ },
33
+ {
34
+ "epoch": 0.031658092599920855,
35
+ "grad_norm": 0.7782989740371704,
36
+ "learning_rate": 3.070866141732284e-05,
37
+ "loss": 0.7230673789978027,
38
+ "step": 40
39
+ },
40
+ {
41
+ "epoch": 0.03957261574990107,
42
+ "grad_norm": 1.0865058898925781,
43
+ "learning_rate": 3.858267716535433e-05,
44
+ "loss": 0.7225656986236573,
45
+ "step": 50
46
+ },
47
+ {
48
+ "epoch": 0.04748713889988128,
49
+ "grad_norm": 0.8686597943305969,
50
+ "learning_rate": 4.645669291338583e-05,
51
+ "loss": 0.6066381454467773,
52
+ "step": 60
53
+ },
54
+ {
55
+ "epoch": 0.055401662049861494,
56
+ "grad_norm": 0.9904829263687134,
57
+ "learning_rate": 5.433070866141733e-05,
58
+ "loss": 0.7127069473266602,
59
+ "step": 70
60
+ },
61
+ {
62
+ "epoch": 0.06331618519984171,
63
+ "grad_norm": 0.9868095517158508,
64
+ "learning_rate": 6.220472440944882e-05,
65
+ "loss": 0.6702294826507569,
66
+ "step": 80
67
+ },
68
+ {
69
+ "epoch": 0.07123070834982193,
70
+ "grad_norm": 1.1342442035675049,
71
+ "learning_rate": 7.007874015748031e-05,
72
+ "loss": 0.6693508148193359,
73
+ "step": 90
74
+ },
75
+ {
76
+ "epoch": 0.07914523149980214,
77
+ "grad_norm": 1.1382489204406738,
78
+ "learning_rate": 7.795275590551181e-05,
79
+ "loss": 0.6664952278137207,
80
+ "step": 100
81
+ },
82
+ {
83
+ "epoch": 0.07914523149980214,
84
+ "eval_loss": 0.6436748504638672,
85
+ "eval_runtime": 461.1275,
86
+ "eval_samples_per_second": 24.29,
87
+ "eval_steps_per_second": 6.074,
88
+ "step": 100
89
+ }
90
+ ],
91
+ "logging_steps": 10,
92
+ "max_steps": 2528,
93
+ "num_input_tokens_seen": 0,
94
+ "num_train_epochs": 2,
95
+ "save_steps": 100,
96
+ "stateful_callbacks": {
97
+ "TrainerControl": {
98
+ "args": {
99
+ "should_epoch_stop": false,
100
+ "should_evaluate": false,
101
+ "should_log": false,
102
+ "should_save": true,
103
+ "should_training_stop": false
104
+ },
105
+ "attributes": {}
106
+ }
107
+ },
108
+ "total_flos": 1.5094044096262656e+16,
109
+ "train_batch_size": 4,
110
+ "trial_name": null,
111
+ "trial_params": null
112
+ }
qwen3.5-2b/best_checkpoint_wer/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:57ac5fc1677740bc79510bd3bb17e1d48a8cb3ceab532db738ff37384e2efc7b
3
+ size 5201
qwen3.5-2b/checkpoint-100/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/checkpoint-100/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
+ "v_proj",
36
+ "up_proj",
37
+ "in_proj_a",
38
+ "gate_proj",
39
+ "out_proj",
40
+ "q_proj",
41
+ "k_proj",
42
+ "in_proj_b",
43
+ "o_proj",
44
+ "in_proj_qkv",
45
+ "in_proj_z",
46
+ "down_proj"
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/checkpoint-100/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b2e249ff273f8df09605bca89a1571361cd7bfc966c1a24f7672d253f488458f
3
+ size 134604152
qwen3.5-2b/checkpoint-100/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:216255d2009f78676e6a71be2913f7485c161494a3b1b2ad6b608621d991684e
3
+ size 269426431
qwen3.5-2b/checkpoint-100/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0bd3c954d401b65bf65ab4946a048a3a709cb2bbbe22c8401248fde3d9108ce2
3
+ size 14645
qwen3.5-2b/checkpoint-100/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5161ab58165f1ba32df42f8d11c180d6e456f2660b1a85bbaf6c2be2e076253a
3
+ size 1465
qwen3.5-2b/checkpoint-100/trainer_state.json ADDED
@@ -0,0 +1,112 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.07914523149980214,
6
+ "eval_steps": 100,
7
+ "global_step": 100,
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.007914523149980214,
14
+ "grad_norm": 1.404574990272522,
15
+ "learning_rate": 7.086614173228347e-06,
16
+ "loss": 0.9118253707885742,
17
+ "step": 10
18
+ },
19
+ {
20
+ "epoch": 0.015829046299960427,
21
+ "grad_norm": 0.868704617023468,
22
+ "learning_rate": 1.4960629921259845e-05,
23
+ "loss": 0.8003530502319336,
24
+ "step": 20
25
+ },
26
+ {
27
+ "epoch": 0.02374356944994064,
28
+ "grad_norm": 0.8633992075920105,
29
+ "learning_rate": 2.283464566929134e-05,
30
+ "loss": 0.7220969676971436,
31
+ "step": 30
32
+ },
33
+ {
34
+ "epoch": 0.031658092599920855,
35
+ "grad_norm": 0.7782989740371704,
36
+ "learning_rate": 3.070866141732284e-05,
37
+ "loss": 0.7230673789978027,
38
+ "step": 40
39
+ },
40
+ {
41
+ "epoch": 0.03957261574990107,
42
+ "grad_norm": 1.0865058898925781,
43
+ "learning_rate": 3.858267716535433e-05,
44
+ "loss": 0.7225656986236573,
45
+ "step": 50
46
+ },
47
+ {
48
+ "epoch": 0.04748713889988128,
49
+ "grad_norm": 0.8686597943305969,
50
+ "learning_rate": 4.645669291338583e-05,
51
+ "loss": 0.6066381454467773,
52
+ "step": 60
53
+ },
54
+ {
55
+ "epoch": 0.055401662049861494,
56
+ "grad_norm": 0.9904829263687134,
57
+ "learning_rate": 5.433070866141733e-05,
58
+ "loss": 0.7127069473266602,
59
+ "step": 70
60
+ },
61
+ {
62
+ "epoch": 0.06331618519984171,
63
+ "grad_norm": 0.9868095517158508,
64
+ "learning_rate": 6.220472440944882e-05,
65
+ "loss": 0.6702294826507569,
66
+ "step": 80
67
+ },
68
+ {
69
+ "epoch": 0.07123070834982193,
70
+ "grad_norm": 1.1342442035675049,
71
+ "learning_rate": 7.007874015748031e-05,
72
+ "loss": 0.6693508148193359,
73
+ "step": 90
74
+ },
75
+ {
76
+ "epoch": 0.07914523149980214,
77
+ "grad_norm": 1.1382489204406738,
78
+ "learning_rate": 7.795275590551181e-05,
79
+ "loss": 0.6664952278137207,
80
+ "step": 100
81
+ },
82
+ {
83
+ "epoch": 0.07914523149980214,
84
+ "eval_loss": 0.6436748504638672,
85
+ "eval_runtime": 461.1275,
86
+ "eval_samples_per_second": 24.29,
87
+ "eval_steps_per_second": 6.074,
88
+ "step": 100
89
+ }
90
+ ],
91
+ "logging_steps": 10,
92
+ "max_steps": 2528,
93
+ "num_input_tokens_seen": 0,
94
+ "num_train_epochs": 2,
95
+ "save_steps": 100,
96
+ "stateful_callbacks": {
97
+ "TrainerControl": {
98
+ "args": {
99
+ "should_epoch_stop": false,
100
+ "should_evaluate": false,
101
+ "should_log": false,
102
+ "should_save": true,
103
+ "should_training_stop": false
104
+ },
105
+ "attributes": {}
106
+ }
107
+ },
108
+ "total_flos": 1.5094044096262656e+16,
109
+ "train_batch_size": 4,
110
+ "trial_name": null,
111
+ "trial_params": null
112
+ }
qwen3.5-2b/checkpoint-100/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:57ac5fc1677740bc79510bd3bb17e1d48a8cb3ceab532db738ff37384e2efc7b
3
+ size 5201
qwen3.5-2b/config.yaml ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model_id: Qwen/Qwen3.5-2B
2
+ dataset_id: ./asr_dataset_curated.jsonl
3
+ train_split: train
4
+ eval_split: null
5
+ eval_fraction: 0.1
6
+ train_fraction: 0.2
7
+ input_column: text_whisper
8
+ target_column: text
9
+ system_prompt: null
10
+ output_dir: outputs/model-comparison/qwen3.5-2b
11
+ max_length: 2048
12
+ on_long_example: drop
13
+ num_train_epochs: 2.0
14
+ per_device_train_batch_size: 4
15
+ per_device_eval_batch_size: 4
16
+ gradient_accumulation_steps: 4
17
+ learning_rate: 0.0001
18
+ warmup_ratio: 0.05
19
+ logging_steps: 10
20
+ eval_steps: 100
21
+ save_steps: 100
22
+ save_total_limit: 3
23
+ seed: 42
24
+ lora_r: 32
25
+ lora_alpha: 64
26
+ lora_dropout: 0.05
27
+ load_in_4bit: false
28
+ push_to_hub: true
29
+ hub_repo_id: PedramR/ASR_Post-processing
30
+ hub_repo_folder: qwen3.5-2b
31
+ hub_private: false
32
+ tensorboard_logging_dir: outputs/model-comparison/qwen3.5-2b/tb
33
+ resume_from_checkpoint: false
34
+ test_dataset_id: ErfanRou/callcc-test-1k
35
+ test_split: test
36
+ test_input_column: text_whisper
37
+ test_target_column: text
38
+ test_max_new_tokens: 1024
39
+ test_batch_size: 32
40
+ test_max_examples: 256
41
+ test_checkpoint_max_examples: 256
42
+ test_hallucination_overlap_floor: 50.0
43
+ test_baseline: true
qwen3.5-2b/resolved_config.yaml ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model_id: Qwen/Qwen3.5-2B
2
+ dataset_id: ./asr_dataset_curated.jsonl
3
+ train_split: train
4
+ eval_split: null
5
+ eval_fraction: 0.1
6
+ train_fraction: 0.2
7
+ input_column: text_whisper
8
+ target_column: text
9
+ system_prompt: null
10
+ output_dir: outputs/model-comparison/qwen3.5-2b
11
+ max_length: 2048
12
+ on_long_example: drop
13
+ training:
14
+ num_train_epochs: 2.0
15
+ per_device_train_batch_size: 4
16
+ per_device_eval_batch_size: 4
17
+ gradient_accumulation_steps: 4
18
+ learning_rate: 0.0001
19
+ warmup_ratio: 0.05
20
+ logging_steps: 10
21
+ eval_steps: 100
22
+ save_steps: 100
23
+ save_total_limit: 3
24
+ seed: 42
25
+ lora:
26
+ r: 32
27
+ alpha: 64
28
+ dropout: 0.05
29
+ quantization:
30
+ load_in_4bit: false
31
+ hub:
32
+ push_to_hub: true
33
+ repo_id: PedramR/ASR_Post-processing
34
+ folder: qwen3.5-2b
35
+ private: false
36
+ tensorboard:
37
+ logging_dir: outputs/model-comparison/qwen3.5-2b/tb
38
+ test:
39
+ dataset_id: ErfanRou/callcc-test-1k
40
+ split: test
41
+ input_column: text_whisper
42
+ target_column: text
43
+ max_new_tokens: 1024
44
+ batch_size: 32
45
+ max_examples: 256
46
+ checkpoint_max_examples: 256
47
+ resume_from_checkpoint: false
qwen3.5-2b/tb/events.out.tfevents.1789917137.fed349279e2c.5234.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d3b069aa8c454d9883a34c2b91f1c35e49efbb70cfda0f42b3562ed70c289156
3
+ size 5626
qwen3.5-2b/tb/events.out.tfevents.1789935605.fed349279e2c.6800.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0139cee80096285d8fea48be42a8f621dd14dbc5d59e0f5e222f23b85ad793ee
3
+ size 8223
qwen3.5-2b/test_eval/metrics.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "wer": 0.46168322007318346,
3
+ "exact_match": 0.00390625,
4
+ "hallucination_rate": 0.09375,
5
+ "n_examples": 256
6
+ }
qwen3.5-2b/test_eval/predictions.jsonl ADDED
The diff for this file is too large to render. See raw diff
 
qwen3.5-2b/test_eval_baseline/metrics.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "wer": 0.27462624150548876,
3
+ "exact_match": 0.0,
4
+ "hallucination_rate": 0.015625,
5
+ "n_examples": 256
6
+ }
qwen3.5-2b/test_eval_baseline/predictions.jsonl ADDED
The diff for this file is too large to render. See raw diff