Sentence Similarity
sentence-transformers
Safetensors
bert
feature-extraction
Generated from Trainer
dataset_size:267
loss:ContrastiveLoss
text-embeddings-inference
Instructions to use officialthinkingstack/Sentence-Matcher-v0.3 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use officialthinkingstack/Sentence-Matcher-v0.3 with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("officialthinkingstack/Sentence-Matcher-v0.3") sentences = [ "hypertension\nThe patient's primary diagnosis is hypertension, as stated in the visit note.\nBP medications\nThe patient is on BP medications which are used to treat hypertension.\nBP management\nThe visit note mentions follow-up on BP management, indicating ongoing treatment for hypertension.\nHTN\nHTN is the abbreviation for hypertension, which is the patient's diagnosed condition.\nBP was measured at 138/90\nThis blood pressure reading supports the diagnosis of hypertension as it is elevated.\nmonthly bp at home have been around that number or higher\nConsistently high blood pressure readings confirm the presence of hypertension.\nmost likely diagnosis for this patient is hypertension\nThe visit note explicitly states that hypertension is the most likely diagnosis.", "Anemia, Unspecified", "Essential (Primary) Hypertension", "Dehydration" ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4] - Notebooks
- Google Colab
- Kaggle
Upload folder using huggingface_hub
Browse files- 1_Pooling/config.json +10 -0
- README.md +438 -0
- config.json +25 -0
- config_sentence_transformers.json +10 -0
- model.safetensors +3 -0
- modules.json +20 -0
- optimizer.pt +3 -0
- rng_state.pth +3 -0
- scheduler.pt +3 -0
- sentence_bert_config.json +4 -0
- special_tokens_map.json +37 -0
- tokenizer.json +0 -0
- tokenizer_config.json +65 -0
- trainer_state.json +153 -0
- training_args.bin +3 -0
- vocab.txt +0 -0
1_Pooling/config.json
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"word_embedding_dimension": 384,
|
| 3 |
+
"pooling_mode_cls_token": false,
|
| 4 |
+
"pooling_mode_mean_tokens": true,
|
| 5 |
+
"pooling_mode_max_tokens": false,
|
| 6 |
+
"pooling_mode_mean_sqrt_len_tokens": false,
|
| 7 |
+
"pooling_mode_weightedmean_tokens": false,
|
| 8 |
+
"pooling_mode_lasttoken": false,
|
| 9 |
+
"include_prompt": true
|
| 10 |
+
}
|
README.md
ADDED
|
@@ -0,0 +1,438 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
tags:
|
| 3 |
+
- sentence-transformers
|
| 4 |
+
- sentence-similarity
|
| 5 |
+
- feature-extraction
|
| 6 |
+
- generated_from_trainer
|
| 7 |
+
- dataset_size:267
|
| 8 |
+
- loss:ContrastiveLoss
|
| 9 |
+
base_model: sentence-transformers/all-MiniLM-L6-v2
|
| 10 |
+
widget:
|
| 11 |
+
- source_sentence: 'hypertension
|
| 12 |
+
|
| 13 |
+
The patient''s primary diagnosis is hypertension, as stated in the visit note.
|
| 14 |
+
|
| 15 |
+
BP medications
|
| 16 |
+
|
| 17 |
+
The patient is on BP medications which are used to treat hypertension.
|
| 18 |
+
|
| 19 |
+
BP management
|
| 20 |
+
|
| 21 |
+
The visit note mentions follow-up on BP management, indicating ongoing treatment
|
| 22 |
+
for hypertension.
|
| 23 |
+
|
| 24 |
+
HTN
|
| 25 |
+
|
| 26 |
+
HTN is the abbreviation for hypertension, which is the patient''s diagnosed condition.
|
| 27 |
+
|
| 28 |
+
BP was measured at 138/90
|
| 29 |
+
|
| 30 |
+
This blood pressure reading supports the diagnosis of hypertension as it is elevated.
|
| 31 |
+
|
| 32 |
+
monthly bp at home have been around that number or higher
|
| 33 |
+
|
| 34 |
+
Consistently high blood pressure readings confirm the presence of hypertension.
|
| 35 |
+
|
| 36 |
+
most likely diagnosis for this patient is hypertension
|
| 37 |
+
|
| 38 |
+
The visit note explicitly states that hypertension is the most likely diagnosis.'
|
| 39 |
+
sentences:
|
| 40 |
+
- Anemia, Unspecified
|
| 41 |
+
- Essential (Primary) Hypertension
|
| 42 |
+
- Dehydration
|
| 43 |
+
- source_sentence: 'BMI ABOVE NORMAL PARAM F/U DOCUMENTED
|
| 44 |
+
|
| 45 |
+
This phrase indicates that the patient''s BMI is above normal parameters and requires
|
| 46 |
+
follow-up, which is a key indicator for obesity classification.
|
| 47 |
+
|
| 48 |
+
34.11
|
| 49 |
+
|
| 50 |
+
The specific BMI value of 34.11 falls within the range for Class 1 obesity (30.0-34.9),
|
| 51 |
+
providing numerical confirmation of the diagnosis.
|
| 52 |
+
|
| 53 |
+
Class 1 obesity
|
| 54 |
+
|
| 55 |
+
This is the explicit statement of the patient''s condition, directly aligning
|
| 56 |
+
with the ICD code E66.811 for Class 1 obesity.'
|
| 57 |
+
sentences:
|
| 58 |
+
- Obesity, Class 1
|
| 59 |
+
- Hypothyroidism, Unspecified
|
| 60 |
+
- Overweight
|
| 61 |
+
- source_sentence: 'anxious and uses food for comfort
|
| 62 |
+
|
| 63 |
+
This phrase indicates the presence of anxiety symptoms, specifically using food
|
| 64 |
+
as a coping mechanism, which aligns with an unspecified anxiety disorder.'
|
| 65 |
+
sentences:
|
| 66 |
+
- Essential (Primary) Hypertension
|
| 67 |
+
- Essential (Primary) Hypertension
|
| 68 |
+
- Anxiety Disorder, Unspecified
|
| 69 |
+
- source_sentence: 'compression stockings
|
| 70 |
+
|
| 71 |
+
Compression stockings are a treatment for venous insufficiency, which can cause
|
| 72 |
+
localized edema.
|
| 73 |
+
|
| 74 |
+
venous insufficiency
|
| 75 |
+
|
| 76 |
+
Venous insufficiency is a condition that leads to leg edema, which is a type of
|
| 77 |
+
localized edema.
|
| 78 |
+
|
| 79 |
+
Leg edema
|
| 80 |
+
|
| 81 |
+
Leg edema is a direct symptom of localized edema.
|
| 82 |
+
|
| 83 |
+
edema
|
| 84 |
+
|
| 85 |
+
Edema refers to swelling caused by fluid retention, which aligns with the ICD
|
| 86 |
+
code R60.0 for Localized Edema.'
|
| 87 |
+
sentences:
|
| 88 |
+
- Nasal Congestion
|
| 89 |
+
- Localized Edema
|
| 90 |
+
- Essential (Primary) Hypertension
|
| 91 |
+
- source_sentence: 'Had lithotripsy and passed an 8x5 mm stone on L.
|
| 92 |
+
|
| 93 |
+
This phrase indicates a history of urinary calculi as evidenced by the treatment
|
| 94 |
+
(lithotripsy) for kidney stones.'
|
| 95 |
+
sentences:
|
| 96 |
+
- Pure Hypercholesterolemia, Unspecified
|
| 97 |
+
- Personal History Of Urinary Calculi
|
| 98 |
+
- Menopausal And Female Climacteric States
|
| 99 |
+
pipeline_tag: sentence-similarity
|
| 100 |
+
library_name: sentence-transformers
|
| 101 |
+
---
|
| 102 |
+
|
| 103 |
+
# SentenceTransformer based on sentence-transformers/all-MiniLM-L6-v2
|
| 104 |
+
|
| 105 |
+
This is a [sentence-transformers](https://www.SBERT.net) model finetuned from [sentence-transformers/all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2). It maps sentences & paragraphs to a 384-dimensional dense vector space and can be used for semantic textual similarity, semantic search, paraphrase mining, text classification, clustering, and more.
|
| 106 |
+
|
| 107 |
+
## Model Details
|
| 108 |
+
|
| 109 |
+
### Model Description
|
| 110 |
+
- **Model Type:** Sentence Transformer
|
| 111 |
+
- **Base model:** [sentence-transformers/all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2) <!-- at revision c9745ed1d9f207416be6d2e6f8de32d1f16199bf -->
|
| 112 |
+
- **Maximum Sequence Length:** 256 tokens
|
| 113 |
+
- **Output Dimensionality:** 384 dimensions
|
| 114 |
+
- **Similarity Function:** Cosine Similarity
|
| 115 |
+
<!-- - **Training Dataset:** Unknown -->
|
| 116 |
+
<!-- - **Language:** Unknown -->
|
| 117 |
+
<!-- - **License:** Unknown -->
|
| 118 |
+
|
| 119 |
+
### Model Sources
|
| 120 |
+
|
| 121 |
+
- **Documentation:** [Sentence Transformers Documentation](https://sbert.net)
|
| 122 |
+
- **Repository:** [Sentence Transformers on GitHub](https://github.com/UKPLab/sentence-transformers)
|
| 123 |
+
- **Hugging Face:** [Sentence Transformers on Hugging Face](https://huggingface.co/models?library=sentence-transformers)
|
| 124 |
+
|
| 125 |
+
### Full Model Architecture
|
| 126 |
+
|
| 127 |
+
```
|
| 128 |
+
SentenceTransformer(
|
| 129 |
+
(0): Transformer({'max_seq_length': 256, 'do_lower_case': False}) with Transformer model: BertModel
|
| 130 |
+
(1): Pooling({'word_embedding_dimension': 384, 'pooling_mode_cls_token': False, 'pooling_mode_mean_tokens': True, 'pooling_mode_max_tokens': False, 'pooling_mode_mean_sqrt_len_tokens': False, 'pooling_mode_weightedmean_tokens': False, 'pooling_mode_lasttoken': False, 'include_prompt': True})
|
| 131 |
+
(2): Normalize()
|
| 132 |
+
)
|
| 133 |
+
```
|
| 134 |
+
|
| 135 |
+
## Usage
|
| 136 |
+
|
| 137 |
+
### Direct Usage (Sentence Transformers)
|
| 138 |
+
|
| 139 |
+
First install the Sentence Transformers library:
|
| 140 |
+
|
| 141 |
+
```bash
|
| 142 |
+
pip install -U sentence-transformers
|
| 143 |
+
```
|
| 144 |
+
|
| 145 |
+
Then you can load this model and run inference.
|
| 146 |
+
```python
|
| 147 |
+
from sentence_transformers import SentenceTransformer
|
| 148 |
+
|
| 149 |
+
# Download from the 🤗 Hub
|
| 150 |
+
model = SentenceTransformer("sentence_transformers_model_id")
|
| 151 |
+
# Run inference
|
| 152 |
+
sentences = [
|
| 153 |
+
'Had lithotripsy and passed an 8x5 mm stone on L.\nThis phrase indicates a history of urinary calculi as evidenced by the treatment (lithotripsy) for kidney stones.',
|
| 154 |
+
'Personal History Of Urinary Calculi',
|
| 155 |
+
'Pure Hypercholesterolemia, Unspecified',
|
| 156 |
+
]
|
| 157 |
+
embeddings = model.encode(sentences)
|
| 158 |
+
print(embeddings.shape)
|
| 159 |
+
# [3, 384]
|
| 160 |
+
|
| 161 |
+
# Get the similarity scores for the embeddings
|
| 162 |
+
similarities = model.similarity(embeddings, embeddings)
|
| 163 |
+
print(similarities.shape)
|
| 164 |
+
# [3, 3]
|
| 165 |
+
```
|
| 166 |
+
|
| 167 |
+
<!--
|
| 168 |
+
### Direct Usage (Transformers)
|
| 169 |
+
|
| 170 |
+
<details><summary>Click to see the direct usage in Transformers</summary>
|
| 171 |
+
|
| 172 |
+
</details>
|
| 173 |
+
-->
|
| 174 |
+
|
| 175 |
+
<!--
|
| 176 |
+
### Downstream Usage (Sentence Transformers)
|
| 177 |
+
|
| 178 |
+
You can finetune this model on your own dataset.
|
| 179 |
+
|
| 180 |
+
<details><summary>Click to expand</summary>
|
| 181 |
+
|
| 182 |
+
</details>
|
| 183 |
+
-->
|
| 184 |
+
|
| 185 |
+
<!--
|
| 186 |
+
### Out-of-Scope Use
|
| 187 |
+
|
| 188 |
+
*List how the model may foreseeably be misused and address what users ought not to do with the model.*
|
| 189 |
+
-->
|
| 190 |
+
|
| 191 |
+
<!--
|
| 192 |
+
## Bias, Risks and Limitations
|
| 193 |
+
|
| 194 |
+
*What are the known or foreseeable issues stemming from this model? You could also flag here known failure cases or weaknesses of the model.*
|
| 195 |
+
-->
|
| 196 |
+
|
| 197 |
+
<!--
|
| 198 |
+
### Recommendations
|
| 199 |
+
|
| 200 |
+
*What are recommendations with respect to the foreseeable issues? For example, filtering explicit content.*
|
| 201 |
+
-->
|
| 202 |
+
|
| 203 |
+
## Training Details
|
| 204 |
+
|
| 205 |
+
### Training Dataset
|
| 206 |
+
|
| 207 |
+
#### Unnamed Dataset
|
| 208 |
+
|
| 209 |
+
* Size: 267 training samples
|
| 210 |
+
* Columns: <code>anchor</code>, <code>positive</code>, and <code>label</code>
|
| 211 |
+
* Approximate statistics based on the first 267 samples:
|
| 212 |
+
| | anchor | positive | label |
|
| 213 |
+
|:--------|:------------------------------------------------------------------------------------|:---------------------------------------------------------------------------------|:--------------------------------------------------------------|
|
| 214 |
+
| type | string | string | float |
|
| 215 |
+
| details | <ul><li>min: 12 tokens</li><li>mean: 94.12 tokens</li><li>max: 256 tokens</li></ul> | <ul><li>min: 3 tokens</li><li>mean: 9.77 tokens</li><li>max: 23 tokens</li></ul> | <ul><li>min: 1.0</li><li>mean: 1.0</li><li>max: 1.0</li></ul> |
|
| 216 |
+
* Samples:
|
| 217 |
+
| anchor | positive | label |
|
| 218 |
+
|:------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|:------------------------------------------------------------|:-----------------|
|
| 219 |
+
| <code>T2DM<br>Directly indicates the diagnosis of Type 2 Diabetes Mellitus without complications as stated in the Problem/Dx section.</code> | <code>Type 2 Diabetes Mellitus Without Complications</code> | <code>1.0</code> |
|
| 220 |
+
| <code>Atorvastatin<br>Atorvastatin is a statin medication prescribed to lower cholesterol levels, directly addressing hypercholesterolemia.<br>Hyperlipidemia<br>Hyperlipidemia is a broader term that includes high cholesterol (hypercholesterolemia), which is explicitly mentioned in the assessment.<br>statin therapy<br>Statin therapy, including Atorvastatin, is specifically noted as part of the treatment plan for managing high cholesterol.<br>Hypercholesterolemia<br>Explicitly listed under assessment as a condition being managed, aligning with the ICD code E78.00.</code> | <code>Pure Hypercholesterolemia, Unspecified</code> | <code>1.0</code> |
|
| 221 |
+
| <code>Encounter for immunization (Z23)<br>This phrase directly indicates the ICD code Z23 and its description as the reason for the encounter.</code> | <code>Encounter For Immunization</code> | <code>1.0</code> |
|
| 222 |
+
* Loss: [<code>ContrastiveLoss</code>](https://sbert.net/docs/package_reference/sentence_transformer/losses.html#contrastiveloss) with these parameters:
|
| 223 |
+
```json
|
| 224 |
+
{
|
| 225 |
+
"distance_metric": "SiameseDistanceMetric.COSINE_DISTANCE",
|
| 226 |
+
"margin": 0.5,
|
| 227 |
+
"size_average": true
|
| 228 |
+
}
|
| 229 |
+
```
|
| 230 |
+
|
| 231 |
+
### Training Hyperparameters
|
| 232 |
+
#### Non-Default Hyperparameters
|
| 233 |
+
|
| 234 |
+
- `per_device_train_batch_size`: 16
|
| 235 |
+
- `per_device_eval_batch_size`: 16
|
| 236 |
+
- `learning_rate`: 2e-05
|
| 237 |
+
- `num_train_epochs`: 1
|
| 238 |
+
- `warmup_ratio`: 0.1
|
| 239 |
+
|
| 240 |
+
#### All Hyperparameters
|
| 241 |
+
<details><summary>Click to expand</summary>
|
| 242 |
+
|
| 243 |
+
- `overwrite_output_dir`: False
|
| 244 |
+
- `do_predict`: False
|
| 245 |
+
- `eval_strategy`: no
|
| 246 |
+
- `prediction_loss_only`: True
|
| 247 |
+
- `per_device_train_batch_size`: 16
|
| 248 |
+
- `per_device_eval_batch_size`: 16
|
| 249 |
+
- `per_gpu_train_batch_size`: None
|
| 250 |
+
- `per_gpu_eval_batch_size`: None
|
| 251 |
+
- `gradient_accumulation_steps`: 1
|
| 252 |
+
- `eval_accumulation_steps`: None
|
| 253 |
+
- `torch_empty_cache_steps`: None
|
| 254 |
+
- `learning_rate`: 2e-05
|
| 255 |
+
- `weight_decay`: 0.0
|
| 256 |
+
- `adam_beta1`: 0.9
|
| 257 |
+
- `adam_beta2`: 0.999
|
| 258 |
+
- `adam_epsilon`: 1e-08
|
| 259 |
+
- `max_grad_norm`: 1.0
|
| 260 |
+
- `num_train_epochs`: 1
|
| 261 |
+
- `max_steps`: -1
|
| 262 |
+
- `lr_scheduler_type`: linear
|
| 263 |
+
- `lr_scheduler_kwargs`: {}
|
| 264 |
+
- `warmup_ratio`: 0.1
|
| 265 |
+
- `warmup_steps`: 0
|
| 266 |
+
- `log_level`: passive
|
| 267 |
+
- `log_level_replica`: warning
|
| 268 |
+
- `log_on_each_node`: True
|
| 269 |
+
- `logging_nan_inf_filter`: True
|
| 270 |
+
- `save_safetensors`: True
|
| 271 |
+
- `save_on_each_node`: False
|
| 272 |
+
- `save_only_model`: False
|
| 273 |
+
- `restore_callback_states_from_checkpoint`: False
|
| 274 |
+
- `no_cuda`: False
|
| 275 |
+
- `use_cpu`: False
|
| 276 |
+
- `use_mps_device`: False
|
| 277 |
+
- `seed`: 42
|
| 278 |
+
- `data_seed`: None
|
| 279 |
+
- `jit_mode_eval`: False
|
| 280 |
+
- `use_ipex`: False
|
| 281 |
+
- `bf16`: False
|
| 282 |
+
- `fp16`: False
|
| 283 |
+
- `fp16_opt_level`: O1
|
| 284 |
+
- `half_precision_backend`: auto
|
| 285 |
+
- `bf16_full_eval`: False
|
| 286 |
+
- `fp16_full_eval`: False
|
| 287 |
+
- `tf32`: None
|
| 288 |
+
- `local_rank`: 0
|
| 289 |
+
- `ddp_backend`: None
|
| 290 |
+
- `tpu_num_cores`: None
|
| 291 |
+
- `tpu_metrics_debug`: False
|
| 292 |
+
- `debug`: []
|
| 293 |
+
- `dataloader_drop_last`: False
|
| 294 |
+
- `dataloader_num_workers`: 0
|
| 295 |
+
- `dataloader_prefetch_factor`: None
|
| 296 |
+
- `past_index`: -1
|
| 297 |
+
- `disable_tqdm`: False
|
| 298 |
+
- `remove_unused_columns`: True
|
| 299 |
+
- `label_names`: None
|
| 300 |
+
- `load_best_model_at_end`: False
|
| 301 |
+
- `ignore_data_skip`: False
|
| 302 |
+
- `fsdp`: []
|
| 303 |
+
- `fsdp_min_num_params`: 0
|
| 304 |
+
- `fsdp_config`: {'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}
|
| 305 |
+
- `tp_size`: 0
|
| 306 |
+
- `fsdp_transformer_layer_cls_to_wrap`: None
|
| 307 |
+
- `accelerator_config`: {'split_batches': False, 'dispatch_batches': None, 'even_batches': True, 'use_seedable_sampler': True, 'non_blocking': False, 'gradient_accumulation_kwargs': None}
|
| 308 |
+
- `deepspeed`: None
|
| 309 |
+
- `label_smoothing_factor`: 0.0
|
| 310 |
+
- `optim`: adamw_torch
|
| 311 |
+
- `optim_args`: None
|
| 312 |
+
- `adafactor`: False
|
| 313 |
+
- `group_by_length`: False
|
| 314 |
+
- `length_column_name`: length
|
| 315 |
+
- `ddp_find_unused_parameters`: None
|
| 316 |
+
- `ddp_bucket_cap_mb`: None
|
| 317 |
+
- `ddp_broadcast_buffers`: False
|
| 318 |
+
- `dataloader_pin_memory`: True
|
| 319 |
+
- `dataloader_persistent_workers`: False
|
| 320 |
+
- `skip_memory_metrics`: True
|
| 321 |
+
- `use_legacy_prediction_loop`: False
|
| 322 |
+
- `push_to_hub`: False
|
| 323 |
+
- `resume_from_checkpoint`: None
|
| 324 |
+
- `hub_model_id`: None
|
| 325 |
+
- `hub_strategy`: every_save
|
| 326 |
+
- `hub_private_repo`: None
|
| 327 |
+
- `hub_always_push`: False
|
| 328 |
+
- `gradient_checkpointing`: False
|
| 329 |
+
- `gradient_checkpointing_kwargs`: None
|
| 330 |
+
- `include_inputs_for_metrics`: False
|
| 331 |
+
- `include_for_metrics`: []
|
| 332 |
+
- `eval_do_concat_batches`: True
|
| 333 |
+
- `fp16_backend`: auto
|
| 334 |
+
- `push_to_hub_model_id`: None
|
| 335 |
+
- `push_to_hub_organization`: None
|
| 336 |
+
- `mp_parameters`:
|
| 337 |
+
- `auto_find_batch_size`: False
|
| 338 |
+
- `full_determinism`: False
|
| 339 |
+
- `torchdynamo`: None
|
| 340 |
+
- `ray_scope`: last
|
| 341 |
+
- `ddp_timeout`: 1800
|
| 342 |
+
- `torch_compile`: False
|
| 343 |
+
- `torch_compile_backend`: None
|
| 344 |
+
- `torch_compile_mode`: None
|
| 345 |
+
- `include_tokens_per_second`: False
|
| 346 |
+
- `include_num_input_tokens_seen`: False
|
| 347 |
+
- `neftune_noise_alpha`: None
|
| 348 |
+
- `optim_target_modules`: None
|
| 349 |
+
- `batch_eval_metrics`: False
|
| 350 |
+
- `eval_on_start`: False
|
| 351 |
+
- `use_liger_kernel`: False
|
| 352 |
+
- `eval_use_gather_object`: False
|
| 353 |
+
- `average_tokens_across_devices`: False
|
| 354 |
+
- `prompts`: None
|
| 355 |
+
- `batch_sampler`: batch_sampler
|
| 356 |
+
- `multi_dataset_batch_sampler`: proportional
|
| 357 |
+
|
| 358 |
+
</details>
|
| 359 |
+
|
| 360 |
+
### Training Logs
|
| 361 |
+
| Epoch | Step | Training Loss |
|
| 362 |
+
|:------:|:----:|:-------------:|
|
| 363 |
+
| 0.0588 | 1 | 0.1007 |
|
| 364 |
+
| 0.1176 | 2 | 0.1131 |
|
| 365 |
+
| 0.1765 | 3 | 0.099 |
|
| 366 |
+
| 0.2353 | 4 | 0.0867 |
|
| 367 |
+
| 0.2941 | 5 | 0.0682 |
|
| 368 |
+
| 0.3529 | 6 | 0.1019 |
|
| 369 |
+
| 0.4118 | 7 | 0.0618 |
|
| 370 |
+
| 0.4706 | 8 | 0.0623 |
|
| 371 |
+
| 0.5294 | 9 | 0.0564 |
|
| 372 |
+
| 0.5882 | 10 | 0.0521 |
|
| 373 |
+
| 0.6471 | 11 | 0.0545 |
|
| 374 |
+
| 0.7059 | 12 | 0.0335 |
|
| 375 |
+
| 0.7647 | 13 | 0.0593 |
|
| 376 |
+
| 0.8235 | 14 | 0.0381 |
|
| 377 |
+
| 0.8824 | 15 | 0.0308 |
|
| 378 |
+
| 0.9412 | 16 | 0.0487 |
|
| 379 |
+
| 1.0 | 17 | 0.0398 |
|
| 380 |
+
|
| 381 |
+
|
| 382 |
+
### Framework Versions
|
| 383 |
+
- Python: 3.11.12
|
| 384 |
+
- Sentence Transformers: 3.4.1
|
| 385 |
+
- Transformers: 4.51.3
|
| 386 |
+
- PyTorch: 2.6.0+cu124
|
| 387 |
+
- Accelerate: 1.5.2
|
| 388 |
+
- Datasets: 3.5.0
|
| 389 |
+
- Tokenizers: 0.21.1
|
| 390 |
+
|
| 391 |
+
## Citation
|
| 392 |
+
|
| 393 |
+
### BibTeX
|
| 394 |
+
|
| 395 |
+
#### Sentence Transformers
|
| 396 |
+
```bibtex
|
| 397 |
+
@inproceedings{reimers-2019-sentence-bert,
|
| 398 |
+
title = "Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks",
|
| 399 |
+
author = "Reimers, Nils and Gurevych, Iryna",
|
| 400 |
+
booktitle = "Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing",
|
| 401 |
+
month = "11",
|
| 402 |
+
year = "2019",
|
| 403 |
+
publisher = "Association for Computational Linguistics",
|
| 404 |
+
url = "https://arxiv.org/abs/1908.10084",
|
| 405 |
+
}
|
| 406 |
+
```
|
| 407 |
+
|
| 408 |
+
#### ContrastiveLoss
|
| 409 |
+
```bibtex
|
| 410 |
+
@inproceedings{hadsell2006dimensionality,
|
| 411 |
+
author={Hadsell, R. and Chopra, S. and LeCun, Y.},
|
| 412 |
+
booktitle={2006 IEEE Computer Society Conference on Computer Vision and Pattern Recognition (CVPR'06)},
|
| 413 |
+
title={Dimensionality Reduction by Learning an Invariant Mapping},
|
| 414 |
+
year={2006},
|
| 415 |
+
volume={2},
|
| 416 |
+
number={},
|
| 417 |
+
pages={1735-1742},
|
| 418 |
+
doi={10.1109/CVPR.2006.100}
|
| 419 |
+
}
|
| 420 |
+
```
|
| 421 |
+
|
| 422 |
+
<!--
|
| 423 |
+
## Glossary
|
| 424 |
+
|
| 425 |
+
*Clearly define terms in order to be accessible across audiences.*
|
| 426 |
+
-->
|
| 427 |
+
|
| 428 |
+
<!--
|
| 429 |
+
## Model Card Authors
|
| 430 |
+
|
| 431 |
+
*Lists the people who create the model card, providing recognition and accountability for the detailed work that goes into its construction.*
|
| 432 |
+
-->
|
| 433 |
+
|
| 434 |
+
<!--
|
| 435 |
+
## Model Card Contact
|
| 436 |
+
|
| 437 |
+
*Provides a way for people who have updates to the Model Card, suggestions, or questions, to contact the Model Card authors.*
|
| 438 |
+
-->
|
config.json
ADDED
|
@@ -0,0 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"BertModel"
|
| 4 |
+
],
|
| 5 |
+
"attention_probs_dropout_prob": 0.1,
|
| 6 |
+
"classifier_dropout": null,
|
| 7 |
+
"gradient_checkpointing": false,
|
| 8 |
+
"hidden_act": "gelu",
|
| 9 |
+
"hidden_dropout_prob": 0.1,
|
| 10 |
+
"hidden_size": 384,
|
| 11 |
+
"initializer_range": 0.02,
|
| 12 |
+
"intermediate_size": 1536,
|
| 13 |
+
"layer_norm_eps": 1e-12,
|
| 14 |
+
"max_position_embeddings": 512,
|
| 15 |
+
"model_type": "bert",
|
| 16 |
+
"num_attention_heads": 12,
|
| 17 |
+
"num_hidden_layers": 6,
|
| 18 |
+
"pad_token_id": 0,
|
| 19 |
+
"position_embedding_type": "absolute",
|
| 20 |
+
"torch_dtype": "float32",
|
| 21 |
+
"transformers_version": "4.51.3",
|
| 22 |
+
"type_vocab_size": 2,
|
| 23 |
+
"use_cache": true,
|
| 24 |
+
"vocab_size": 30522
|
| 25 |
+
}
|
config_sentence_transformers.json
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"__version__": {
|
| 3 |
+
"sentence_transformers": "3.4.1",
|
| 4 |
+
"transformers": "4.51.3",
|
| 5 |
+
"pytorch": "2.6.0+cu124"
|
| 6 |
+
},
|
| 7 |
+
"prompts": {},
|
| 8 |
+
"default_prompt_name": null,
|
| 9 |
+
"similarity_fn_name": "cosine"
|
| 10 |
+
}
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:808a45d80f4c8edf10b99da34271c61376209440d587d4b85213e9735fcd6ea8
|
| 3 |
+
size 90864192
|
modules.json
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[
|
| 2 |
+
{
|
| 3 |
+
"idx": 0,
|
| 4 |
+
"name": "0",
|
| 5 |
+
"path": "",
|
| 6 |
+
"type": "sentence_transformers.models.Transformer"
|
| 7 |
+
},
|
| 8 |
+
{
|
| 9 |
+
"idx": 1,
|
| 10 |
+
"name": "1",
|
| 11 |
+
"path": "1_Pooling",
|
| 12 |
+
"type": "sentence_transformers.models.Pooling"
|
| 13 |
+
},
|
| 14 |
+
{
|
| 15 |
+
"idx": 2,
|
| 16 |
+
"name": "2",
|
| 17 |
+
"path": "2_Normalize",
|
| 18 |
+
"type": "sentence_transformers.models.Normalize"
|
| 19 |
+
}
|
| 20 |
+
]
|
optimizer.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1801eded5c67cff972a43417c225fe9626f4012a74f5967823bf0f694e059b24
|
| 3 |
+
size 180607738
|
rng_state.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:be9876d8593203eb43609aa7e32536f6eff78deb124e080058fb7685c86aa1b7
|
| 3 |
+
size 14244
|
scheduler.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:713b9413f23613a4e7a09196bef36c96e72093dde2a9bc9b9c3a3215a82bcd1f
|
| 3 |
+
size 1064
|
sentence_bert_config.json
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"max_seq_length": 256,
|
| 3 |
+
"do_lower_case": false
|
| 4 |
+
}
|
special_tokens_map.json
ADDED
|
@@ -0,0 +1,37 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"cls_token": {
|
| 3 |
+
"content": "[CLS]",
|
| 4 |
+
"lstrip": false,
|
| 5 |
+
"normalized": false,
|
| 6 |
+
"rstrip": false,
|
| 7 |
+
"single_word": false
|
| 8 |
+
},
|
| 9 |
+
"mask_token": {
|
| 10 |
+
"content": "[MASK]",
|
| 11 |
+
"lstrip": false,
|
| 12 |
+
"normalized": false,
|
| 13 |
+
"rstrip": false,
|
| 14 |
+
"single_word": false
|
| 15 |
+
},
|
| 16 |
+
"pad_token": {
|
| 17 |
+
"content": "[PAD]",
|
| 18 |
+
"lstrip": false,
|
| 19 |
+
"normalized": false,
|
| 20 |
+
"rstrip": false,
|
| 21 |
+
"single_word": false
|
| 22 |
+
},
|
| 23 |
+
"sep_token": {
|
| 24 |
+
"content": "[SEP]",
|
| 25 |
+
"lstrip": false,
|
| 26 |
+
"normalized": false,
|
| 27 |
+
"rstrip": false,
|
| 28 |
+
"single_word": false
|
| 29 |
+
},
|
| 30 |
+
"unk_token": {
|
| 31 |
+
"content": "[UNK]",
|
| 32 |
+
"lstrip": false,
|
| 33 |
+
"normalized": false,
|
| 34 |
+
"rstrip": false,
|
| 35 |
+
"single_word": false
|
| 36 |
+
}
|
| 37 |
+
}
|
tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
tokenizer_config.json
ADDED
|
@@ -0,0 +1,65 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"added_tokens_decoder": {
|
| 3 |
+
"0": {
|
| 4 |
+
"content": "[PAD]",
|
| 5 |
+
"lstrip": false,
|
| 6 |
+
"normalized": false,
|
| 7 |
+
"rstrip": false,
|
| 8 |
+
"single_word": false,
|
| 9 |
+
"special": true
|
| 10 |
+
},
|
| 11 |
+
"100": {
|
| 12 |
+
"content": "[UNK]",
|
| 13 |
+
"lstrip": false,
|
| 14 |
+
"normalized": false,
|
| 15 |
+
"rstrip": false,
|
| 16 |
+
"single_word": false,
|
| 17 |
+
"special": true
|
| 18 |
+
},
|
| 19 |
+
"101": {
|
| 20 |
+
"content": "[CLS]",
|
| 21 |
+
"lstrip": false,
|
| 22 |
+
"normalized": false,
|
| 23 |
+
"rstrip": false,
|
| 24 |
+
"single_word": false,
|
| 25 |
+
"special": true
|
| 26 |
+
},
|
| 27 |
+
"102": {
|
| 28 |
+
"content": "[SEP]",
|
| 29 |
+
"lstrip": false,
|
| 30 |
+
"normalized": false,
|
| 31 |
+
"rstrip": false,
|
| 32 |
+
"single_word": false,
|
| 33 |
+
"special": true
|
| 34 |
+
},
|
| 35 |
+
"103": {
|
| 36 |
+
"content": "[MASK]",
|
| 37 |
+
"lstrip": false,
|
| 38 |
+
"normalized": false,
|
| 39 |
+
"rstrip": false,
|
| 40 |
+
"single_word": false,
|
| 41 |
+
"special": true
|
| 42 |
+
}
|
| 43 |
+
},
|
| 44 |
+
"clean_up_tokenization_spaces": false,
|
| 45 |
+
"cls_token": "[CLS]",
|
| 46 |
+
"do_basic_tokenize": true,
|
| 47 |
+
"do_lower_case": true,
|
| 48 |
+
"extra_special_tokens": {},
|
| 49 |
+
"mask_token": "[MASK]",
|
| 50 |
+
"max_length": 128,
|
| 51 |
+
"model_max_length": 256,
|
| 52 |
+
"never_split": null,
|
| 53 |
+
"pad_to_multiple_of": null,
|
| 54 |
+
"pad_token": "[PAD]",
|
| 55 |
+
"pad_token_type_id": 0,
|
| 56 |
+
"padding_side": "right",
|
| 57 |
+
"sep_token": "[SEP]",
|
| 58 |
+
"stride": 0,
|
| 59 |
+
"strip_accents": null,
|
| 60 |
+
"tokenize_chinese_chars": true,
|
| 61 |
+
"tokenizer_class": "BertTokenizer",
|
| 62 |
+
"truncation_side": "right",
|
| 63 |
+
"truncation_strategy": "longest_first",
|
| 64 |
+
"unk_token": "[UNK]"
|
| 65 |
+
}
|
trainer_state.json
ADDED
|
@@ -0,0 +1,153 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"best_global_step": null,
|
| 3 |
+
"best_metric": null,
|
| 4 |
+
"best_model_checkpoint": null,
|
| 5 |
+
"epoch": 1.0,
|
| 6 |
+
"eval_steps": 500,
|
| 7 |
+
"global_step": 17,
|
| 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.058823529411764705,
|
| 14 |
+
"grad_norm": 1.1143922805786133,
|
| 15 |
+
"learning_rate": 0.0,
|
| 16 |
+
"loss": 0.1007,
|
| 17 |
+
"step": 1
|
| 18 |
+
},
|
| 19 |
+
{
|
| 20 |
+
"epoch": 0.11764705882352941,
|
| 21 |
+
"grad_norm": 1.2392655611038208,
|
| 22 |
+
"learning_rate": 1e-05,
|
| 23 |
+
"loss": 0.1131,
|
| 24 |
+
"step": 2
|
| 25 |
+
},
|
| 26 |
+
{
|
| 27 |
+
"epoch": 0.17647058823529413,
|
| 28 |
+
"grad_norm": 1.21766197681427,
|
| 29 |
+
"learning_rate": 2e-05,
|
| 30 |
+
"loss": 0.099,
|
| 31 |
+
"step": 3
|
| 32 |
+
},
|
| 33 |
+
{
|
| 34 |
+
"epoch": 0.23529411764705882,
|
| 35 |
+
"grad_norm": 1.0714961290359497,
|
| 36 |
+
"learning_rate": 1.866666666666667e-05,
|
| 37 |
+
"loss": 0.0867,
|
| 38 |
+
"step": 4
|
| 39 |
+
},
|
| 40 |
+
{
|
| 41 |
+
"epoch": 0.29411764705882354,
|
| 42 |
+
"grad_norm": 0.8589863777160645,
|
| 43 |
+
"learning_rate": 1.7333333333333336e-05,
|
| 44 |
+
"loss": 0.0682,
|
| 45 |
+
"step": 5
|
| 46 |
+
},
|
| 47 |
+
{
|
| 48 |
+
"epoch": 0.35294117647058826,
|
| 49 |
+
"grad_norm": 0.9731529951095581,
|
| 50 |
+
"learning_rate": 1.6000000000000003e-05,
|
| 51 |
+
"loss": 0.1019,
|
| 52 |
+
"step": 6
|
| 53 |
+
},
|
| 54 |
+
{
|
| 55 |
+
"epoch": 0.4117647058823529,
|
| 56 |
+
"grad_norm": 0.7355734705924988,
|
| 57 |
+
"learning_rate": 1.4666666666666666e-05,
|
| 58 |
+
"loss": 0.0618,
|
| 59 |
+
"step": 7
|
| 60 |
+
},
|
| 61 |
+
{
|
| 62 |
+
"epoch": 0.47058823529411764,
|
| 63 |
+
"grad_norm": 0.70526123046875,
|
| 64 |
+
"learning_rate": 1.3333333333333333e-05,
|
| 65 |
+
"loss": 0.0623,
|
| 66 |
+
"step": 8
|
| 67 |
+
},
|
| 68 |
+
{
|
| 69 |
+
"epoch": 0.5294117647058824,
|
| 70 |
+
"grad_norm": 0.5791162848472595,
|
| 71 |
+
"learning_rate": 1.2e-05,
|
| 72 |
+
"loss": 0.0564,
|
| 73 |
+
"step": 9
|
| 74 |
+
},
|
| 75 |
+
{
|
| 76 |
+
"epoch": 0.5882352941176471,
|
| 77 |
+
"grad_norm": 0.6825107336044312,
|
| 78 |
+
"learning_rate": 1.0666666666666667e-05,
|
| 79 |
+
"loss": 0.0521,
|
| 80 |
+
"step": 10
|
| 81 |
+
},
|
| 82 |
+
{
|
| 83 |
+
"epoch": 0.6470588235294118,
|
| 84 |
+
"grad_norm": 0.5632421970367432,
|
| 85 |
+
"learning_rate": 9.333333333333334e-06,
|
| 86 |
+
"loss": 0.0545,
|
| 87 |
+
"step": 11
|
| 88 |
+
},
|
| 89 |
+
{
|
| 90 |
+
"epoch": 0.7058823529411765,
|
| 91 |
+
"grad_norm": 0.4283078610897064,
|
| 92 |
+
"learning_rate": 8.000000000000001e-06,
|
| 93 |
+
"loss": 0.0335,
|
| 94 |
+
"step": 12
|
| 95 |
+
},
|
| 96 |
+
{
|
| 97 |
+
"epoch": 0.7647058823529411,
|
| 98 |
+
"grad_norm": 0.5495983958244324,
|
| 99 |
+
"learning_rate": 6.666666666666667e-06,
|
| 100 |
+
"loss": 0.0593,
|
| 101 |
+
"step": 13
|
| 102 |
+
},
|
| 103 |
+
{
|
| 104 |
+
"epoch": 0.8235294117647058,
|
| 105 |
+
"grad_norm": 0.45312389731407166,
|
| 106 |
+
"learning_rate": 5.333333333333334e-06,
|
| 107 |
+
"loss": 0.0381,
|
| 108 |
+
"step": 14
|
| 109 |
+
},
|
| 110 |
+
{
|
| 111 |
+
"epoch": 0.8823529411764706,
|
| 112 |
+
"grad_norm": 0.397487610578537,
|
| 113 |
+
"learning_rate": 4.000000000000001e-06,
|
| 114 |
+
"loss": 0.0308,
|
| 115 |
+
"step": 15
|
| 116 |
+
},
|
| 117 |
+
{
|
| 118 |
+
"epoch": 0.9411764705882353,
|
| 119 |
+
"grad_norm": 0.5438785552978516,
|
| 120 |
+
"learning_rate": 2.666666666666667e-06,
|
| 121 |
+
"loss": 0.0487,
|
| 122 |
+
"step": 16
|
| 123 |
+
},
|
| 124 |
+
{
|
| 125 |
+
"epoch": 1.0,
|
| 126 |
+
"grad_norm": 0.5302333235740662,
|
| 127 |
+
"learning_rate": 1.3333333333333334e-06,
|
| 128 |
+
"loss": 0.0398,
|
| 129 |
+
"step": 17
|
| 130 |
+
}
|
| 131 |
+
],
|
| 132 |
+
"logging_steps": 1,
|
| 133 |
+
"max_steps": 17,
|
| 134 |
+
"num_input_tokens_seen": 0,
|
| 135 |
+
"num_train_epochs": 1,
|
| 136 |
+
"save_steps": 100,
|
| 137 |
+
"stateful_callbacks": {
|
| 138 |
+
"TrainerControl": {
|
| 139 |
+
"args": {
|
| 140 |
+
"should_epoch_stop": false,
|
| 141 |
+
"should_evaluate": false,
|
| 142 |
+
"should_log": false,
|
| 143 |
+
"should_save": true,
|
| 144 |
+
"should_training_stop": true
|
| 145 |
+
},
|
| 146 |
+
"attributes": {}
|
| 147 |
+
}
|
| 148 |
+
},
|
| 149 |
+
"total_flos": 0.0,
|
| 150 |
+
"train_batch_size": 16,
|
| 151 |
+
"trial_name": null,
|
| 152 |
+
"trial_params": null
|
| 153 |
+
}
|
training_args.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:03986a621c8a967582b8c75e6d85cf4cdc8e83d874acd21f1b6d57b1406266ac
|
| 3 |
+
size 5560
|
vocab.txt
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|