agentlans commited on
Commit
da2db33
·
verified ·
1 Parent(s): 279776a

Upload 12 files

Browse files
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