Text Classification
Transformers
Safetensors
bert
Generated from Trainer
regression
Eval Results (legacy)
text-embeddings-inference
Instructions to use agentlans/bge-small-en-text-quality with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use agentlans/bge-small-en-text-quality with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="agentlans/bge-small-en-text-quality")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("agentlans/bge-small-en-text-quality") model = AutoModelForSequenceClassification.from_pretrained("agentlans/bge-small-en-text-quality", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Upload 12 files
Browse files- README.md +125 -0
- all_results.json +15 -0
- config.json +32 -0
- eval_results.json +9 -0
- model.safetensors +3 -0
- special_tokens_map.json +37 -0
- tokenizer.json +0 -0
- tokenizer_config.json +58 -0
- train_results.json +9 -0
- trainer_state.json +490 -0
- training_args.bin +3 -0
- vocab.txt +0 -0
README.md
CHANGED
|
@@ -1,3 +1,128 @@
|
|
| 1 |
---
|
|
|
|
| 2 |
license: mit
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
+
library_name: transformers
|
| 3 |
license: mit
|
| 4 |
+
base_model: BAAI/bge-small-en-v1.5
|
| 5 |
+
dataset: agentlans/en-document-classification
|
| 6 |
+
tags:
|
| 7 |
+
- generated_from_trainer
|
| 8 |
+
- text-classification
|
| 9 |
+
- regression
|
| 10 |
+
model-index:
|
| 11 |
+
- name: bge-small-en-text-quality
|
| 12 |
+
results:
|
| 13 |
+
- task:
|
| 14 |
+
type: text-classification
|
| 15 |
+
name: Text Regression
|
| 16 |
+
dataset:
|
| 17 |
+
name: en-document-classification
|
| 18 |
+
type: agentlans/en-document-classification
|
| 19 |
+
metrics:
|
| 20 |
+
- name: MSE
|
| 21 |
+
type: mse
|
| 22 |
+
value: 0.1176
|
| 23 |
---
|
| 24 |
+
# bge-small-en-text-quality
|
| 25 |
+
|
| 26 |
+
This model evaluates English language text quality by mapping text inputs to a continuous quality score.
|
| 27 |
+
|
| 28 |
+
It is a fine-tuned version of [BAAI/bge-small-en-v1.5](https://huggingface.co/BAAI/bge-small-en-v1.5) optimized on the [agentlans/en-document-classification](https://huggingface.co/datasets/agentlans/en-document-classification) dataset for text quality regression.
|
| 29 |
+
|
| 30 |
+
### Evaluation Performance
|
| 31 |
+
* **Validation Loss:** 0.1176
|
| 32 |
+
* **Mean Squared Error (MSE):** 0.1176 *(achieved at Epoch 2)*
|
| 33 |
+
|
| 34 |
+
## Model Description
|
| 35 |
+
|
| 36 |
+
This model acts as a scoring mechanism for the informational value and professional quality of English text. It outputs a single scalar value: higher scores generally indicate structured, formal, and high-information content (e.g., academic, technical, or well-structured reference text), while lower scores indicate spam, unverified chatter, or low-context snippets.
|
| 37 |
+
|
| 38 |
+
### How to Use
|
| 39 |
+
|
| 40 |
+
You can use this model directly with the Hugging Face `transformers` library for sequence classification/regression:
|
| 41 |
+
|
| 42 |
+
```python
|
| 43 |
+
import torch
|
| 44 |
+
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
| 45 |
+
|
| 46 |
+
# Load model and tokenizer
|
| 47 |
+
model_name = "agentlans/bge-small-en-text-quality"
|
| 48 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 49 |
+
model = AutoModelForSequenceClassification.from_pretrained(model_name)
|
| 50 |
+
|
| 51 |
+
# Sample texts
|
| 52 |
+
texts = [
|
| 53 |
+
"Original publication and designation: Decaisne, J. (1842). Essais sur une classification des algues...",
|
| 54 |
+
"Modern Action Figure Toys & Collectibles!! You've been outbid to E****y! to YOU!"
|
| 55 |
+
]
|
| 56 |
+
|
| 57 |
+
# Tokenize and predict
|
| 58 |
+
inputs = tokenizer(texts, padding=True, truncation=True, return_tensors="pt")
|
| 59 |
+
with torch.no_grad():
|
| 60 |
+
outputs = model(**inputs)
|
| 61 |
+
# The model outputs a single continuous value per text
|
| 62 |
+
scores = outputs.logits.squeeze(-1).tolist()
|
| 63 |
+
|
| 64 |
+
for text, score in zip(texts, scores):
|
| 65 |
+
print(f"Score: {score:.4f} | Text: {text[:80]}...")
|
| 66 |
+
|
| 67 |
+
```
|
| 68 |
+
|
| 69 |
+
## Intended Uses & Limitations
|
| 70 |
+
|
| 71 |
+
### Primary Use Cases
|
| 72 |
+
|
| 73 |
+
* **Content Moderation:** A rapid, lightweight method to filter out spam, SEO filler, and corrupted web-scraped text.
|
| 74 |
+
* **Data Filtering:** Sorting or ranking documents by informational value before using them for LLM pre-training or RAG (Retrieval-Augmented Generation) pipelines.
|
| 75 |
+
|
| 76 |
+
### Known Limitations
|
| 77 |
+
|
| 78 |
+
* **Genre Bias:** The model is heavily weighted toward structured, informative prose. Subjective, opinionated, or creative writing (e.g., personal blogs, fiction, entertainment reviews) will inherently score lower regardless of their actual artistic or human value.
|
| 79 |
+
* **Lack of Explainability:** It returns a raw numerical score without indicating *why* a text was deemed high or low quality.
|
| 80 |
+
* **No Fact-Checking Capability:** The model scores structural and stylistic indicators of quality. It cannot verify the objective accuracy or truthfulness of a statement; highly structured misinformation may still receive a high rating.
|
| 81 |
+
|
| 82 |
+
## Training and Evaluation Data
|
| 83 |
+
|
| 84 |
+
The model was trained and evaluated on the `agentlans/en-document-classification` dataset. Below is a sample of text inputs alongside their predicted scores versus their true dataset targets.
|
| 85 |
+
|
| 86 |
+
### Sample Evaluation Results
|
| 87 |
+
|
| 88 |
+
| Input Sample | Predicted Value | Actual Target |
|
| 89 |
+
| --- | --- | --- |
|
| 90 |
+
| Modern Action Figure Toys & Collectibles - Vintage, Rare and Hard to Find Toys including Convention and Store Exclusives! You've been outbid to E****y! to YOU! | **-1.6846** | -1.8808 |
|
| 91 |
+
| Find 2 listings related to afterglow hair salon in Defuniak Springs on YP.com. See reviews, photos, directions, phone numbers and more for Afterglow Hair Salon... | **-1.6602** | -1.7688 |
|
| 92 |
+
| We had a little bit of down time yesterday due to some technical difficulties. In other words Martin accidentally deleted the website! But while one hand destroys... | **-1.3730** | -1.4030 |
|
| 93 |
+
| Should we talk with U.S. Congressmen who we know support campaign finance reform? Our Theory of Change only requires support from our State Legislators... | **-0.2137** | -0.6712 |
|
| 94 |
+
| Sarah Hofstetter is the global CEO at 360i, the hotshot agency that’s behind some of the most buzzed-about work in the industry. The funny thing is... | **-0.3315** | -0.3483 |
|
| 95 |
+
| LJW lawyers have extensive experience litigating commercial disputes on a broad range of issues and in a variety of forums. LJW's commercial practice includes... | **0.3911** | 0.3204 |
|
| 96 |
+
| If you own a beautiful hand-knotted Oriental rug, you want to keep it looking beautiful forever. There are certain things that you can do that will help... | **0.9932** | 1.0737 |
|
| 97 |
+
| Original publication and holotype designation: Decaisne, J. (1842). Essais sur une classification des algues et des polypiers calcifères de Lamouroux... | **1.6182** | 1.6823 |
|
| 98 |
+
| Chemical additives are used in foodstuffs and sometimes it results into adulteration due to processing and hygiene aids. The discovery of an increasing number... | **2.4180** | 2.4192 |
|
| 99 |
+
|
| 100 |
+
## Training Procedure
|
| 101 |
+
|
| 102 |
+
### Training Hyperparameters
|
| 103 |
+
|
| 104 |
+
* **Learning Rate:** 5e-05
|
| 105 |
+
* **Train Batch Size:** 8
|
| 106 |
+
* **Eval Batch Size:** 8
|
| 107 |
+
* **Seed:** 42
|
| 108 |
+
* **Optimizer:** AdamW (`adamw_torch_fused`) with $\beta_1 = 0.9$, $\beta_2 = 0.999$, and $\epsilon = 10^{-8}$
|
| 109 |
+
* **Lr Scheduler Type:** Linear
|
| 110 |
+
* **Num Epochs:** 3.0
|
| 111 |
+
|
| 112 |
+
### Epoch-by-Epoch Progress
|
| 113 |
+
|
| 114 |
+
| Training Loss | Epoch | Step | Validation Loss | Mse |
|
| 115 |
+
| --- | --- | --- | --- | --- |
|
| 116 |
+
| 0.1426 | 1.0 | 10000 | 0.1778 | 0.1778 |
|
| 117 |
+
| 0.0895 | 2.0 | 20000 | **0.1176** | **0.1176** |
|
| 118 |
+
| 0.0521 | 3.0 | 30000 | 0.1261 | 0.1261 |
|
| 119 |
+
|
| 120 |
+
> **Note:** The model begins over-fitting slightly by Epoch 3; the checkpoint at Epoch 2 provides the lowest Validation Loss / MSE.
|
| 121 |
+
|
| 122 |
+
### Framework Versions
|
| 123 |
+
|
| 124 |
+
* Transformers 5.0.0.dev0
|
| 125 |
+
* Pytorch 2.9.1+cu128
|
| 126 |
+
* Datasets 4.4.1
|
| 127 |
+
* Tokenizers 0.22.1
|
| 128 |
+
|
all_results.json
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"epoch": 3.0,
|
| 3 |
+
"eval_loss": 0.1175876185297966,
|
| 4 |
+
"eval_mse": 0.11758762182878052,
|
| 5 |
+
"eval_runtime": 11.4071,
|
| 6 |
+
"eval_samples": 20000,
|
| 7 |
+
"eval_samples_per_second": 1753.295,
|
| 8 |
+
"eval_steps_per_second": 219.162,
|
| 9 |
+
"total_flos": 3952243439859456.0,
|
| 10 |
+
"train_loss": 0.10308435103098551,
|
| 11 |
+
"train_runtime": 528.9789,
|
| 12 |
+
"train_samples": 79999,
|
| 13 |
+
"train_samples_per_second": 453.699,
|
| 14 |
+
"train_steps_per_second": 56.713
|
| 15 |
+
}
|
config.json
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"architectures": [
|
| 3 |
+
"BertForSequenceClassification"
|
| 4 |
+
],
|
| 5 |
+
"attention_probs_dropout_prob": 0.1,
|
| 6 |
+
"classifier_dropout": null,
|
| 7 |
+
"dtype": "float32",
|
| 8 |
+
"finetuning_task": "text-classification",
|
| 9 |
+
"hidden_act": "gelu",
|
| 10 |
+
"hidden_dropout_prob": 0.1,
|
| 11 |
+
"hidden_size": 384,
|
| 12 |
+
"id2label": {
|
| 13 |
+
"0": "LABEL_0"
|
| 14 |
+
},
|
| 15 |
+
"initializer_range": 0.02,
|
| 16 |
+
"intermediate_size": 1536,
|
| 17 |
+
"label2id": {
|
| 18 |
+
"LABEL_0": 0
|
| 19 |
+
},
|
| 20 |
+
"layer_norm_eps": 1e-12,
|
| 21 |
+
"max_position_embeddings": 512,
|
| 22 |
+
"model_type": "bert",
|
| 23 |
+
"num_attention_heads": 12,
|
| 24 |
+
"num_hidden_layers": 12,
|
| 25 |
+
"pad_token_id": 0,
|
| 26 |
+
"position_embedding_type": "absolute",
|
| 27 |
+
"problem_type": "regression",
|
| 28 |
+
"transformers_version": "5.0.0.dev0",
|
| 29 |
+
"type_vocab_size": 2,
|
| 30 |
+
"use_cache": false,
|
| 31 |
+
"vocab_size": 30522
|
| 32 |
+
}
|
eval_results.json
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"epoch": 3.0,
|
| 3 |
+
"eval_loss": 0.1175876185297966,
|
| 4 |
+
"eval_mse": 0.11758762182878052,
|
| 5 |
+
"eval_runtime": 11.4071,
|
| 6 |
+
"eval_samples": 20000,
|
| 7 |
+
"eval_samples_per_second": 1753.295,
|
| 8 |
+
"eval_steps_per_second": 219.162
|
| 9 |
+
}
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4dc475a5c1e27cbd481ca7507df9c52555a8adedc9f72f3e91783410eb4831b0
|
| 3 |
+
size 133464836
|
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,58 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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": true,
|
| 45 |
+
"cls_token": "[CLS]",
|
| 46 |
+
"do_basic_tokenize": true,
|
| 47 |
+
"do_lower_case": true,
|
| 48 |
+
"extra_special_tokens": {},
|
| 49 |
+
"mask_token": "[MASK]",
|
| 50 |
+
"model_max_length": 512,
|
| 51 |
+
"never_split": null,
|
| 52 |
+
"pad_token": "[PAD]",
|
| 53 |
+
"sep_token": "[SEP]",
|
| 54 |
+
"strip_accents": null,
|
| 55 |
+
"tokenize_chinese_chars": true,
|
| 56 |
+
"tokenizer_class": "BertTokenizer",
|
| 57 |
+
"unk_token": "[UNK]"
|
| 58 |
+
}
|
train_results.json
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"epoch": 3.0,
|
| 3 |
+
"total_flos": 3952243439859456.0,
|
| 4 |
+
"train_loss": 0.10308435103098551,
|
| 5 |
+
"train_runtime": 528.9789,
|
| 6 |
+
"train_samples": 79999,
|
| 7 |
+
"train_samples_per_second": 453.699,
|
| 8 |
+
"train_steps_per_second": 56.713
|
| 9 |
+
}
|
trainer_state.json
ADDED
|
@@ -0,0 +1,490 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"best_global_step": 20000,
|
| 3 |
+
"best_metric": 0.1175876185297966,
|
| 4 |
+
"best_model_checkpoint": "./bge-small-en-text-quality/checkpoint-20000",
|
| 5 |
+
"epoch": 3.0,
|
| 6 |
+
"eval_steps": 500,
|
| 7 |
+
"global_step": 30000,
|
| 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.05,
|
| 14 |
+
"grad_norm": 3.9433600902557373,
|
| 15 |
+
"learning_rate": 4.9168333333333335e-05,
|
| 16 |
+
"loss": 0.2981,
|
| 17 |
+
"step": 500
|
| 18 |
+
},
|
| 19 |
+
{
|
| 20 |
+
"epoch": 0.1,
|
| 21 |
+
"grad_norm": 3.3599181175231934,
|
| 22 |
+
"learning_rate": 4.8335000000000004e-05,
|
| 23 |
+
"loss": 0.2038,
|
| 24 |
+
"step": 1000
|
| 25 |
+
},
|
| 26 |
+
{
|
| 27 |
+
"epoch": 0.15,
|
| 28 |
+
"grad_norm": 10.630352973937988,
|
| 29 |
+
"learning_rate": 4.7501666666666666e-05,
|
| 30 |
+
"loss": 0.1913,
|
| 31 |
+
"step": 1500
|
| 32 |
+
},
|
| 33 |
+
{
|
| 34 |
+
"epoch": 0.2,
|
| 35 |
+
"grad_norm": 2.7322261333465576,
|
| 36 |
+
"learning_rate": 4.6668333333333335e-05,
|
| 37 |
+
"loss": 0.1819,
|
| 38 |
+
"step": 2000
|
| 39 |
+
},
|
| 40 |
+
{
|
| 41 |
+
"epoch": 0.25,
|
| 42 |
+
"grad_norm": 3.438952922821045,
|
| 43 |
+
"learning_rate": 4.5835e-05,
|
| 44 |
+
"loss": 0.1678,
|
| 45 |
+
"step": 2500
|
| 46 |
+
},
|
| 47 |
+
{
|
| 48 |
+
"epoch": 0.3,
|
| 49 |
+
"grad_norm": 6.013777732849121,
|
| 50 |
+
"learning_rate": 4.5001666666666666e-05,
|
| 51 |
+
"loss": 0.1635,
|
| 52 |
+
"step": 3000
|
| 53 |
+
},
|
| 54 |
+
{
|
| 55 |
+
"epoch": 0.35,
|
| 56 |
+
"grad_norm": 3.261183977127075,
|
| 57 |
+
"learning_rate": 4.4168333333333335e-05,
|
| 58 |
+
"loss": 0.1564,
|
| 59 |
+
"step": 3500
|
| 60 |
+
},
|
| 61 |
+
{
|
| 62 |
+
"epoch": 0.4,
|
| 63 |
+
"grad_norm": 4.93856954574585,
|
| 64 |
+
"learning_rate": 4.3335000000000004e-05,
|
| 65 |
+
"loss": 0.1555,
|
| 66 |
+
"step": 4000
|
| 67 |
+
},
|
| 68 |
+
{
|
| 69 |
+
"epoch": 0.45,
|
| 70 |
+
"grad_norm": 1.6399900913238525,
|
| 71 |
+
"learning_rate": 4.2501666666666667e-05,
|
| 72 |
+
"loss": 0.1527,
|
| 73 |
+
"step": 4500
|
| 74 |
+
},
|
| 75 |
+
{
|
| 76 |
+
"epoch": 0.5,
|
| 77 |
+
"grad_norm": 4.572032451629639,
|
| 78 |
+
"learning_rate": 4.1668333333333336e-05,
|
| 79 |
+
"loss": 0.1472,
|
| 80 |
+
"step": 5000
|
| 81 |
+
},
|
| 82 |
+
{
|
| 83 |
+
"epoch": 0.55,
|
| 84 |
+
"grad_norm": 2.234090566635132,
|
| 85 |
+
"learning_rate": 4.0835e-05,
|
| 86 |
+
"loss": 0.1551,
|
| 87 |
+
"step": 5500
|
| 88 |
+
},
|
| 89 |
+
{
|
| 90 |
+
"epoch": 0.6,
|
| 91 |
+
"grad_norm": 2.3441779613494873,
|
| 92 |
+
"learning_rate": 4.0001666666666674e-05,
|
| 93 |
+
"loss": 0.1435,
|
| 94 |
+
"step": 6000
|
| 95 |
+
},
|
| 96 |
+
{
|
| 97 |
+
"epoch": 0.65,
|
| 98 |
+
"grad_norm": 3.1698434352874756,
|
| 99 |
+
"learning_rate": 3.9168333333333336e-05,
|
| 100 |
+
"loss": 0.1484,
|
| 101 |
+
"step": 6500
|
| 102 |
+
},
|
| 103 |
+
{
|
| 104 |
+
"epoch": 0.7,
|
| 105 |
+
"grad_norm": 4.67405891418457,
|
| 106 |
+
"learning_rate": 3.8335000000000005e-05,
|
| 107 |
+
"loss": 0.1489,
|
| 108 |
+
"step": 7000
|
| 109 |
+
},
|
| 110 |
+
{
|
| 111 |
+
"epoch": 0.75,
|
| 112 |
+
"grad_norm": 4.494452953338623,
|
| 113 |
+
"learning_rate": 3.750166666666667e-05,
|
| 114 |
+
"loss": 0.1376,
|
| 115 |
+
"step": 7500
|
| 116 |
+
},
|
| 117 |
+
{
|
| 118 |
+
"epoch": 0.8,
|
| 119 |
+
"grad_norm": 8.088772773742676,
|
| 120 |
+
"learning_rate": 3.666833333333333e-05,
|
| 121 |
+
"loss": 0.1477,
|
| 122 |
+
"step": 8000
|
| 123 |
+
},
|
| 124 |
+
{
|
| 125 |
+
"epoch": 0.85,
|
| 126 |
+
"grad_norm": 2.655813217163086,
|
| 127 |
+
"learning_rate": 3.5835000000000005e-05,
|
| 128 |
+
"loss": 0.1414,
|
| 129 |
+
"step": 8500
|
| 130 |
+
},
|
| 131 |
+
{
|
| 132 |
+
"epoch": 0.9,
|
| 133 |
+
"grad_norm": 3.0818183422088623,
|
| 134 |
+
"learning_rate": 3.500166666666667e-05,
|
| 135 |
+
"loss": 0.1355,
|
| 136 |
+
"step": 9000
|
| 137 |
+
},
|
| 138 |
+
{
|
| 139 |
+
"epoch": 0.95,
|
| 140 |
+
"grad_norm": 1.5601372718811035,
|
| 141 |
+
"learning_rate": 3.4168333333333336e-05,
|
| 142 |
+
"loss": 0.1364,
|
| 143 |
+
"step": 9500
|
| 144 |
+
},
|
| 145 |
+
{
|
| 146 |
+
"epoch": 1.0,
|
| 147 |
+
"grad_norm": 3.5111470222473145,
|
| 148 |
+
"learning_rate": 3.3335e-05,
|
| 149 |
+
"loss": 0.1426,
|
| 150 |
+
"step": 10000
|
| 151 |
+
},
|
| 152 |
+
{
|
| 153 |
+
"epoch": 1.0,
|
| 154 |
+
"eval_loss": 0.1778368353843689,
|
| 155 |
+
"eval_mse": 0.1778368374988932,
|
| 156 |
+
"eval_runtime": 11.2433,
|
| 157 |
+
"eval_samples_per_second": 1778.834,
|
| 158 |
+
"eval_steps_per_second": 222.354,
|
| 159 |
+
"step": 10000
|
| 160 |
+
},
|
| 161 |
+
{
|
| 162 |
+
"epoch": 1.05,
|
| 163 |
+
"grad_norm": 2.7848703861236572,
|
| 164 |
+
"learning_rate": 3.250166666666667e-05,
|
| 165 |
+
"loss": 0.0963,
|
| 166 |
+
"step": 10500
|
| 167 |
+
},
|
| 168 |
+
{
|
| 169 |
+
"epoch": 1.1,
|
| 170 |
+
"grad_norm": 1.4169955253601074,
|
| 171 |
+
"learning_rate": 3.1668333333333336e-05,
|
| 172 |
+
"loss": 0.0937,
|
| 173 |
+
"step": 11000
|
| 174 |
+
},
|
| 175 |
+
{
|
| 176 |
+
"epoch": 1.15,
|
| 177 |
+
"grad_norm": 1.4891533851623535,
|
| 178 |
+
"learning_rate": 3.0835000000000005e-05,
|
| 179 |
+
"loss": 0.0955,
|
| 180 |
+
"step": 11500
|
| 181 |
+
},
|
| 182 |
+
{
|
| 183 |
+
"epoch": 1.2,
|
| 184 |
+
"grad_norm": 3.7263450622558594,
|
| 185 |
+
"learning_rate": 3.0001666666666668e-05,
|
| 186 |
+
"loss": 0.0925,
|
| 187 |
+
"step": 12000
|
| 188 |
+
},
|
| 189 |
+
{
|
| 190 |
+
"epoch": 1.25,
|
| 191 |
+
"grad_norm": 2.7233424186706543,
|
| 192 |
+
"learning_rate": 2.9168333333333337e-05,
|
| 193 |
+
"loss": 0.0874,
|
| 194 |
+
"step": 12500
|
| 195 |
+
},
|
| 196 |
+
{
|
| 197 |
+
"epoch": 1.3,
|
| 198 |
+
"grad_norm": 1.419253945350647,
|
| 199 |
+
"learning_rate": 2.8335e-05,
|
| 200 |
+
"loss": 0.0921,
|
| 201 |
+
"step": 13000
|
| 202 |
+
},
|
| 203 |
+
{
|
| 204 |
+
"epoch": 1.35,
|
| 205 |
+
"grad_norm": 3.7091524600982666,
|
| 206 |
+
"learning_rate": 2.750166666666667e-05,
|
| 207 |
+
"loss": 0.0922,
|
| 208 |
+
"step": 13500
|
| 209 |
+
},
|
| 210 |
+
{
|
| 211 |
+
"epoch": 1.4,
|
| 212 |
+
"grad_norm": 1.8904818296432495,
|
| 213 |
+
"learning_rate": 2.6668333333333334e-05,
|
| 214 |
+
"loss": 0.0893,
|
| 215 |
+
"step": 14000
|
| 216 |
+
},
|
| 217 |
+
{
|
| 218 |
+
"epoch": 1.45,
|
| 219 |
+
"grad_norm": 2.954240322113037,
|
| 220 |
+
"learning_rate": 2.5835000000000003e-05,
|
| 221 |
+
"loss": 0.091,
|
| 222 |
+
"step": 14500
|
| 223 |
+
},
|
| 224 |
+
{
|
| 225 |
+
"epoch": 1.5,
|
| 226 |
+
"grad_norm": 1.840616226196289,
|
| 227 |
+
"learning_rate": 2.5001666666666668e-05,
|
| 228 |
+
"loss": 0.0907,
|
| 229 |
+
"step": 15000
|
| 230 |
+
},
|
| 231 |
+
{
|
| 232 |
+
"epoch": 1.55,
|
| 233 |
+
"grad_norm": 1.0130300521850586,
|
| 234 |
+
"learning_rate": 2.4168333333333334e-05,
|
| 235 |
+
"loss": 0.0941,
|
| 236 |
+
"step": 15500
|
| 237 |
+
},
|
| 238 |
+
{
|
| 239 |
+
"epoch": 1.6,
|
| 240 |
+
"grad_norm": 0.8689394593238831,
|
| 241 |
+
"learning_rate": 2.3335000000000003e-05,
|
| 242 |
+
"loss": 0.0912,
|
| 243 |
+
"step": 16000
|
| 244 |
+
},
|
| 245 |
+
{
|
| 246 |
+
"epoch": 1.65,
|
| 247 |
+
"grad_norm": 1.185209035873413,
|
| 248 |
+
"learning_rate": 2.250166666666667e-05,
|
| 249 |
+
"loss": 0.0913,
|
| 250 |
+
"step": 16500
|
| 251 |
+
},
|
| 252 |
+
{
|
| 253 |
+
"epoch": 1.7,
|
| 254 |
+
"grad_norm": 1.7481811046600342,
|
| 255 |
+
"learning_rate": 2.1668333333333334e-05,
|
| 256 |
+
"loss": 0.0876,
|
| 257 |
+
"step": 17000
|
| 258 |
+
},
|
| 259 |
+
{
|
| 260 |
+
"epoch": 1.75,
|
| 261 |
+
"grad_norm": 1.983954906463623,
|
| 262 |
+
"learning_rate": 2.0835000000000003e-05,
|
| 263 |
+
"loss": 0.0907,
|
| 264 |
+
"step": 17500
|
| 265 |
+
},
|
| 266 |
+
{
|
| 267 |
+
"epoch": 1.8,
|
| 268 |
+
"grad_norm": 1.7682359218597412,
|
| 269 |
+
"learning_rate": 2.000166666666667e-05,
|
| 270 |
+
"loss": 0.0917,
|
| 271 |
+
"step": 18000
|
| 272 |
+
},
|
| 273 |
+
{
|
| 274 |
+
"epoch": 1.85,
|
| 275 |
+
"grad_norm": 1.4291614294052124,
|
| 276 |
+
"learning_rate": 1.9168333333333334e-05,
|
| 277 |
+
"loss": 0.0889,
|
| 278 |
+
"step": 18500
|
| 279 |
+
},
|
| 280 |
+
{
|
| 281 |
+
"epoch": 1.9,
|
| 282 |
+
"grad_norm": 2.0772218704223633,
|
| 283 |
+
"learning_rate": 1.8335000000000003e-05,
|
| 284 |
+
"loss": 0.0931,
|
| 285 |
+
"step": 19000
|
| 286 |
+
},
|
| 287 |
+
{
|
| 288 |
+
"epoch": 1.95,
|
| 289 |
+
"grad_norm": 4.12407922744751,
|
| 290 |
+
"learning_rate": 1.7501666666666665e-05,
|
| 291 |
+
"loss": 0.0922,
|
| 292 |
+
"step": 19500
|
| 293 |
+
},
|
| 294 |
+
{
|
| 295 |
+
"epoch": 2.0,
|
| 296 |
+
"grad_norm": 1.343153476715088,
|
| 297 |
+
"learning_rate": 1.6668333333333334e-05,
|
| 298 |
+
"loss": 0.0895,
|
| 299 |
+
"step": 20000
|
| 300 |
+
},
|
| 301 |
+
{
|
| 302 |
+
"epoch": 2.0,
|
| 303 |
+
"eval_loss": 0.1175876185297966,
|
| 304 |
+
"eval_mse": 0.11758762182878052,
|
| 305 |
+
"eval_runtime": 11.2725,
|
| 306 |
+
"eval_samples_per_second": 1774.228,
|
| 307 |
+
"eval_steps_per_second": 221.779,
|
| 308 |
+
"step": 20000
|
| 309 |
+
},
|
| 310 |
+
{
|
| 311 |
+
"epoch": 2.05,
|
| 312 |
+
"grad_norm": 1.369104027748108,
|
| 313 |
+
"learning_rate": 1.5835e-05,
|
| 314 |
+
"loss": 0.0583,
|
| 315 |
+
"step": 20500
|
| 316 |
+
},
|
| 317 |
+
{
|
| 318 |
+
"epoch": 2.1,
|
| 319 |
+
"grad_norm": 4.734041213989258,
|
| 320 |
+
"learning_rate": 1.5001666666666667e-05,
|
| 321 |
+
"loss": 0.0556,
|
| 322 |
+
"step": 21000
|
| 323 |
+
},
|
| 324 |
+
{
|
| 325 |
+
"epoch": 2.15,
|
| 326 |
+
"grad_norm": 2.007387638092041,
|
| 327 |
+
"learning_rate": 1.4168333333333333e-05,
|
| 328 |
+
"loss": 0.0576,
|
| 329 |
+
"step": 21500
|
| 330 |
+
},
|
| 331 |
+
{
|
| 332 |
+
"epoch": 2.2,
|
| 333 |
+
"grad_norm": 2.5438034534454346,
|
| 334 |
+
"learning_rate": 1.3335e-05,
|
| 335 |
+
"loss": 0.0564,
|
| 336 |
+
"step": 22000
|
| 337 |
+
},
|
| 338 |
+
{
|
| 339 |
+
"epoch": 2.25,
|
| 340 |
+
"grad_norm": 4.6716132164001465,
|
| 341 |
+
"learning_rate": 1.2501666666666668e-05,
|
| 342 |
+
"loss": 0.0566,
|
| 343 |
+
"step": 22500
|
| 344 |
+
},
|
| 345 |
+
{
|
| 346 |
+
"epoch": 2.3,
|
| 347 |
+
"grad_norm": 3.9979987144470215,
|
| 348 |
+
"learning_rate": 1.1668333333333333e-05,
|
| 349 |
+
"loss": 0.0591,
|
| 350 |
+
"step": 23000
|
| 351 |
+
},
|
| 352 |
+
{
|
| 353 |
+
"epoch": 2.35,
|
| 354 |
+
"grad_norm": 2.851638078689575,
|
| 355 |
+
"learning_rate": 1.0835e-05,
|
| 356 |
+
"loss": 0.0582,
|
| 357 |
+
"step": 23500
|
| 358 |
+
},
|
| 359 |
+
{
|
| 360 |
+
"epoch": 2.4,
|
| 361 |
+
"grad_norm": 2.4053075313568115,
|
| 362 |
+
"learning_rate": 1.0001666666666668e-05,
|
| 363 |
+
"loss": 0.0577,
|
| 364 |
+
"step": 24000
|
| 365 |
+
},
|
| 366 |
+
{
|
| 367 |
+
"epoch": 2.45,
|
| 368 |
+
"grad_norm": 2.3901314735412598,
|
| 369 |
+
"learning_rate": 9.168333333333335e-06,
|
| 370 |
+
"loss": 0.0543,
|
| 371 |
+
"step": 24500
|
| 372 |
+
},
|
| 373 |
+
{
|
| 374 |
+
"epoch": 2.5,
|
| 375 |
+
"grad_norm": 1.9612762928009033,
|
| 376 |
+
"learning_rate": 8.334999999999999e-06,
|
| 377 |
+
"loss": 0.0522,
|
| 378 |
+
"step": 25000
|
| 379 |
+
},
|
| 380 |
+
{
|
| 381 |
+
"epoch": 2.55,
|
| 382 |
+
"grad_norm": 1.849785566329956,
|
| 383 |
+
"learning_rate": 7.501666666666666e-06,
|
| 384 |
+
"loss": 0.0518,
|
| 385 |
+
"step": 25500
|
| 386 |
+
},
|
| 387 |
+
{
|
| 388 |
+
"epoch": 2.6,
|
| 389 |
+
"grad_norm": 2.1459758281707764,
|
| 390 |
+
"learning_rate": 6.668333333333334e-06,
|
| 391 |
+
"loss": 0.052,
|
| 392 |
+
"step": 26000
|
| 393 |
+
},
|
| 394 |
+
{
|
| 395 |
+
"epoch": 2.65,
|
| 396 |
+
"grad_norm": 1.2251453399658203,
|
| 397 |
+
"learning_rate": 5.835e-06,
|
| 398 |
+
"loss": 0.0546,
|
| 399 |
+
"step": 26500
|
| 400 |
+
},
|
| 401 |
+
{
|
| 402 |
+
"epoch": 2.7,
|
| 403 |
+
"grad_norm": 2.3431410789489746,
|
| 404 |
+
"learning_rate": 5.001666666666667e-06,
|
| 405 |
+
"loss": 0.0571,
|
| 406 |
+
"step": 27000
|
| 407 |
+
},
|
| 408 |
+
{
|
| 409 |
+
"epoch": 2.75,
|
| 410 |
+
"grad_norm": 2.283482074737549,
|
| 411 |
+
"learning_rate": 4.168333333333334e-06,
|
| 412 |
+
"loss": 0.053,
|
| 413 |
+
"step": 27500
|
| 414 |
+
},
|
| 415 |
+
{
|
| 416 |
+
"epoch": 2.8,
|
| 417 |
+
"grad_norm": 1.6145457029342651,
|
| 418 |
+
"learning_rate": 3.335e-06,
|
| 419 |
+
"loss": 0.0538,
|
| 420 |
+
"step": 28000
|
| 421 |
+
},
|
| 422 |
+
{
|
| 423 |
+
"epoch": 2.85,
|
| 424 |
+
"grad_norm": 1.8216482400894165,
|
| 425 |
+
"learning_rate": 2.5016666666666667e-06,
|
| 426 |
+
"loss": 0.0535,
|
| 427 |
+
"step": 28500
|
| 428 |
+
},
|
| 429 |
+
{
|
| 430 |
+
"epoch": 2.9,
|
| 431 |
+
"grad_norm": 1.7675800323486328,
|
| 432 |
+
"learning_rate": 1.6683333333333336e-06,
|
| 433 |
+
"loss": 0.0533,
|
| 434 |
+
"step": 29000
|
| 435 |
+
},
|
| 436 |
+
{
|
| 437 |
+
"epoch": 2.95,
|
| 438 |
+
"grad_norm": 1.378221869468689,
|
| 439 |
+
"learning_rate": 8.35e-07,
|
| 440 |
+
"loss": 0.0513,
|
| 441 |
+
"step": 29500
|
| 442 |
+
},
|
| 443 |
+
{
|
| 444 |
+
"epoch": 3.0,
|
| 445 |
+
"grad_norm": 1.2478489875793457,
|
| 446 |
+
"learning_rate": 1.666666666666667e-09,
|
| 447 |
+
"loss": 0.0521,
|
| 448 |
+
"step": 30000
|
| 449 |
+
},
|
| 450 |
+
{
|
| 451 |
+
"epoch": 3.0,
|
| 452 |
+
"eval_loss": 0.12606243789196014,
|
| 453 |
+
"eval_mse": 0.12606243263783948,
|
| 454 |
+
"eval_runtime": 11.2592,
|
| 455 |
+
"eval_samples_per_second": 1776.328,
|
| 456 |
+
"eval_steps_per_second": 222.041,
|
| 457 |
+
"step": 30000
|
| 458 |
+
},
|
| 459 |
+
{
|
| 460 |
+
"epoch": 3.0,
|
| 461 |
+
"step": 30000,
|
| 462 |
+
"total_flos": 3952243439859456.0,
|
| 463 |
+
"train_loss": 0.10308435103098551,
|
| 464 |
+
"train_runtime": 528.9789,
|
| 465 |
+
"train_samples_per_second": 453.699,
|
| 466 |
+
"train_steps_per_second": 56.713
|
| 467 |
+
}
|
| 468 |
+
],
|
| 469 |
+
"logging_steps": 500,
|
| 470 |
+
"max_steps": 30000,
|
| 471 |
+
"num_input_tokens_seen": 0,
|
| 472 |
+
"num_train_epochs": 3,
|
| 473 |
+
"save_steps": 500,
|
| 474 |
+
"stateful_callbacks": {
|
| 475 |
+
"TrainerControl": {
|
| 476 |
+
"args": {
|
| 477 |
+
"should_epoch_stop": false,
|
| 478 |
+
"should_evaluate": false,
|
| 479 |
+
"should_log": false,
|
| 480 |
+
"should_save": true,
|
| 481 |
+
"should_training_stop": true
|
| 482 |
+
},
|
| 483 |
+
"attributes": {}
|
| 484 |
+
}
|
| 485 |
+
},
|
| 486 |
+
"total_flos": 3952243439859456.0,
|
| 487 |
+
"train_batch_size": 8,
|
| 488 |
+
"trial_name": null,
|
| 489 |
+
"trial_params": null
|
| 490 |
+
}
|
training_args.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3d8dd202ded227543427b2893c75589574e6a44546b0deeef2de35e337715c16
|
| 3 |
+
size 5201
|
vocab.txt
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|