checkpoint-500
Browse files- qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/README.md +207 -0
- qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/adapter_config.json +56 -0
- qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/adapter_model.safetensors +3 -0
- qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/best_metrics.json +4 -0
- qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/optimizer.pt +3 -0
- qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/rng_state.pth +3 -0
- qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/scheduler.pt +3 -0
- qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/trainer_state.json +384 -0
- qwen3.5-2b-100pct-masked-weighted-v2/best_checkpoint_wer/training_args.bin +3 -0
- qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/README.md +207 -0
- qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/adapter_config.json +56 -0
- qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/adapter_model.safetensors +3 -0
- qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/optimizer.pt +3 -0
- qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/rng_state.pth +3 -0
- qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/scheduler.pt +3 -0
- qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/trainer_state.json +384 -0
- qwen3.5-2b-100pct-masked-weighted-v2/checkpoint-500/training_args.bin +3 -0
- qwen3.5-2b-100pct-masked-weighted-v2/config.yaml +67 -0
- qwen3.5-2b-100pct-masked-weighted-v2/resolved_config.yaml +57 -0
- qwen3.5-2b-100pct-masked-weighted-v2/tb/events.out.tfevents.1790605476.02b48645612f.4580.0 +3 -0
- qwen3.5-2b-100pct-masked-weighted-v2/test_eval_baseline/metrics.json +16 -0
- qwen3.5-2b-100pct-masked-weighted-v2/test_eval_baseline/predictions.jsonl +0 -0
- qwen3.5-2b-100pct-masked-weighted-v2/test_eval_entity/metrics.json +16 -0
- qwen3.5-2b-100pct-masked-weighted-v2/test_eval_entity/predictions.jsonl +0 -0
- qwen3.5-2b-100pct-masked-weighted-v2/test_eval_entity_baseline/metrics.json +16 -0
- qwen3.5-2b-100pct-masked-weighted-v2/test_eval_entity_baseline/predictions.jsonl +0 -0
- qwen3.5-2b-100pct-masked-weighted-v2/test_eval_typo/metrics.json +16 -0
- qwen3.5-2b-100pct-masked-weighted-v2/test_eval_typo/predictions.jsonl +0 -0
- qwen3.5-2b-100pct-masked-weighted-v2/test_eval_typo_baseline/metrics.json +16 -0
- 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
|
|
|