agentlans commited on
Commit
d667e5f
·
verified ·
1 Parent(s): 5887b38

Upload 14 files

Browse files
README.md CHANGED
@@ -1,3 +1,100 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ datasets:
5
+ - Universal-NER/Pile-NER-type
6
+ - Universal-NER/Pile-NER-definition
7
+ language:
8
+ - en
9
+ base_model:
10
+ - google/flan-t5-small
11
+ pipeline_tag: text2text-generation
12
+ tags:
13
+ - named-entity-recognition
14
+ - generated_from_trainer
15
+ ---
16
+ # flan-t5-small-ner
17
+
18
+ This model is a fine-tuned version of [google/flan-t5-small](https://huggingface.co/google/flan-t5-small)
19
+ on 200 000 random (text, entity) combinations from the
20
+ [Universal-NER/Pile-NER-type](https://huggingface.co/datasets/Universal-NER/Pile-NER-type) and
21
+ [Universal-NER/Pile-NER-definition](https://huggingface.co/datasets/Universal-NER/Pile-NER-definition) datasets.
22
+
23
+ - Loss: 0.5393
24
+ - Num Input Tokens Seen: 332318598
25
+
26
+ ## Model Description
27
+
28
+ flan-t5-small-ner can extract entities of specific types or definitions from text such as person, company, school, technology, and many more.
29
+ It builds upon the FLAN-T5 architecture, which has strong performance across natural language processing tasks.
30
+
31
+ Example:
32
+
33
+ ```python
34
+ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
35
+ import torch
36
+
37
+ model_path = "agentlans/flan-t5-small-ner"
38
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_path).to("cuda" if torch.cuda.is_available() else "cpu")
39
+ tokenizer = AutoTokenizer.from_pretrained(model_path)
40
+
41
+ def custom_split(s): # Processes the output from the model
42
+ parts = s.split("<|sep|>")
43
+ if not s.endswith("<|end|>"):
44
+ parts = parts[:-1] # If output is truncated, then don't include last item
45
+ else:
46
+ parts[-1] = parts[-1].replace("<|end|>", "") # Remove the marker tokens
47
+ return [p.strip() for p in parts if p.strip()]
48
+
49
+ def find_entities(input_text, entity_type):
50
+ txt = entity_type + "<|sep|>" + input_text + "<|end|>" # Important: need exact input format
51
+ inputs = tokenizer(txt, return_tensors="pt").to(model.device)
52
+ outputs = model.generate(**inputs, max_new_tokens=100)
53
+ decoded = tokenizer.decode(outputs[0], skip_special_tokens=True)
54
+ return custom_split(decoded)
55
+
56
+ # Example usage
57
+ input_text = "In the bustling metropolis of New York City, Apple Inc. sponsored a conference where Dr. Elena Rodriguez presented groundbreaking research about neuroscience and AI."
58
+ print(find_entities(input_text, "person")) # ['Elena Rodriguez']
59
+ print(find_entities(input_text, "company")) # ['Apple Inc.']
60
+ print(find_entities(input_text, "fruit")) # []
61
+ print(find_entities(input_text, "subject")) # ['neuroscience', 'AI']
62
+ ```
63
+
64
+ ## Limitations
65
+
66
+ - False positives and negatives are possible.
67
+ - May struggle with specialized knowledge or fine distinctions.
68
+ - Performance may vary for very short or long texts.
69
+ - English language only.
70
+ - Consider privacy when processing sensitive text.
71
+
72
+ ## Training Procedure
73
+
74
+ ### Training hyperparameters
75
+
76
+ The following hyperparameters were used during training:
77
+ - learning_rate: 5e-05
78
+ - train_batch_size: 8
79
+ - eval_batch_size: 8
80
+ - seed: 42
81
+ - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
82
+ - lr_scheduler_type: linear
83
+ - num_epochs: 5.0
84
+
85
+ ### Training results
86
+
87
+ | Training Loss | Epoch | Step | Validation Loss | Input Tokens Seen |
88
+ |:-------------:|:-----:|:-----:|:---------------:|:-----------------:|
89
+ | 0.8398 | 1.0 | 19991 | 0.6227 | 66451084 |
90
+ | 0.7203 | 2.0 | 39982 | 0.5679 | 132976438 |
91
+ | 0.6479 | 3.0 | 59973 | 0.5605 | 199402582 |
92
+ | 0.6023 | 4.0 | 79964 | 0.5427 | 265875340 |
93
+ | 0.5879 | 5.0 | 99955 | 0.5393 | 332318598 |
94
+
95
+ ## Framework Versions
96
+
97
+ - Transformers: 4.46.3
98
+ - PyTorch: 2.5.1+cu124
99
+ - Datasets: 3.2.0
100
+ - Tokenizers: 0.20.3
added_tokens.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "<|end|>": 32101,
3
+ "<|sep|>": 32100
4
+ }
all_results.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 5.0,
3
+ "eval_loss": 0.5393198132514954,
4
+ "eval_runtime": 96.9251,
5
+ "eval_samples": 39981,
6
+ "eval_samples_per_second": 412.494,
7
+ "eval_steps_per_second": 51.566,
8
+ "num_input_tokens_seen": 332318598,
9
+ "total_flos": 1.2062750373789696e+17,
10
+ "train_loss": 0.7601320325591829,
11
+ "train_runtime": 7988.2275,
12
+ "train_samples": 159922,
13
+ "train_samples_per_second": 100.099,
14
+ "train_steps_per_second": 12.513,
15
+ "train_tokens_per_second": 41593.134
16
+ }
config.json ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "flan-t5-small-with-list-tokens",
3
+ "architectures": [
4
+ "T5ForConditionalGeneration"
5
+ ],
6
+ "classifier_dropout": 0.0,
7
+ "d_ff": 1024,
8
+ "d_kv": 64,
9
+ "d_model": 512,
10
+ "decoder_start_token_id": 0,
11
+ "dense_act_fn": "gelu_new",
12
+ "dropout_rate": 0.1,
13
+ "eos_token_id": 1,
14
+ "feed_forward_proj": "gated-gelu",
15
+ "initializer_factor": 1.0,
16
+ "is_encoder_decoder": true,
17
+ "is_gated_act": true,
18
+ "layer_norm_epsilon": 1e-06,
19
+ "model_type": "t5",
20
+ "n_positions": 512,
21
+ "num_decoder_layers": 8,
22
+ "num_heads": 6,
23
+ "num_layers": 8,
24
+ "output_past": true,
25
+ "pad_token_id": 0,
26
+ "relative_attention_max_distance": 128,
27
+ "relative_attention_num_buckets": 32,
28
+ "task_specific_params": {
29
+ "summarization": {
30
+ "early_stopping": true,
31
+ "length_penalty": 2.0,
32
+ "max_length": 200,
33
+ "min_length": 30,
34
+ "no_repeat_ngram_size": 3,
35
+ "num_beams": 4,
36
+ "prefix": "summarize: "
37
+ },
38
+ "translation_en_to_de": {
39
+ "early_stopping": true,
40
+ "max_length": 300,
41
+ "num_beams": 4,
42
+ "prefix": "translate English to German: "
43
+ },
44
+ "translation_en_to_fr": {
45
+ "early_stopping": true,
46
+ "max_length": 300,
47
+ "num_beams": 4,
48
+ "prefix": "translate English to French: "
49
+ },
50
+ "translation_en_to_ro": {
51
+ "early_stopping": true,
52
+ "max_length": 300,
53
+ "num_beams": 4,
54
+ "prefix": "translate English to Romanian: "
55
+ }
56
+ },
57
+ "tie_word_embeddings": false,
58
+ "torch_dtype": "float32",
59
+ "transformers_version": "4.46.3",
60
+ "use_cache": true,
61
+ "vocab_size": 32102
62
+ }
eval_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 5.0,
3
+ "eval_loss": 0.5393198132514954,
4
+ "eval_runtime": 96.9251,
5
+ "eval_samples": 39981,
6
+ "eval_samples_per_second": 412.494,
7
+ "eval_steps_per_second": 51.566,
8
+ "num_input_tokens_seen": 332318598
9
+ }
generation_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "decoder_start_token_id": 0,
4
+ "eos_token_id": 1,
5
+ "pad_token_id": 0,
6
+ "transformers_version": "4.46.3"
7
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:573c84dccd8e73c9f70e0c012c482bcc591e45ca79645251fb6487bbfc0d8456
3
+ size 307760552
special_tokens_map.json ADDED
@@ -0,0 +1,125 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "<extra_id_0>",
4
+ "<extra_id_1>",
5
+ "<extra_id_2>",
6
+ "<extra_id_3>",
7
+ "<extra_id_4>",
8
+ "<extra_id_5>",
9
+ "<extra_id_6>",
10
+ "<extra_id_7>",
11
+ "<extra_id_8>",
12
+ "<extra_id_9>",
13
+ "<extra_id_10>",
14
+ "<extra_id_11>",
15
+ "<extra_id_12>",
16
+ "<extra_id_13>",
17
+ "<extra_id_14>",
18
+ "<extra_id_15>",
19
+ "<extra_id_16>",
20
+ "<extra_id_17>",
21
+ "<extra_id_18>",
22
+ "<extra_id_19>",
23
+ "<extra_id_20>",
24
+ "<extra_id_21>",
25
+ "<extra_id_22>",
26
+ "<extra_id_23>",
27
+ "<extra_id_24>",
28
+ "<extra_id_25>",
29
+ "<extra_id_26>",
30
+ "<extra_id_27>",
31
+ "<extra_id_28>",
32
+ "<extra_id_29>",
33
+ "<extra_id_30>",
34
+ "<extra_id_31>",
35
+ "<extra_id_32>",
36
+ "<extra_id_33>",
37
+ "<extra_id_34>",
38
+ "<extra_id_35>",
39
+ "<extra_id_36>",
40
+ "<extra_id_37>",
41
+ "<extra_id_38>",
42
+ "<extra_id_39>",
43
+ "<extra_id_40>",
44
+ "<extra_id_41>",
45
+ "<extra_id_42>",
46
+ "<extra_id_43>",
47
+ "<extra_id_44>",
48
+ "<extra_id_45>",
49
+ "<extra_id_46>",
50
+ "<extra_id_47>",
51
+ "<extra_id_48>",
52
+ "<extra_id_49>",
53
+ "<extra_id_50>",
54
+ "<extra_id_51>",
55
+ "<extra_id_52>",
56
+ "<extra_id_53>",
57
+ "<extra_id_54>",
58
+ "<extra_id_55>",
59
+ "<extra_id_56>",
60
+ "<extra_id_57>",
61
+ "<extra_id_58>",
62
+ "<extra_id_59>",
63
+ "<extra_id_60>",
64
+ "<extra_id_61>",
65
+ "<extra_id_62>",
66
+ "<extra_id_63>",
67
+ "<extra_id_64>",
68
+ "<extra_id_65>",
69
+ "<extra_id_66>",
70
+ "<extra_id_67>",
71
+ "<extra_id_68>",
72
+ "<extra_id_69>",
73
+ "<extra_id_70>",
74
+ "<extra_id_71>",
75
+ "<extra_id_72>",
76
+ "<extra_id_73>",
77
+ "<extra_id_74>",
78
+ "<extra_id_75>",
79
+ "<extra_id_76>",
80
+ "<extra_id_77>",
81
+ "<extra_id_78>",
82
+ "<extra_id_79>",
83
+ "<extra_id_80>",
84
+ "<extra_id_81>",
85
+ "<extra_id_82>",
86
+ "<extra_id_83>",
87
+ "<extra_id_84>",
88
+ "<extra_id_85>",
89
+ "<extra_id_86>",
90
+ "<extra_id_87>",
91
+ "<extra_id_88>",
92
+ "<extra_id_89>",
93
+ "<extra_id_90>",
94
+ "<extra_id_91>",
95
+ "<extra_id_92>",
96
+ "<extra_id_93>",
97
+ "<extra_id_94>",
98
+ "<extra_id_95>",
99
+ "<extra_id_96>",
100
+ "<extra_id_97>",
101
+ "<extra_id_98>",
102
+ "<extra_id_99>"
103
+ ],
104
+ "eos_token": {
105
+ "content": "</s>",
106
+ "lstrip": false,
107
+ "normalized": false,
108
+ "rstrip": false,
109
+ "single_word": false
110
+ },
111
+ "pad_token": {
112
+ "content": "<pad>",
113
+ "lstrip": false,
114
+ "normalized": false,
115
+ "rstrip": false,
116
+ "single_word": false
117
+ },
118
+ "unk_token": {
119
+ "content": "<unk>",
120
+ "lstrip": false,
121
+ "normalized": false,
122
+ "rstrip": false,
123
+ "single_word": false
124
+ }
125
+ }
spiece.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d60acb128cf7b7f2536e8f38a5b18a05535c9e14c7a355904270e15b0945ea86
3
+ size 791656
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,956 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": true,
3
+ "added_tokens_decoder": {
4
+ "0": {
5
+ "content": "<pad>",
6
+ "lstrip": false,
7
+ "normalized": false,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "1": {
13
+ "content": "</s>",
14
+ "lstrip": false,
15
+ "normalized": false,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "2": {
21
+ "content": "<unk>",
22
+ "lstrip": false,
23
+ "normalized": false,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ },
28
+ "32000": {
29
+ "content": "<extra_id_99>",
30
+ "lstrip": false,
31
+ "normalized": false,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": true
35
+ },
36
+ "32001": {
37
+ "content": "<extra_id_98>",
38
+ "lstrip": false,
39
+ "normalized": false,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": true
43
+ },
44
+ "32002": {
45
+ "content": "<extra_id_97>",
46
+ "lstrip": false,
47
+ "normalized": false,
48
+ "rstrip": false,
49
+ "single_word": false,
50
+ "special": true
51
+ },
52
+ "32003": {
53
+ "content": "<extra_id_96>",
54
+ "lstrip": false,
55
+ "normalized": false,
56
+ "rstrip": false,
57
+ "single_word": false,
58
+ "special": true
59
+ },
60
+ "32004": {
61
+ "content": "<extra_id_95>",
62
+ "lstrip": false,
63
+ "normalized": false,
64
+ "rstrip": false,
65
+ "single_word": false,
66
+ "special": true
67
+ },
68
+ "32005": {
69
+ "content": "<extra_id_94>",
70
+ "lstrip": false,
71
+ "normalized": false,
72
+ "rstrip": false,
73
+ "single_word": false,
74
+ "special": true
75
+ },
76
+ "32006": {
77
+ "content": "<extra_id_93>",
78
+ "lstrip": false,
79
+ "normalized": false,
80
+ "rstrip": false,
81
+ "single_word": false,
82
+ "special": true
83
+ },
84
+ "32007": {
85
+ "content": "<extra_id_92>",
86
+ "lstrip": false,
87
+ "normalized": false,
88
+ "rstrip": false,
89
+ "single_word": false,
90
+ "special": true
91
+ },
92
+ "32008": {
93
+ "content": "<extra_id_91>",
94
+ "lstrip": false,
95
+ "normalized": false,
96
+ "rstrip": false,
97
+ "single_word": false,
98
+ "special": true
99
+ },
100
+ "32009": {
101
+ "content": "<extra_id_90>",
102
+ "lstrip": false,
103
+ "normalized": false,
104
+ "rstrip": false,
105
+ "single_word": false,
106
+ "special": true
107
+ },
108
+ "32010": {
109
+ "content": "<extra_id_89>",
110
+ "lstrip": false,
111
+ "normalized": false,
112
+ "rstrip": false,
113
+ "single_word": false,
114
+ "special": true
115
+ },
116
+ "32011": {
117
+ "content": "<extra_id_88>",
118
+ "lstrip": false,
119
+ "normalized": false,
120
+ "rstrip": false,
121
+ "single_word": false,
122
+ "special": true
123
+ },
124
+ "32012": {
125
+ "content": "<extra_id_87>",
126
+ "lstrip": false,
127
+ "normalized": false,
128
+ "rstrip": false,
129
+ "single_word": false,
130
+ "special": true
131
+ },
132
+ "32013": {
133
+ "content": "<extra_id_86>",
134
+ "lstrip": false,
135
+ "normalized": false,
136
+ "rstrip": false,
137
+ "single_word": false,
138
+ "special": true
139
+ },
140
+ "32014": {
141
+ "content": "<extra_id_85>",
142
+ "lstrip": false,
143
+ "normalized": false,
144
+ "rstrip": false,
145
+ "single_word": false,
146
+ "special": true
147
+ },
148
+ "32015": {
149
+ "content": "<extra_id_84>",
150
+ "lstrip": false,
151
+ "normalized": false,
152
+ "rstrip": false,
153
+ "single_word": false,
154
+ "special": true
155
+ },
156
+ "32016": {
157
+ "content": "<extra_id_83>",
158
+ "lstrip": false,
159
+ "normalized": false,
160
+ "rstrip": false,
161
+ "single_word": false,
162
+ "special": true
163
+ },
164
+ "32017": {
165
+ "content": "<extra_id_82>",
166
+ "lstrip": false,
167
+ "normalized": false,
168
+ "rstrip": false,
169
+ "single_word": false,
170
+ "special": true
171
+ },
172
+ "32018": {
173
+ "content": "<extra_id_81>",
174
+ "lstrip": false,
175
+ "normalized": false,
176
+ "rstrip": false,
177
+ "single_word": false,
178
+ "special": true
179
+ },
180
+ "32019": {
181
+ "content": "<extra_id_80>",
182
+ "lstrip": false,
183
+ "normalized": false,
184
+ "rstrip": false,
185
+ "single_word": false,
186
+ "special": true
187
+ },
188
+ "32020": {
189
+ "content": "<extra_id_79>",
190
+ "lstrip": false,
191
+ "normalized": false,
192
+ "rstrip": false,
193
+ "single_word": false,
194
+ "special": true
195
+ },
196
+ "32021": {
197
+ "content": "<extra_id_78>",
198
+ "lstrip": false,
199
+ "normalized": false,
200
+ "rstrip": false,
201
+ "single_word": false,
202
+ "special": true
203
+ },
204
+ "32022": {
205
+ "content": "<extra_id_77>",
206
+ "lstrip": false,
207
+ "normalized": false,
208
+ "rstrip": false,
209
+ "single_word": false,
210
+ "special": true
211
+ },
212
+ "32023": {
213
+ "content": "<extra_id_76>",
214
+ "lstrip": false,
215
+ "normalized": false,
216
+ "rstrip": false,
217
+ "single_word": false,
218
+ "special": true
219
+ },
220
+ "32024": {
221
+ "content": "<extra_id_75>",
222
+ "lstrip": false,
223
+ "normalized": false,
224
+ "rstrip": false,
225
+ "single_word": false,
226
+ "special": true
227
+ },
228
+ "32025": {
229
+ "content": "<extra_id_74>",
230
+ "lstrip": false,
231
+ "normalized": false,
232
+ "rstrip": false,
233
+ "single_word": false,
234
+ "special": true
235
+ },
236
+ "32026": {
237
+ "content": "<extra_id_73>",
238
+ "lstrip": false,
239
+ "normalized": false,
240
+ "rstrip": false,
241
+ "single_word": false,
242
+ "special": true
243
+ },
244
+ "32027": {
245
+ "content": "<extra_id_72>",
246
+ "lstrip": false,
247
+ "normalized": false,
248
+ "rstrip": false,
249
+ "single_word": false,
250
+ "special": true
251
+ },
252
+ "32028": {
253
+ "content": "<extra_id_71>",
254
+ "lstrip": false,
255
+ "normalized": false,
256
+ "rstrip": false,
257
+ "single_word": false,
258
+ "special": true
259
+ },
260
+ "32029": {
261
+ "content": "<extra_id_70>",
262
+ "lstrip": false,
263
+ "normalized": false,
264
+ "rstrip": false,
265
+ "single_word": false,
266
+ "special": true
267
+ },
268
+ "32030": {
269
+ "content": "<extra_id_69>",
270
+ "lstrip": false,
271
+ "normalized": false,
272
+ "rstrip": false,
273
+ "single_word": false,
274
+ "special": true
275
+ },
276
+ "32031": {
277
+ "content": "<extra_id_68>",
278
+ "lstrip": false,
279
+ "normalized": false,
280
+ "rstrip": false,
281
+ "single_word": false,
282
+ "special": true
283
+ },
284
+ "32032": {
285
+ "content": "<extra_id_67>",
286
+ "lstrip": false,
287
+ "normalized": false,
288
+ "rstrip": false,
289
+ "single_word": false,
290
+ "special": true
291
+ },
292
+ "32033": {
293
+ "content": "<extra_id_66>",
294
+ "lstrip": false,
295
+ "normalized": false,
296
+ "rstrip": false,
297
+ "single_word": false,
298
+ "special": true
299
+ },
300
+ "32034": {
301
+ "content": "<extra_id_65>",
302
+ "lstrip": false,
303
+ "normalized": false,
304
+ "rstrip": false,
305
+ "single_word": false,
306
+ "special": true
307
+ },
308
+ "32035": {
309
+ "content": "<extra_id_64>",
310
+ "lstrip": false,
311
+ "normalized": false,
312
+ "rstrip": false,
313
+ "single_word": false,
314
+ "special": true
315
+ },
316
+ "32036": {
317
+ "content": "<extra_id_63>",
318
+ "lstrip": false,
319
+ "normalized": false,
320
+ "rstrip": false,
321
+ "single_word": false,
322
+ "special": true
323
+ },
324
+ "32037": {
325
+ "content": "<extra_id_62>",
326
+ "lstrip": false,
327
+ "normalized": false,
328
+ "rstrip": false,
329
+ "single_word": false,
330
+ "special": true
331
+ },
332
+ "32038": {
333
+ "content": "<extra_id_61>",
334
+ "lstrip": false,
335
+ "normalized": false,
336
+ "rstrip": false,
337
+ "single_word": false,
338
+ "special": true
339
+ },
340
+ "32039": {
341
+ "content": "<extra_id_60>",
342
+ "lstrip": false,
343
+ "normalized": false,
344
+ "rstrip": false,
345
+ "single_word": false,
346
+ "special": true
347
+ },
348
+ "32040": {
349
+ "content": "<extra_id_59>",
350
+ "lstrip": false,
351
+ "normalized": false,
352
+ "rstrip": false,
353
+ "single_word": false,
354
+ "special": true
355
+ },
356
+ "32041": {
357
+ "content": "<extra_id_58>",
358
+ "lstrip": false,
359
+ "normalized": false,
360
+ "rstrip": false,
361
+ "single_word": false,
362
+ "special": true
363
+ },
364
+ "32042": {
365
+ "content": "<extra_id_57>",
366
+ "lstrip": false,
367
+ "normalized": false,
368
+ "rstrip": false,
369
+ "single_word": false,
370
+ "special": true
371
+ },
372
+ "32043": {
373
+ "content": "<extra_id_56>",
374
+ "lstrip": false,
375
+ "normalized": false,
376
+ "rstrip": false,
377
+ "single_word": false,
378
+ "special": true
379
+ },
380
+ "32044": {
381
+ "content": "<extra_id_55>",
382
+ "lstrip": false,
383
+ "normalized": false,
384
+ "rstrip": false,
385
+ "single_word": false,
386
+ "special": true
387
+ },
388
+ "32045": {
389
+ "content": "<extra_id_54>",
390
+ "lstrip": false,
391
+ "normalized": false,
392
+ "rstrip": false,
393
+ "single_word": false,
394
+ "special": true
395
+ },
396
+ "32046": {
397
+ "content": "<extra_id_53>",
398
+ "lstrip": false,
399
+ "normalized": false,
400
+ "rstrip": false,
401
+ "single_word": false,
402
+ "special": true
403
+ },
404
+ "32047": {
405
+ "content": "<extra_id_52>",
406
+ "lstrip": false,
407
+ "normalized": false,
408
+ "rstrip": false,
409
+ "single_word": false,
410
+ "special": true
411
+ },
412
+ "32048": {
413
+ "content": "<extra_id_51>",
414
+ "lstrip": false,
415
+ "normalized": false,
416
+ "rstrip": false,
417
+ "single_word": false,
418
+ "special": true
419
+ },
420
+ "32049": {
421
+ "content": "<extra_id_50>",
422
+ "lstrip": false,
423
+ "normalized": false,
424
+ "rstrip": false,
425
+ "single_word": false,
426
+ "special": true
427
+ },
428
+ "32050": {
429
+ "content": "<extra_id_49>",
430
+ "lstrip": false,
431
+ "normalized": false,
432
+ "rstrip": false,
433
+ "single_word": false,
434
+ "special": true
435
+ },
436
+ "32051": {
437
+ "content": "<extra_id_48>",
438
+ "lstrip": false,
439
+ "normalized": false,
440
+ "rstrip": false,
441
+ "single_word": false,
442
+ "special": true
443
+ },
444
+ "32052": {
445
+ "content": "<extra_id_47>",
446
+ "lstrip": false,
447
+ "normalized": false,
448
+ "rstrip": false,
449
+ "single_word": false,
450
+ "special": true
451
+ },
452
+ "32053": {
453
+ "content": "<extra_id_46>",
454
+ "lstrip": false,
455
+ "normalized": false,
456
+ "rstrip": false,
457
+ "single_word": false,
458
+ "special": true
459
+ },
460
+ "32054": {
461
+ "content": "<extra_id_45>",
462
+ "lstrip": false,
463
+ "normalized": false,
464
+ "rstrip": false,
465
+ "single_word": false,
466
+ "special": true
467
+ },
468
+ "32055": {
469
+ "content": "<extra_id_44>",
470
+ "lstrip": false,
471
+ "normalized": false,
472
+ "rstrip": false,
473
+ "single_word": false,
474
+ "special": true
475
+ },
476
+ "32056": {
477
+ "content": "<extra_id_43>",
478
+ "lstrip": false,
479
+ "normalized": false,
480
+ "rstrip": false,
481
+ "single_word": false,
482
+ "special": true
483
+ },
484
+ "32057": {
485
+ "content": "<extra_id_42>",
486
+ "lstrip": false,
487
+ "normalized": false,
488
+ "rstrip": false,
489
+ "single_word": false,
490
+ "special": true
491
+ },
492
+ "32058": {
493
+ "content": "<extra_id_41>",
494
+ "lstrip": false,
495
+ "normalized": false,
496
+ "rstrip": false,
497
+ "single_word": false,
498
+ "special": true
499
+ },
500
+ "32059": {
501
+ "content": "<extra_id_40>",
502
+ "lstrip": false,
503
+ "normalized": false,
504
+ "rstrip": false,
505
+ "single_word": false,
506
+ "special": true
507
+ },
508
+ "32060": {
509
+ "content": "<extra_id_39>",
510
+ "lstrip": false,
511
+ "normalized": false,
512
+ "rstrip": false,
513
+ "single_word": false,
514
+ "special": true
515
+ },
516
+ "32061": {
517
+ "content": "<extra_id_38>",
518
+ "lstrip": false,
519
+ "normalized": false,
520
+ "rstrip": false,
521
+ "single_word": false,
522
+ "special": true
523
+ },
524
+ "32062": {
525
+ "content": "<extra_id_37>",
526
+ "lstrip": false,
527
+ "normalized": false,
528
+ "rstrip": false,
529
+ "single_word": false,
530
+ "special": true
531
+ },
532
+ "32063": {
533
+ "content": "<extra_id_36>",
534
+ "lstrip": false,
535
+ "normalized": false,
536
+ "rstrip": false,
537
+ "single_word": false,
538
+ "special": true
539
+ },
540
+ "32064": {
541
+ "content": "<extra_id_35>",
542
+ "lstrip": false,
543
+ "normalized": false,
544
+ "rstrip": false,
545
+ "single_word": false,
546
+ "special": true
547
+ },
548
+ "32065": {
549
+ "content": "<extra_id_34>",
550
+ "lstrip": false,
551
+ "normalized": false,
552
+ "rstrip": false,
553
+ "single_word": false,
554
+ "special": true
555
+ },
556
+ "32066": {
557
+ "content": "<extra_id_33>",
558
+ "lstrip": false,
559
+ "normalized": false,
560
+ "rstrip": false,
561
+ "single_word": false,
562
+ "special": true
563
+ },
564
+ "32067": {
565
+ "content": "<extra_id_32>",
566
+ "lstrip": false,
567
+ "normalized": false,
568
+ "rstrip": false,
569
+ "single_word": false,
570
+ "special": true
571
+ },
572
+ "32068": {
573
+ "content": "<extra_id_31>",
574
+ "lstrip": false,
575
+ "normalized": false,
576
+ "rstrip": false,
577
+ "single_word": false,
578
+ "special": true
579
+ },
580
+ "32069": {
581
+ "content": "<extra_id_30>",
582
+ "lstrip": false,
583
+ "normalized": false,
584
+ "rstrip": false,
585
+ "single_word": false,
586
+ "special": true
587
+ },
588
+ "32070": {
589
+ "content": "<extra_id_29>",
590
+ "lstrip": false,
591
+ "normalized": false,
592
+ "rstrip": false,
593
+ "single_word": false,
594
+ "special": true
595
+ },
596
+ "32071": {
597
+ "content": "<extra_id_28>",
598
+ "lstrip": false,
599
+ "normalized": false,
600
+ "rstrip": false,
601
+ "single_word": false,
602
+ "special": true
603
+ },
604
+ "32072": {
605
+ "content": "<extra_id_27>",
606
+ "lstrip": false,
607
+ "normalized": false,
608
+ "rstrip": false,
609
+ "single_word": false,
610
+ "special": true
611
+ },
612
+ "32073": {
613
+ "content": "<extra_id_26>",
614
+ "lstrip": false,
615
+ "normalized": false,
616
+ "rstrip": false,
617
+ "single_word": false,
618
+ "special": true
619
+ },
620
+ "32074": {
621
+ "content": "<extra_id_25>",
622
+ "lstrip": false,
623
+ "normalized": false,
624
+ "rstrip": false,
625
+ "single_word": false,
626
+ "special": true
627
+ },
628
+ "32075": {
629
+ "content": "<extra_id_24>",
630
+ "lstrip": false,
631
+ "normalized": false,
632
+ "rstrip": false,
633
+ "single_word": false,
634
+ "special": true
635
+ },
636
+ "32076": {
637
+ "content": "<extra_id_23>",
638
+ "lstrip": false,
639
+ "normalized": false,
640
+ "rstrip": false,
641
+ "single_word": false,
642
+ "special": true
643
+ },
644
+ "32077": {
645
+ "content": "<extra_id_22>",
646
+ "lstrip": false,
647
+ "normalized": false,
648
+ "rstrip": false,
649
+ "single_word": false,
650
+ "special": true
651
+ },
652
+ "32078": {
653
+ "content": "<extra_id_21>",
654
+ "lstrip": false,
655
+ "normalized": false,
656
+ "rstrip": false,
657
+ "single_word": false,
658
+ "special": true
659
+ },
660
+ "32079": {
661
+ "content": "<extra_id_20>",
662
+ "lstrip": false,
663
+ "normalized": false,
664
+ "rstrip": false,
665
+ "single_word": false,
666
+ "special": true
667
+ },
668
+ "32080": {
669
+ "content": "<extra_id_19>",
670
+ "lstrip": false,
671
+ "normalized": false,
672
+ "rstrip": false,
673
+ "single_word": false,
674
+ "special": true
675
+ },
676
+ "32081": {
677
+ "content": "<extra_id_18>",
678
+ "lstrip": false,
679
+ "normalized": false,
680
+ "rstrip": false,
681
+ "single_word": false,
682
+ "special": true
683
+ },
684
+ "32082": {
685
+ "content": "<extra_id_17>",
686
+ "lstrip": false,
687
+ "normalized": false,
688
+ "rstrip": false,
689
+ "single_word": false,
690
+ "special": true
691
+ },
692
+ "32083": {
693
+ "content": "<extra_id_16>",
694
+ "lstrip": false,
695
+ "normalized": false,
696
+ "rstrip": false,
697
+ "single_word": false,
698
+ "special": true
699
+ },
700
+ "32084": {
701
+ "content": "<extra_id_15>",
702
+ "lstrip": false,
703
+ "normalized": false,
704
+ "rstrip": false,
705
+ "single_word": false,
706
+ "special": true
707
+ },
708
+ "32085": {
709
+ "content": "<extra_id_14>",
710
+ "lstrip": false,
711
+ "normalized": false,
712
+ "rstrip": false,
713
+ "single_word": false,
714
+ "special": true
715
+ },
716
+ "32086": {
717
+ "content": "<extra_id_13>",
718
+ "lstrip": false,
719
+ "normalized": false,
720
+ "rstrip": false,
721
+ "single_word": false,
722
+ "special": true
723
+ },
724
+ "32087": {
725
+ "content": "<extra_id_12>",
726
+ "lstrip": false,
727
+ "normalized": false,
728
+ "rstrip": false,
729
+ "single_word": false,
730
+ "special": true
731
+ },
732
+ "32088": {
733
+ "content": "<extra_id_11>",
734
+ "lstrip": false,
735
+ "normalized": false,
736
+ "rstrip": false,
737
+ "single_word": false,
738
+ "special": true
739
+ },
740
+ "32089": {
741
+ "content": "<extra_id_10>",
742
+ "lstrip": false,
743
+ "normalized": false,
744
+ "rstrip": false,
745
+ "single_word": false,
746
+ "special": true
747
+ },
748
+ "32090": {
749
+ "content": "<extra_id_9>",
750
+ "lstrip": false,
751
+ "normalized": false,
752
+ "rstrip": false,
753
+ "single_word": false,
754
+ "special": true
755
+ },
756
+ "32091": {
757
+ "content": "<extra_id_8>",
758
+ "lstrip": false,
759
+ "normalized": false,
760
+ "rstrip": false,
761
+ "single_word": false,
762
+ "special": true
763
+ },
764
+ "32092": {
765
+ "content": "<extra_id_7>",
766
+ "lstrip": false,
767
+ "normalized": false,
768
+ "rstrip": false,
769
+ "single_word": false,
770
+ "special": true
771
+ },
772
+ "32093": {
773
+ "content": "<extra_id_6>",
774
+ "lstrip": false,
775
+ "normalized": false,
776
+ "rstrip": false,
777
+ "single_word": false,
778
+ "special": true
779
+ },
780
+ "32094": {
781
+ "content": "<extra_id_5>",
782
+ "lstrip": false,
783
+ "normalized": false,
784
+ "rstrip": false,
785
+ "single_word": false,
786
+ "special": true
787
+ },
788
+ "32095": {
789
+ "content": "<extra_id_4>",
790
+ "lstrip": false,
791
+ "normalized": false,
792
+ "rstrip": false,
793
+ "single_word": false,
794
+ "special": true
795
+ },
796
+ "32096": {
797
+ "content": "<extra_id_3>",
798
+ "lstrip": false,
799
+ "normalized": false,
800
+ "rstrip": false,
801
+ "single_word": false,
802
+ "special": true
803
+ },
804
+ "32097": {
805
+ "content": "<extra_id_2>",
806
+ "lstrip": false,
807
+ "normalized": false,
808
+ "rstrip": false,
809
+ "single_word": false,
810
+ "special": true
811
+ },
812
+ "32098": {
813
+ "content": "<extra_id_1>",
814
+ "lstrip": false,
815
+ "normalized": false,
816
+ "rstrip": false,
817
+ "single_word": false,
818
+ "special": true
819
+ },
820
+ "32099": {
821
+ "content": "<extra_id_0>",
822
+ "lstrip": false,
823
+ "normalized": false,
824
+ "rstrip": false,
825
+ "single_word": false,
826
+ "special": true
827
+ },
828
+ "32100": {
829
+ "content": "<|sep|>",
830
+ "lstrip": false,
831
+ "normalized": true,
832
+ "rstrip": false,
833
+ "single_word": false,
834
+ "special": false
835
+ },
836
+ "32101": {
837
+ "content": "<|end|>",
838
+ "lstrip": false,
839
+ "normalized": true,
840
+ "rstrip": false,
841
+ "single_word": false,
842
+ "special": false
843
+ }
844
+ },
845
+ "additional_special_tokens": [
846
+ "<extra_id_0>",
847
+ "<extra_id_1>",
848
+ "<extra_id_2>",
849
+ "<extra_id_3>",
850
+ "<extra_id_4>",
851
+ "<extra_id_5>",
852
+ "<extra_id_6>",
853
+ "<extra_id_7>",
854
+ "<extra_id_8>",
855
+ "<extra_id_9>",
856
+ "<extra_id_10>",
857
+ "<extra_id_11>",
858
+ "<extra_id_12>",
859
+ "<extra_id_13>",
860
+ "<extra_id_14>",
861
+ "<extra_id_15>",
862
+ "<extra_id_16>",
863
+ "<extra_id_17>",
864
+ "<extra_id_18>",
865
+ "<extra_id_19>",
866
+ "<extra_id_20>",
867
+ "<extra_id_21>",
868
+ "<extra_id_22>",
869
+ "<extra_id_23>",
870
+ "<extra_id_24>",
871
+ "<extra_id_25>",
872
+ "<extra_id_26>",
873
+ "<extra_id_27>",
874
+ "<extra_id_28>",
875
+ "<extra_id_29>",
876
+ "<extra_id_30>",
877
+ "<extra_id_31>",
878
+ "<extra_id_32>",
879
+ "<extra_id_33>",
880
+ "<extra_id_34>",
881
+ "<extra_id_35>",
882
+ "<extra_id_36>",
883
+ "<extra_id_37>",
884
+ "<extra_id_38>",
885
+ "<extra_id_39>",
886
+ "<extra_id_40>",
887
+ "<extra_id_41>",
888
+ "<extra_id_42>",
889
+ "<extra_id_43>",
890
+ "<extra_id_44>",
891
+ "<extra_id_45>",
892
+ "<extra_id_46>",
893
+ "<extra_id_47>",
894
+ "<extra_id_48>",
895
+ "<extra_id_49>",
896
+ "<extra_id_50>",
897
+ "<extra_id_51>",
898
+ "<extra_id_52>",
899
+ "<extra_id_53>",
900
+ "<extra_id_54>",
901
+ "<extra_id_55>",
902
+ "<extra_id_56>",
903
+ "<extra_id_57>",
904
+ "<extra_id_58>",
905
+ "<extra_id_59>",
906
+ "<extra_id_60>",
907
+ "<extra_id_61>",
908
+ "<extra_id_62>",
909
+ "<extra_id_63>",
910
+ "<extra_id_64>",
911
+ "<extra_id_65>",
912
+ "<extra_id_66>",
913
+ "<extra_id_67>",
914
+ "<extra_id_68>",
915
+ "<extra_id_69>",
916
+ "<extra_id_70>",
917
+ "<extra_id_71>",
918
+ "<extra_id_72>",
919
+ "<extra_id_73>",
920
+ "<extra_id_74>",
921
+ "<extra_id_75>",
922
+ "<extra_id_76>",
923
+ "<extra_id_77>",
924
+ "<extra_id_78>",
925
+ "<extra_id_79>",
926
+ "<extra_id_80>",
927
+ "<extra_id_81>",
928
+ "<extra_id_82>",
929
+ "<extra_id_83>",
930
+ "<extra_id_84>",
931
+ "<extra_id_85>",
932
+ "<extra_id_86>",
933
+ "<extra_id_87>",
934
+ "<extra_id_88>",
935
+ "<extra_id_89>",
936
+ "<extra_id_90>",
937
+ "<extra_id_91>",
938
+ "<extra_id_92>",
939
+ "<extra_id_93>",
940
+ "<extra_id_94>",
941
+ "<extra_id_95>",
942
+ "<extra_id_96>",
943
+ "<extra_id_97>",
944
+ "<extra_id_98>",
945
+ "<extra_id_99>"
946
+ ],
947
+ "clean_up_tokenization_spaces": true,
948
+ "eos_token": "</s>",
949
+ "extra_ids": 100,
950
+ "legacy": true,
951
+ "model_max_length": 512,
952
+ "pad_token": "<pad>",
953
+ "sp_model_kwargs": {},
954
+ "tokenizer_class": "T5Tokenizer",
955
+ "unk_token": "<unk>"
956
+ }
train_results.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 5.0,
3
+ "num_input_tokens_seen": 332318598,
4
+ "total_flos": 1.2062750373789696e+17,
5
+ "train_loss": 0.7601320325591829,
6
+ "train_runtime": 7988.2275,
7
+ "train_samples": 159922,
8
+ "train_samples_per_second": 100.099,
9
+ "train_steps_per_second": 12.513,
10
+ "train_tokens_per_second": 41593.134
11
+ }
trainer_state.json ADDED
@@ -0,0 +1,1681 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.5393198132514954,
3
+ "best_model_checkpoint": "/media/user/Expansion/flan-t5-small-ner/checkpoint-99955",
4
+ "epoch": 5.0,
5
+ "eval_steps": 500,
6
+ "global_step": 99955,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.02501125506477915,
13
+ "grad_norm": 27.66358757019043,
14
+ "learning_rate": 4.974988744935221e-05,
15
+ "loss": 4.6267,
16
+ "num_input_tokens_seen": 1673336,
17
+ "step": 500
18
+ },
19
+ {
20
+ "epoch": 0.0500225101295583,
21
+ "grad_norm": 9.223219871520996,
22
+ "learning_rate": 4.949977489870442e-05,
23
+ "loss": 2.1909,
24
+ "num_input_tokens_seen": 3361736,
25
+ "step": 1000
26
+ },
27
+ {
28
+ "epoch": 0.07503376519433745,
29
+ "grad_norm": 15.481683731079102,
30
+ "learning_rate": 4.924966234805663e-05,
31
+ "loss": 1.8355,
32
+ "num_input_tokens_seen": 5013800,
33
+ "step": 1500
34
+ },
35
+ {
36
+ "epoch": 0.1000450202591166,
37
+ "grad_norm": 14.396512985229492,
38
+ "learning_rate": 4.8999549797408836e-05,
39
+ "loss": 1.5648,
40
+ "num_input_tokens_seen": 6667312,
41
+ "step": 2000
42
+ },
43
+ {
44
+ "epoch": 0.12505627532389577,
45
+ "grad_norm": 7.068989276885986,
46
+ "learning_rate": 4.8749437246761046e-05,
47
+ "loss": 1.4252,
48
+ "num_input_tokens_seen": 8347016,
49
+ "step": 2500
50
+ },
51
+ {
52
+ "epoch": 0.1500675303886749,
53
+ "grad_norm": 11.330971717834473,
54
+ "learning_rate": 4.849932469611325e-05,
55
+ "loss": 1.3972,
56
+ "num_input_tokens_seen": 10008296,
57
+ "step": 3000
58
+ },
59
+ {
60
+ "epoch": 0.17507878545345407,
61
+ "grad_norm": 9.403321266174316,
62
+ "learning_rate": 4.824921214546546e-05,
63
+ "loss": 1.3001,
64
+ "num_input_tokens_seen": 11658808,
65
+ "step": 3500
66
+ },
67
+ {
68
+ "epoch": 0.2000900405182332,
69
+ "grad_norm": 8.147115707397461,
70
+ "learning_rate": 4.799909959481767e-05,
71
+ "loss": 1.2625,
72
+ "num_input_tokens_seen": 13331648,
73
+ "step": 4000
74
+ },
75
+ {
76
+ "epoch": 0.22510129558301237,
77
+ "grad_norm": 13.405184745788574,
78
+ "learning_rate": 4.774898704416988e-05,
79
+ "loss": 1.1839,
80
+ "num_input_tokens_seen": 14982440,
81
+ "step": 4500
82
+ },
83
+ {
84
+ "epoch": 0.25011255064779153,
85
+ "grad_norm": 20.70949363708496,
86
+ "learning_rate": 4.749887449352209e-05,
87
+ "loss": 1.1598,
88
+ "num_input_tokens_seen": 16633632,
89
+ "step": 5000
90
+ },
91
+ {
92
+ "epoch": 0.27512380571257067,
93
+ "grad_norm": 16.94267463684082,
94
+ "learning_rate": 4.72487619428743e-05,
95
+ "loss": 1.1473,
96
+ "num_input_tokens_seen": 18311672,
97
+ "step": 5500
98
+ },
99
+ {
100
+ "epoch": 0.3001350607773498,
101
+ "grad_norm": 8.609989166259766,
102
+ "learning_rate": 4.69986493922265e-05,
103
+ "loss": 1.1098,
104
+ "num_input_tokens_seen": 19980456,
105
+ "step": 6000
106
+ },
107
+ {
108
+ "epoch": 0.32514631584212894,
109
+ "grad_norm": 9.003643989562988,
110
+ "learning_rate": 4.674853684157871e-05,
111
+ "loss": 1.0973,
112
+ "num_input_tokens_seen": 21646328,
113
+ "step": 6500
114
+ },
115
+ {
116
+ "epoch": 0.35015757090690813,
117
+ "grad_norm": 18.364194869995117,
118
+ "learning_rate": 4.649842429093092e-05,
119
+ "loss": 1.0987,
120
+ "num_input_tokens_seen": 23277400,
121
+ "step": 7000
122
+ },
123
+ {
124
+ "epoch": 0.37516882597168727,
125
+ "grad_norm": 13.544733047485352,
126
+ "learning_rate": 4.624831174028313e-05,
127
+ "loss": 1.0642,
128
+ "num_input_tokens_seen": 24915304,
129
+ "step": 7500
130
+ },
131
+ {
132
+ "epoch": 0.4001800810364664,
133
+ "grad_norm": 14.257452011108398,
134
+ "learning_rate": 4.5998199189635336e-05,
135
+ "loss": 1.0414,
136
+ "num_input_tokens_seen": 26590576,
137
+ "step": 8000
138
+ },
139
+ {
140
+ "epoch": 0.42519133610124554,
141
+ "grad_norm": 10.29515266418457,
142
+ "learning_rate": 4.5748086638987546e-05,
143
+ "loss": 1.0634,
144
+ "num_input_tokens_seen": 28236280,
145
+ "step": 8500
146
+ },
147
+ {
148
+ "epoch": 0.45020259116602473,
149
+ "grad_norm": 13.840631484985352,
150
+ "learning_rate": 4.5497974088339756e-05,
151
+ "loss": 0.9817,
152
+ "num_input_tokens_seen": 29891480,
153
+ "step": 9000
154
+ },
155
+ {
156
+ "epoch": 0.47521384623080387,
157
+ "grad_norm": 12.118327140808105,
158
+ "learning_rate": 4.5247861537691966e-05,
159
+ "loss": 1.0122,
160
+ "num_input_tokens_seen": 31551000,
161
+ "step": 9500
162
+ },
163
+ {
164
+ "epoch": 0.5002251012955831,
165
+ "grad_norm": 8.115203857421875,
166
+ "learning_rate": 4.499774898704417e-05,
167
+ "loss": 0.9802,
168
+ "num_input_tokens_seen": 33221384,
169
+ "step": 10000
170
+ },
171
+ {
172
+ "epoch": 0.5252363563603621,
173
+ "grad_norm": 8.905954360961914,
174
+ "learning_rate": 4.474763643639638e-05,
175
+ "loss": 0.9796,
176
+ "num_input_tokens_seen": 34891392,
177
+ "step": 10500
178
+ },
179
+ {
180
+ "epoch": 0.5502476114251413,
181
+ "grad_norm": 10.70656681060791,
182
+ "learning_rate": 4.449752388574859e-05,
183
+ "loss": 1.0031,
184
+ "num_input_tokens_seen": 36518768,
185
+ "step": 11000
186
+ },
187
+ {
188
+ "epoch": 0.5752588664899204,
189
+ "grad_norm": 12.424896240234375,
190
+ "learning_rate": 4.42474113351008e-05,
191
+ "loss": 0.9591,
192
+ "num_input_tokens_seen": 38147456,
193
+ "step": 11500
194
+ },
195
+ {
196
+ "epoch": 0.6002701215546996,
197
+ "grad_norm": 10.77695083618164,
198
+ "learning_rate": 4.399729878445301e-05,
199
+ "loss": 0.9338,
200
+ "num_input_tokens_seen": 39823976,
201
+ "step": 12000
202
+ },
203
+ {
204
+ "epoch": 0.6252813766194788,
205
+ "grad_norm": 12.77743911743164,
206
+ "learning_rate": 4.374718623380521e-05,
207
+ "loss": 0.9112,
208
+ "num_input_tokens_seen": 41493480,
209
+ "step": 12500
210
+ },
211
+ {
212
+ "epoch": 0.6502926316842579,
213
+ "grad_norm": 16.060897827148438,
214
+ "learning_rate": 4.349707368315742e-05,
215
+ "loss": 0.915,
216
+ "num_input_tokens_seen": 43130832,
217
+ "step": 13000
218
+ },
219
+ {
220
+ "epoch": 0.6753038867490371,
221
+ "grad_norm": 17.562183380126953,
222
+ "learning_rate": 4.324696113250963e-05,
223
+ "loss": 0.9096,
224
+ "num_input_tokens_seen": 44779392,
225
+ "step": 13500
226
+ },
227
+ {
228
+ "epoch": 0.7003151418138163,
229
+ "grad_norm": 12.406323432922363,
230
+ "learning_rate": 4.2996848581861835e-05,
231
+ "loss": 0.9499,
232
+ "num_input_tokens_seen": 46433856,
233
+ "step": 14000
234
+ },
235
+ {
236
+ "epoch": 0.7253263968785953,
237
+ "grad_norm": 15.567843437194824,
238
+ "learning_rate": 4.2746736031214045e-05,
239
+ "loss": 0.923,
240
+ "num_input_tokens_seen": 48102016,
241
+ "step": 14500
242
+ },
243
+ {
244
+ "epoch": 0.7503376519433745,
245
+ "grad_norm": 9.45335578918457,
246
+ "learning_rate": 4.2496623480566255e-05,
247
+ "loss": 0.9285,
248
+ "num_input_tokens_seen": 49796432,
249
+ "step": 15000
250
+ },
251
+ {
252
+ "epoch": 0.7753489070081536,
253
+ "grad_norm": 7.158623695373535,
254
+ "learning_rate": 4.2246510929918465e-05,
255
+ "loss": 0.9023,
256
+ "num_input_tokens_seen": 51432848,
257
+ "step": 15500
258
+ },
259
+ {
260
+ "epoch": 0.8003601620729328,
261
+ "grad_norm": 9.542813301086426,
262
+ "learning_rate": 4.1996398379270675e-05,
263
+ "loss": 0.9237,
264
+ "num_input_tokens_seen": 53083496,
265
+ "step": 16000
266
+ },
267
+ {
268
+ "epoch": 0.825371417137712,
269
+ "grad_norm": 10.027923583984375,
270
+ "learning_rate": 4.1746285828622885e-05,
271
+ "loss": 0.8813,
272
+ "num_input_tokens_seen": 54755032,
273
+ "step": 16500
274
+ },
275
+ {
276
+ "epoch": 0.8503826722024911,
277
+ "grad_norm": 18.8748722076416,
278
+ "learning_rate": 4.1496173277975095e-05,
279
+ "loss": 0.9036,
280
+ "num_input_tokens_seen": 56411184,
281
+ "step": 17000
282
+ },
283
+ {
284
+ "epoch": 0.8753939272672703,
285
+ "grad_norm": 12.792276382446289,
286
+ "learning_rate": 4.12460607273273e-05,
287
+ "loss": 0.8589,
288
+ "num_input_tokens_seen": 58070520,
289
+ "step": 17500
290
+ },
291
+ {
292
+ "epoch": 0.9004051823320495,
293
+ "grad_norm": 6.8420491218566895,
294
+ "learning_rate": 4.09959481766795e-05,
295
+ "loss": 0.8855,
296
+ "num_input_tokens_seen": 59745800,
297
+ "step": 18000
298
+ },
299
+ {
300
+ "epoch": 0.9254164373968286,
301
+ "grad_norm": 9.066823959350586,
302
+ "learning_rate": 4.074583562603171e-05,
303
+ "loss": 0.8773,
304
+ "num_input_tokens_seen": 61457288,
305
+ "step": 18500
306
+ },
307
+ {
308
+ "epoch": 0.9504276924616077,
309
+ "grad_norm": 7.002307415008545,
310
+ "learning_rate": 4.049572307538392e-05,
311
+ "loss": 0.8747,
312
+ "num_input_tokens_seen": 63139928,
313
+ "step": 19000
314
+ },
315
+ {
316
+ "epoch": 0.9754389475263868,
317
+ "grad_norm": 14.685755729675293,
318
+ "learning_rate": 4.024561052473613e-05,
319
+ "loss": 0.8398,
320
+ "num_input_tokens_seen": 64811920,
321
+ "step": 19500
322
+ },
323
+ {
324
+ "epoch": 1.0,
325
+ "eval_loss": 0.6227446794509888,
326
+ "eval_runtime": 96.3481,
327
+ "eval_samples_per_second": 414.964,
328
+ "eval_steps_per_second": 51.874,
329
+ "num_input_tokens_seen": 66451084,
330
+ "step": 19991
331
+ },
332
+ {
333
+ "epoch": 1.0004502025911661,
334
+ "grad_norm": 13.560747146606445,
335
+ "learning_rate": 3.999549797408834e-05,
336
+ "loss": 0.852,
337
+ "num_input_tokens_seen": 66482076,
338
+ "step": 20000
339
+ },
340
+ {
341
+ "epoch": 1.025461457655945,
342
+ "grad_norm": 4.446373462677002,
343
+ "learning_rate": 3.974538542344055e-05,
344
+ "loss": 0.7973,
345
+ "num_input_tokens_seen": 68132180,
346
+ "step": 20500
347
+ },
348
+ {
349
+ "epoch": 1.0504727127207243,
350
+ "grad_norm": 3.456674098968506,
351
+ "learning_rate": 3.949527287279276e-05,
352
+ "loss": 0.8215,
353
+ "num_input_tokens_seen": 69804380,
354
+ "step": 21000
355
+ },
356
+ {
357
+ "epoch": 1.0754839677855035,
358
+ "grad_norm": 8.283075332641602,
359
+ "learning_rate": 3.924516032214497e-05,
360
+ "loss": 0.8081,
361
+ "num_input_tokens_seen": 71452668,
362
+ "step": 21500
363
+ },
364
+ {
365
+ "epoch": 1.1004952228502827,
366
+ "grad_norm": 9.358149528503418,
367
+ "learning_rate": 3.8995047771497175e-05,
368
+ "loss": 0.7991,
369
+ "num_input_tokens_seen": 73104948,
370
+ "step": 22000
371
+ },
372
+ {
373
+ "epoch": 1.1255064779150619,
374
+ "grad_norm": 9.011244773864746,
375
+ "learning_rate": 3.8744935220849385e-05,
376
+ "loss": 0.7839,
377
+ "num_input_tokens_seen": 74751164,
378
+ "step": 22500
379
+ },
380
+ {
381
+ "epoch": 1.1505177329798408,
382
+ "grad_norm": 5.775268077850342,
383
+ "learning_rate": 3.849482267020159e-05,
384
+ "loss": 0.7515,
385
+ "num_input_tokens_seen": 76431460,
386
+ "step": 23000
387
+ },
388
+ {
389
+ "epoch": 1.17552898804462,
390
+ "grad_norm": 13.273436546325684,
391
+ "learning_rate": 3.82447101195538e-05,
392
+ "loss": 0.7821,
393
+ "num_input_tokens_seen": 78092124,
394
+ "step": 23500
395
+ },
396
+ {
397
+ "epoch": 1.2005402431093992,
398
+ "grad_norm": 10.351176261901855,
399
+ "learning_rate": 3.799459756890601e-05,
400
+ "loss": 0.772,
401
+ "num_input_tokens_seen": 79736012,
402
+ "step": 24000
403
+ },
404
+ {
405
+ "epoch": 1.2255514981741784,
406
+ "grad_norm": 14.834792137145996,
407
+ "learning_rate": 3.774448501825822e-05,
408
+ "loss": 0.78,
409
+ "num_input_tokens_seen": 81414220,
410
+ "step": 24500
411
+ },
412
+ {
413
+ "epoch": 1.2505627532389576,
414
+ "grad_norm": 14.160717964172363,
415
+ "learning_rate": 3.749437246761043e-05,
416
+ "loss": 0.7767,
417
+ "num_input_tokens_seen": 83081932,
418
+ "step": 25000
419
+ },
420
+ {
421
+ "epoch": 1.2755740083037366,
422
+ "grad_norm": 8.410615921020508,
423
+ "learning_rate": 3.724425991696264e-05,
424
+ "loss": 0.7665,
425
+ "num_input_tokens_seen": 84745948,
426
+ "step": 25500
427
+ },
428
+ {
429
+ "epoch": 1.3005852633685158,
430
+ "grad_norm": 7.881125450134277,
431
+ "learning_rate": 3.699414736631484e-05,
432
+ "loss": 0.7626,
433
+ "num_input_tokens_seen": 86421180,
434
+ "step": 26000
435
+ },
436
+ {
437
+ "epoch": 1.325596518433295,
438
+ "grad_norm": 21.633901596069336,
439
+ "learning_rate": 3.674403481566705e-05,
440
+ "loss": 0.7645,
441
+ "num_input_tokens_seen": 88075204,
442
+ "step": 26500
443
+ },
444
+ {
445
+ "epoch": 1.3506077734980741,
446
+ "grad_norm": 14.725602149963379,
447
+ "learning_rate": 3.649392226501926e-05,
448
+ "loss": 0.751,
449
+ "num_input_tokens_seen": 89740116,
450
+ "step": 27000
451
+ },
452
+ {
453
+ "epoch": 1.3756190285628533,
454
+ "grad_norm": 6.119060039520264,
455
+ "learning_rate": 3.6243809714371465e-05,
456
+ "loss": 0.756,
457
+ "num_input_tokens_seen": 91410556,
458
+ "step": 27500
459
+ },
460
+ {
461
+ "epoch": 1.4006302836276325,
462
+ "grad_norm": 6.520070552825928,
463
+ "learning_rate": 3.5993697163723675e-05,
464
+ "loss": 0.7526,
465
+ "num_input_tokens_seen": 93116396,
466
+ "step": 28000
467
+ },
468
+ {
469
+ "epoch": 1.4256415386924115,
470
+ "grad_norm": 7.963521480560303,
471
+ "learning_rate": 3.5743584613075885e-05,
472
+ "loss": 0.7645,
473
+ "num_input_tokens_seen": 94761716,
474
+ "step": 28500
475
+ },
476
+ {
477
+ "epoch": 1.4506527937571907,
478
+ "grad_norm": 11.38167953491211,
479
+ "learning_rate": 3.5493472062428095e-05,
480
+ "loss": 0.7624,
481
+ "num_input_tokens_seen": 96449700,
482
+ "step": 29000
483
+ },
484
+ {
485
+ "epoch": 1.4756640488219699,
486
+ "grad_norm": 15.715912818908691,
487
+ "learning_rate": 3.5243359511780305e-05,
488
+ "loss": 0.7509,
489
+ "num_input_tokens_seen": 98102252,
490
+ "step": 29500
491
+ },
492
+ {
493
+ "epoch": 1.500675303886749,
494
+ "grad_norm": 7.735713005065918,
495
+ "learning_rate": 3.499324696113251e-05,
496
+ "loss": 0.7738,
497
+ "num_input_tokens_seen": 99780396,
498
+ "step": 30000
499
+ },
500
+ {
501
+ "epoch": 1.525686558951528,
502
+ "grad_norm": 8.079352378845215,
503
+ "learning_rate": 3.474313441048472e-05,
504
+ "loss": 0.7522,
505
+ "num_input_tokens_seen": 101479956,
506
+ "step": 30500
507
+ },
508
+ {
509
+ "epoch": 1.5506978140163072,
510
+ "grad_norm": 8.290655136108398,
511
+ "learning_rate": 3.449302185983693e-05,
512
+ "loss": 0.7381,
513
+ "num_input_tokens_seen": 103149500,
514
+ "step": 31000
515
+ },
516
+ {
517
+ "epoch": 1.5757090690810864,
518
+ "grad_norm": 8.904264450073242,
519
+ "learning_rate": 3.424290930918914e-05,
520
+ "loss": 0.7467,
521
+ "num_input_tokens_seen": 104812996,
522
+ "step": 31500
523
+ },
524
+ {
525
+ "epoch": 1.6007203241458656,
526
+ "grad_norm": 7.439008712768555,
527
+ "learning_rate": 3.399279675854135e-05,
528
+ "loss": 0.7507,
529
+ "num_input_tokens_seen": 106479036,
530
+ "step": 32000
531
+ },
532
+ {
533
+ "epoch": 1.6257315792106448,
534
+ "grad_norm": 7.584664344787598,
535
+ "learning_rate": 3.374268420789355e-05,
536
+ "loss": 0.7168,
537
+ "num_input_tokens_seen": 108141364,
538
+ "step": 32500
539
+ },
540
+ {
541
+ "epoch": 1.650742834275424,
542
+ "grad_norm": 8.953302383422852,
543
+ "learning_rate": 3.349257165724576e-05,
544
+ "loss": 0.7469,
545
+ "num_input_tokens_seen": 109799916,
546
+ "step": 33000
547
+ },
548
+ {
549
+ "epoch": 1.6757540893402032,
550
+ "grad_norm": 10.678362846374512,
551
+ "learning_rate": 3.324245910659797e-05,
552
+ "loss": 0.7468,
553
+ "num_input_tokens_seen": 111436748,
554
+ "step": 33500
555
+ },
556
+ {
557
+ "epoch": 1.7007653444049824,
558
+ "grad_norm": 11.628217697143555,
559
+ "learning_rate": 3.2992346555950175e-05,
560
+ "loss": 0.7358,
561
+ "num_input_tokens_seen": 113068476,
562
+ "step": 34000
563
+ },
564
+ {
565
+ "epoch": 1.7257765994697614,
566
+ "grad_norm": 12.741203308105469,
567
+ "learning_rate": 3.2742234005302385e-05,
568
+ "loss": 0.7402,
569
+ "num_input_tokens_seen": 114748860,
570
+ "step": 34500
571
+ },
572
+ {
573
+ "epoch": 1.7507878545345406,
574
+ "grad_norm": 9.066828727722168,
575
+ "learning_rate": 3.2492121454654595e-05,
576
+ "loss": 0.7728,
577
+ "num_input_tokens_seen": 116441684,
578
+ "step": 35000
579
+ },
580
+ {
581
+ "epoch": 1.7757991095993197,
582
+ "grad_norm": 7.780086517333984,
583
+ "learning_rate": 3.2242008904006805e-05,
584
+ "loss": 0.7424,
585
+ "num_input_tokens_seen": 118093652,
586
+ "step": 35500
587
+ },
588
+ {
589
+ "epoch": 1.8008103646640987,
590
+ "grad_norm": 5.290003299713135,
591
+ "learning_rate": 3.1991896353359015e-05,
592
+ "loss": 0.7121,
593
+ "num_input_tokens_seen": 119756772,
594
+ "step": 36000
595
+ },
596
+ {
597
+ "epoch": 1.825821619728878,
598
+ "grad_norm": 13.356730461120605,
599
+ "learning_rate": 3.1741783802711225e-05,
600
+ "loss": 0.789,
601
+ "num_input_tokens_seen": 121419852,
602
+ "step": 36500
603
+ },
604
+ {
605
+ "epoch": 1.850832874793657,
606
+ "grad_norm": 4.2140727043151855,
607
+ "learning_rate": 3.149167125206343e-05,
608
+ "loss": 0.7501,
609
+ "num_input_tokens_seen": 123080420,
610
+ "step": 37000
611
+ },
612
+ {
613
+ "epoch": 1.8758441298584363,
614
+ "grad_norm": 15.408193588256836,
615
+ "learning_rate": 3.124155870141564e-05,
616
+ "loss": 0.7576,
617
+ "num_input_tokens_seen": 124733724,
618
+ "step": 37500
619
+ },
620
+ {
621
+ "epoch": 1.9008553849232155,
622
+ "grad_norm": 8.88025951385498,
623
+ "learning_rate": 3.099144615076784e-05,
624
+ "loss": 0.7315,
625
+ "num_input_tokens_seen": 126386636,
626
+ "step": 38000
627
+ },
628
+ {
629
+ "epoch": 1.9258666399879947,
630
+ "grad_norm": 15.850674629211426,
631
+ "learning_rate": 3.074133360012005e-05,
632
+ "loss": 0.7289,
633
+ "num_input_tokens_seen": 128054932,
634
+ "step": 38500
635
+ },
636
+ {
637
+ "epoch": 1.9508778950527739,
638
+ "grad_norm": 10.460667610168457,
639
+ "learning_rate": 3.049122104947226e-05,
640
+ "loss": 0.7375,
641
+ "num_input_tokens_seen": 129731780,
642
+ "step": 39000
643
+ },
644
+ {
645
+ "epoch": 1.975889150117553,
646
+ "grad_norm": 4.816532135009766,
647
+ "learning_rate": 3.024110849882447e-05,
648
+ "loss": 0.7203,
649
+ "num_input_tokens_seen": 131377564,
650
+ "step": 39500
651
+ },
652
+ {
653
+ "epoch": 2.0,
654
+ "eval_loss": 0.5678554773330688,
655
+ "eval_runtime": 97.2769,
656
+ "eval_samples_per_second": 411.002,
657
+ "eval_steps_per_second": 51.379,
658
+ "num_input_tokens_seen": 132976438,
659
+ "step": 39982
660
+ },
661
+ {
662
+ "epoch": 2.0009004051823323,
663
+ "grad_norm": 8.531465530395508,
664
+ "learning_rate": 2.999099594817668e-05,
665
+ "loss": 0.7337,
666
+ "num_input_tokens_seen": 133038726,
667
+ "step": 40000
668
+ },
669
+ {
670
+ "epoch": 2.025911660247111,
671
+ "grad_norm": 17.74102783203125,
672
+ "learning_rate": 2.974088339752889e-05,
673
+ "loss": 0.6798,
674
+ "num_input_tokens_seen": 134681590,
675
+ "step": 40500
676
+ },
677
+ {
678
+ "epoch": 2.05092291531189,
679
+ "grad_norm": 16.203670501708984,
680
+ "learning_rate": 2.9490770846881098e-05,
681
+ "loss": 0.692,
682
+ "num_input_tokens_seen": 136354910,
683
+ "step": 41000
684
+ },
685
+ {
686
+ "epoch": 2.0759341703766694,
687
+ "grad_norm": 11.238871574401855,
688
+ "learning_rate": 2.9240658296233308e-05,
689
+ "loss": 0.653,
690
+ "num_input_tokens_seen": 138014246,
691
+ "step": 41500
692
+ },
693
+ {
694
+ "epoch": 2.1009454254414486,
695
+ "grad_norm": 8.781373023986816,
696
+ "learning_rate": 2.899054574558551e-05,
697
+ "loss": 0.6742,
698
+ "num_input_tokens_seen": 139676526,
699
+ "step": 42000
700
+ },
701
+ {
702
+ "epoch": 2.1259566805062278,
703
+ "grad_norm": 7.73007869720459,
704
+ "learning_rate": 2.874043319493772e-05,
705
+ "loss": 0.6739,
706
+ "num_input_tokens_seen": 141326846,
707
+ "step": 42500
708
+ },
709
+ {
710
+ "epoch": 2.150967935571007,
711
+ "grad_norm": 6.6758904457092285,
712
+ "learning_rate": 2.849032064428993e-05,
713
+ "loss": 0.6767,
714
+ "num_input_tokens_seen": 142999126,
715
+ "step": 43000
716
+ },
717
+ {
718
+ "epoch": 2.175979190635786,
719
+ "grad_norm": 9.964508056640625,
720
+ "learning_rate": 2.824020809364214e-05,
721
+ "loss": 0.6649,
722
+ "num_input_tokens_seen": 144643454,
723
+ "step": 43500
724
+ },
725
+ {
726
+ "epoch": 2.2009904457005653,
727
+ "grad_norm": 7.9148664474487305,
728
+ "learning_rate": 2.7990095542994348e-05,
729
+ "loss": 0.678,
730
+ "num_input_tokens_seen": 146327686,
731
+ "step": 44000
732
+ },
733
+ {
734
+ "epoch": 2.2260017007653445,
735
+ "grad_norm": 5.838576316833496,
736
+ "learning_rate": 2.7739982992346558e-05,
737
+ "loss": 0.6629,
738
+ "num_input_tokens_seen": 147996750,
739
+ "step": 44500
740
+ },
741
+ {
742
+ "epoch": 2.2510129558301237,
743
+ "grad_norm": 9.018148422241211,
744
+ "learning_rate": 2.7489870441698768e-05,
745
+ "loss": 0.6673,
746
+ "num_input_tokens_seen": 149658382,
747
+ "step": 45000
748
+ },
749
+ {
750
+ "epoch": 2.276024210894903,
751
+ "grad_norm": 5.56981897354126,
752
+ "learning_rate": 2.7239757891050978e-05,
753
+ "loss": 0.658,
754
+ "num_input_tokens_seen": 151279470,
755
+ "step": 45500
756
+ },
757
+ {
758
+ "epoch": 2.3010354659596817,
759
+ "grad_norm": 3.9373059272766113,
760
+ "learning_rate": 2.698964534040318e-05,
761
+ "loss": 0.6747,
762
+ "num_input_tokens_seen": 152950878,
763
+ "step": 46000
764
+ },
765
+ {
766
+ "epoch": 2.326046721024461,
767
+ "grad_norm": 7.596631050109863,
768
+ "learning_rate": 2.6739532789755388e-05,
769
+ "loss": 0.6824,
770
+ "num_input_tokens_seen": 154603110,
771
+ "step": 46500
772
+ },
773
+ {
774
+ "epoch": 2.35105797608924,
775
+ "grad_norm": 7.714618682861328,
776
+ "learning_rate": 2.6489420239107598e-05,
777
+ "loss": 0.6662,
778
+ "num_input_tokens_seen": 156262254,
779
+ "step": 47000
780
+ },
781
+ {
782
+ "epoch": 2.3760692311540192,
783
+ "grad_norm": 11.400321006774902,
784
+ "learning_rate": 2.6239307688459808e-05,
785
+ "loss": 0.6478,
786
+ "num_input_tokens_seen": 157940526,
787
+ "step": 47500
788
+ },
789
+ {
790
+ "epoch": 2.4010804862187984,
791
+ "grad_norm": 5.944780349731445,
792
+ "learning_rate": 2.5989195137812018e-05,
793
+ "loss": 0.6701,
794
+ "num_input_tokens_seen": 159597926,
795
+ "step": 48000
796
+ },
797
+ {
798
+ "epoch": 2.4260917412835776,
799
+ "grad_norm": 7.971735954284668,
800
+ "learning_rate": 2.5739082587164225e-05,
801
+ "loss": 0.6815,
802
+ "num_input_tokens_seen": 161249054,
803
+ "step": 48500
804
+ },
805
+ {
806
+ "epoch": 2.451102996348357,
807
+ "grad_norm": 8.019645690917969,
808
+ "learning_rate": 2.5488970036516435e-05,
809
+ "loss": 0.6823,
810
+ "num_input_tokens_seen": 162937710,
811
+ "step": 49000
812
+ },
813
+ {
814
+ "epoch": 2.476114251413136,
815
+ "grad_norm": 14.52238655090332,
816
+ "learning_rate": 2.5238857485868645e-05,
817
+ "loss": 0.6662,
818
+ "num_input_tokens_seen": 164579550,
819
+ "step": 49500
820
+ },
821
+ {
822
+ "epoch": 2.501125506477915,
823
+ "grad_norm": 8.065009117126465,
824
+ "learning_rate": 2.498874493522085e-05,
825
+ "loss": 0.6855,
826
+ "num_input_tokens_seen": 166259486,
827
+ "step": 50000
828
+ },
829
+ {
830
+ "epoch": 2.526136761542694,
831
+ "grad_norm": 3.0121171474456787,
832
+ "learning_rate": 2.473863238457306e-05,
833
+ "loss": 0.6597,
834
+ "num_input_tokens_seen": 167925014,
835
+ "step": 50500
836
+ },
837
+ {
838
+ "epoch": 2.551148016607473,
839
+ "grad_norm": 9.93840217590332,
840
+ "learning_rate": 2.4488519833925268e-05,
841
+ "loss": 0.6672,
842
+ "num_input_tokens_seen": 169584230,
843
+ "step": 51000
844
+ },
845
+ {
846
+ "epoch": 2.5761592716722523,
847
+ "grad_norm": 7.8001627922058105,
848
+ "learning_rate": 2.4238407283277475e-05,
849
+ "loss": 0.6419,
850
+ "num_input_tokens_seen": 171205846,
851
+ "step": 51500
852
+ },
853
+ {
854
+ "epoch": 2.6011705267370315,
855
+ "grad_norm": 5.621837139129639,
856
+ "learning_rate": 2.3988294732629685e-05,
857
+ "loss": 0.6679,
858
+ "num_input_tokens_seen": 172867766,
859
+ "step": 52000
860
+ },
861
+ {
862
+ "epoch": 2.6261817818018107,
863
+ "grad_norm": 18.287431716918945,
864
+ "learning_rate": 2.3738182181981895e-05,
865
+ "loss": 0.6601,
866
+ "num_input_tokens_seen": 174508502,
867
+ "step": 52500
868
+ },
869
+ {
870
+ "epoch": 2.65119303686659,
871
+ "grad_norm": 7.687650203704834,
872
+ "learning_rate": 2.34880696313341e-05,
873
+ "loss": 0.6722,
874
+ "num_input_tokens_seen": 176179174,
875
+ "step": 53000
876
+ },
877
+ {
878
+ "epoch": 2.676204291931369,
879
+ "grad_norm": 9.807682037353516,
880
+ "learning_rate": 2.3237957080686308e-05,
881
+ "loss": 0.666,
882
+ "num_input_tokens_seen": 177874198,
883
+ "step": 53500
884
+ },
885
+ {
886
+ "epoch": 2.7012155469961483,
887
+ "grad_norm": 9.2701416015625,
888
+ "learning_rate": 2.2987844530038518e-05,
889
+ "loss": 0.6811,
890
+ "num_input_tokens_seen": 179531678,
891
+ "step": 54000
892
+ },
893
+ {
894
+ "epoch": 2.7262268020609275,
895
+ "grad_norm": 8.37064266204834,
896
+ "learning_rate": 2.2737731979390728e-05,
897
+ "loss": 0.6505,
898
+ "num_input_tokens_seen": 181197542,
899
+ "step": 54500
900
+ },
901
+ {
902
+ "epoch": 2.7512380571257067,
903
+ "grad_norm": 5.556591033935547,
904
+ "learning_rate": 2.2487619428742935e-05,
905
+ "loss": 0.6711,
906
+ "num_input_tokens_seen": 182849270,
907
+ "step": 55000
908
+ },
909
+ {
910
+ "epoch": 2.776249312190486,
911
+ "grad_norm": 7.93866491317749,
912
+ "learning_rate": 2.2237506878095145e-05,
913
+ "loss": 0.6664,
914
+ "num_input_tokens_seen": 184520526,
915
+ "step": 55500
916
+ },
917
+ {
918
+ "epoch": 2.801260567255265,
919
+ "grad_norm": 6.768641471862793,
920
+ "learning_rate": 2.198739432744735e-05,
921
+ "loss": 0.6699,
922
+ "num_input_tokens_seen": 186239974,
923
+ "step": 56000
924
+ },
925
+ {
926
+ "epoch": 2.8262718223200443,
927
+ "grad_norm": 5.911066055297852,
928
+ "learning_rate": 2.173728177679956e-05,
929
+ "loss": 0.6649,
930
+ "num_input_tokens_seen": 187875982,
931
+ "step": 56500
932
+ },
933
+ {
934
+ "epoch": 2.851283077384823,
935
+ "grad_norm": 9.964897155761719,
936
+ "learning_rate": 2.1487169226151768e-05,
937
+ "loss": 0.6874,
938
+ "num_input_tokens_seen": 189505118,
939
+ "step": 57000
940
+ },
941
+ {
942
+ "epoch": 2.876294332449602,
943
+ "grad_norm": 8.109452247619629,
944
+ "learning_rate": 2.1237056675503978e-05,
945
+ "loss": 0.6762,
946
+ "num_input_tokens_seen": 191184886,
947
+ "step": 57500
948
+ },
949
+ {
950
+ "epoch": 2.9013055875143814,
951
+ "grad_norm": 8.556594848632812,
952
+ "learning_rate": 2.0986944124856188e-05,
953
+ "loss": 0.6491,
954
+ "num_input_tokens_seen": 192859070,
955
+ "step": 58000
956
+ },
957
+ {
958
+ "epoch": 2.9263168425791606,
959
+ "grad_norm": 5.430099010467529,
960
+ "learning_rate": 2.0736831574208394e-05,
961
+ "loss": 0.661,
962
+ "num_input_tokens_seen": 194533102,
963
+ "step": 58500
964
+ },
965
+ {
966
+ "epoch": 2.9513280976439398,
967
+ "grad_norm": 9.806259155273438,
968
+ "learning_rate": 2.04867190235606e-05,
969
+ "loss": 0.645,
970
+ "num_input_tokens_seen": 196171870,
971
+ "step": 59000
972
+ },
973
+ {
974
+ "epoch": 2.976339352708719,
975
+ "grad_norm": 8.950848579406738,
976
+ "learning_rate": 2.023660647291281e-05,
977
+ "loss": 0.6479,
978
+ "num_input_tokens_seen": 197877830,
979
+ "step": 59500
980
+ },
981
+ {
982
+ "epoch": 3.0,
983
+ "eval_loss": 0.560497522354126,
984
+ "eval_runtime": 98.7193,
985
+ "eval_samples_per_second": 404.997,
986
+ "eval_steps_per_second": 50.628,
987
+ "num_input_tokens_seen": 199402582,
988
+ "step": 59973
989
+ },
990
+ {
991
+ "epoch": 3.001350607773498,
992
+ "grad_norm": 6.855441093444824,
993
+ "learning_rate": 1.998649392226502e-05,
994
+ "loss": 0.6187,
995
+ "num_input_tokens_seen": 199490934,
996
+ "step": 60000
997
+ },
998
+ {
999
+ "epoch": 3.0263618628382774,
1000
+ "grad_norm": 8.57907772064209,
1001
+ "learning_rate": 1.973638137161723e-05,
1002
+ "loss": 0.652,
1003
+ "num_input_tokens_seen": 201164870,
1004
+ "step": 60500
1005
+ },
1006
+ {
1007
+ "epoch": 3.0513731179030565,
1008
+ "grad_norm": 15.578742027282715,
1009
+ "learning_rate": 1.9486268820969438e-05,
1010
+ "loss": 0.6127,
1011
+ "num_input_tokens_seen": 202824222,
1012
+ "step": 61000
1013
+ },
1014
+ {
1015
+ "epoch": 3.0763843729678357,
1016
+ "grad_norm": 9.083669662475586,
1017
+ "learning_rate": 1.9236156270321644e-05,
1018
+ "loss": 0.6146,
1019
+ "num_input_tokens_seen": 204495334,
1020
+ "step": 61500
1021
+ },
1022
+ {
1023
+ "epoch": 3.1013956280326145,
1024
+ "grad_norm": 10.12027359008789,
1025
+ "learning_rate": 1.8986043719673854e-05,
1026
+ "loss": 0.6341,
1027
+ "num_input_tokens_seen": 206136214,
1028
+ "step": 62000
1029
+ },
1030
+ {
1031
+ "epoch": 3.1264068830973937,
1032
+ "grad_norm": 10.482580184936523,
1033
+ "learning_rate": 1.8735931169026064e-05,
1034
+ "loss": 0.603,
1035
+ "num_input_tokens_seen": 207809294,
1036
+ "step": 62500
1037
+ },
1038
+ {
1039
+ "epoch": 3.151418138162173,
1040
+ "grad_norm": 7.722796440124512,
1041
+ "learning_rate": 1.848581861837827e-05,
1042
+ "loss": 0.6184,
1043
+ "num_input_tokens_seen": 209485534,
1044
+ "step": 63000
1045
+ },
1046
+ {
1047
+ "epoch": 3.176429393226952,
1048
+ "grad_norm": 7.449066162109375,
1049
+ "learning_rate": 1.8235706067730478e-05,
1050
+ "loss": 0.621,
1051
+ "num_input_tokens_seen": 211143158,
1052
+ "step": 63500
1053
+ },
1054
+ {
1055
+ "epoch": 3.2014406482917313,
1056
+ "grad_norm": 8.766199111938477,
1057
+ "learning_rate": 1.7985593517082688e-05,
1058
+ "loss": 0.6165,
1059
+ "num_input_tokens_seen": 212777414,
1060
+ "step": 64000
1061
+ },
1062
+ {
1063
+ "epoch": 3.2264519033565104,
1064
+ "grad_norm": 4.193557262420654,
1065
+ "learning_rate": 1.7735480966434898e-05,
1066
+ "loss": 0.6188,
1067
+ "num_input_tokens_seen": 214420798,
1068
+ "step": 64500
1069
+ },
1070
+ {
1071
+ "epoch": 3.2514631584212896,
1072
+ "grad_norm": 6.699706554412842,
1073
+ "learning_rate": 1.7485368415787104e-05,
1074
+ "loss": 0.6095,
1075
+ "num_input_tokens_seen": 216073590,
1076
+ "step": 65000
1077
+ },
1078
+ {
1079
+ "epoch": 3.276474413486069,
1080
+ "grad_norm": 8.79476547241211,
1081
+ "learning_rate": 1.7235255865139314e-05,
1082
+ "loss": 0.6208,
1083
+ "num_input_tokens_seen": 217746214,
1084
+ "step": 65500
1085
+ },
1086
+ {
1087
+ "epoch": 3.301485668550848,
1088
+ "grad_norm": 6.685282230377197,
1089
+ "learning_rate": 1.698514331449152e-05,
1090
+ "loss": 0.6058,
1091
+ "num_input_tokens_seen": 219433446,
1092
+ "step": 66000
1093
+ },
1094
+ {
1095
+ "epoch": 3.326496923615627,
1096
+ "grad_norm": 10.743680953979492,
1097
+ "learning_rate": 1.673503076384373e-05,
1098
+ "loss": 0.6318,
1099
+ "num_input_tokens_seen": 221089694,
1100
+ "step": 66500
1101
+ },
1102
+ {
1103
+ "epoch": 3.3515081786804064,
1104
+ "grad_norm": 8.36410903930664,
1105
+ "learning_rate": 1.6484918213195938e-05,
1106
+ "loss": 0.6236,
1107
+ "num_input_tokens_seen": 222760502,
1108
+ "step": 67000
1109
+ },
1110
+ {
1111
+ "epoch": 3.376519433745185,
1112
+ "grad_norm": 7.1238274574279785,
1113
+ "learning_rate": 1.6234805662548148e-05,
1114
+ "loss": 0.6103,
1115
+ "num_input_tokens_seen": 224417582,
1116
+ "step": 67500
1117
+ },
1118
+ {
1119
+ "epoch": 3.4015306888099643,
1120
+ "grad_norm": 7.042121887207031,
1121
+ "learning_rate": 1.5984693111900358e-05,
1122
+ "loss": 0.6157,
1123
+ "num_input_tokens_seen": 226068982,
1124
+ "step": 68000
1125
+ },
1126
+ {
1127
+ "epoch": 3.4265419438747435,
1128
+ "grad_norm": 9.31881332397461,
1129
+ "learning_rate": 1.5734580561252564e-05,
1130
+ "loss": 0.6263,
1131
+ "num_input_tokens_seen": 227701038,
1132
+ "step": 68500
1133
+ },
1134
+ {
1135
+ "epoch": 3.4515531989395227,
1136
+ "grad_norm": 7.049442768096924,
1137
+ "learning_rate": 1.548446801060477e-05,
1138
+ "loss": 0.6237,
1139
+ "num_input_tokens_seen": 229359710,
1140
+ "step": 69000
1141
+ },
1142
+ {
1143
+ "epoch": 3.476564454004302,
1144
+ "grad_norm": 7.746445178985596,
1145
+ "learning_rate": 1.5234355459956981e-05,
1146
+ "loss": 0.6376,
1147
+ "num_input_tokens_seen": 231028950,
1148
+ "step": 69500
1149
+ },
1150
+ {
1151
+ "epoch": 3.501575709069081,
1152
+ "grad_norm": 4.588512420654297,
1153
+ "learning_rate": 1.4984242909309191e-05,
1154
+ "loss": 0.6189,
1155
+ "num_input_tokens_seen": 232663446,
1156
+ "step": 70000
1157
+ },
1158
+ {
1159
+ "epoch": 3.5265869641338603,
1160
+ "grad_norm": 9.873016357421875,
1161
+ "learning_rate": 1.47341303586614e-05,
1162
+ "loss": 0.5935,
1163
+ "num_input_tokens_seen": 234333558,
1164
+ "step": 70500
1165
+ },
1166
+ {
1167
+ "epoch": 3.5515982191986395,
1168
+ "grad_norm": 8.153191566467285,
1169
+ "learning_rate": 1.4484017808013606e-05,
1170
+ "loss": 0.6403,
1171
+ "num_input_tokens_seen": 236006758,
1172
+ "step": 71000
1173
+ },
1174
+ {
1175
+ "epoch": 3.5766094742634187,
1176
+ "grad_norm": 5.909561634063721,
1177
+ "learning_rate": 1.4233905257365814e-05,
1178
+ "loss": 0.6152,
1179
+ "num_input_tokens_seen": 237655630,
1180
+ "step": 71500
1181
+ },
1182
+ {
1183
+ "epoch": 3.6016207293281974,
1184
+ "grad_norm": 9.481532096862793,
1185
+ "learning_rate": 1.3983792706718024e-05,
1186
+ "loss": 0.5916,
1187
+ "num_input_tokens_seen": 239300238,
1188
+ "step": 72000
1189
+ },
1190
+ {
1191
+ "epoch": 3.6266319843929766,
1192
+ "grad_norm": 4.988440990447998,
1193
+ "learning_rate": 1.3733680156070232e-05,
1194
+ "loss": 0.6275,
1195
+ "num_input_tokens_seen": 240971214,
1196
+ "step": 72500
1197
+ },
1198
+ {
1199
+ "epoch": 3.651643239457756,
1200
+ "grad_norm": 6.159299850463867,
1201
+ "learning_rate": 1.3483567605422439e-05,
1202
+ "loss": 0.6101,
1203
+ "num_input_tokens_seen": 242634286,
1204
+ "step": 73000
1205
+ },
1206
+ {
1207
+ "epoch": 3.676654494522535,
1208
+ "grad_norm": 4.264859199523926,
1209
+ "learning_rate": 1.3233455054774649e-05,
1210
+ "loss": 0.6045,
1211
+ "num_input_tokens_seen": 244293870,
1212
+ "step": 73500
1213
+ },
1214
+ {
1215
+ "epoch": 3.701665749587314,
1216
+ "grad_norm": 5.82095193862915,
1217
+ "learning_rate": 1.2983342504126857e-05,
1218
+ "loss": 0.624,
1219
+ "num_input_tokens_seen": 245956374,
1220
+ "step": 74000
1221
+ },
1222
+ {
1223
+ "epoch": 3.7266770046520934,
1224
+ "grad_norm": 10.4242525100708,
1225
+ "learning_rate": 1.2733229953479067e-05,
1226
+ "loss": 0.6231,
1227
+ "num_input_tokens_seen": 247566166,
1228
+ "step": 74500
1229
+ },
1230
+ {
1231
+ "epoch": 3.7516882597168726,
1232
+ "grad_norm": 6.536423206329346,
1233
+ "learning_rate": 1.2483117402831276e-05,
1234
+ "loss": 0.6159,
1235
+ "num_input_tokens_seen": 249233118,
1236
+ "step": 75000
1237
+ },
1238
+ {
1239
+ "epoch": 3.776699514781652,
1240
+ "grad_norm": 10.467476844787598,
1241
+ "learning_rate": 1.2233004852183482e-05,
1242
+ "loss": 0.6252,
1243
+ "num_input_tokens_seen": 250919822,
1244
+ "step": 75500
1245
+ },
1246
+ {
1247
+ "epoch": 3.801710769846431,
1248
+ "grad_norm": 13.297423362731934,
1249
+ "learning_rate": 1.1982892301535692e-05,
1250
+ "loss": 0.6133,
1251
+ "num_input_tokens_seen": 252600838,
1252
+ "step": 76000
1253
+ },
1254
+ {
1255
+ "epoch": 3.82672202491121,
1256
+ "grad_norm": 6.729821681976318,
1257
+ "learning_rate": 1.1732779750887899e-05,
1258
+ "loss": 0.6201,
1259
+ "num_input_tokens_seen": 254292558,
1260
+ "step": 76500
1261
+ },
1262
+ {
1263
+ "epoch": 3.8517332799759894,
1264
+ "grad_norm": 5.975412845611572,
1265
+ "learning_rate": 1.1482667200240109e-05,
1266
+ "loss": 0.5976,
1267
+ "num_input_tokens_seen": 255961510,
1268
+ "step": 77000
1269
+ },
1270
+ {
1271
+ "epoch": 3.8767445350407685,
1272
+ "grad_norm": 16.30948257446289,
1273
+ "learning_rate": 1.1232554649592317e-05,
1274
+ "loss": 0.6023,
1275
+ "num_input_tokens_seen": 257630246,
1276
+ "step": 77500
1277
+ },
1278
+ {
1279
+ "epoch": 3.9017557901055477,
1280
+ "grad_norm": 7.327265739440918,
1281
+ "learning_rate": 1.0982442098944526e-05,
1282
+ "loss": 0.6145,
1283
+ "num_input_tokens_seen": 259305118,
1284
+ "step": 78000
1285
+ },
1286
+ {
1287
+ "epoch": 3.9267670451703265,
1288
+ "grad_norm": 12.45727825164795,
1289
+ "learning_rate": 1.0732329548296734e-05,
1290
+ "loss": 0.6311,
1291
+ "num_input_tokens_seen": 260978934,
1292
+ "step": 78500
1293
+ },
1294
+ {
1295
+ "epoch": 3.9517783002351057,
1296
+ "grad_norm": 10.317325592041016,
1297
+ "learning_rate": 1.0482216997648942e-05,
1298
+ "loss": 0.6346,
1299
+ "num_input_tokens_seen": 262670814,
1300
+ "step": 79000
1301
+ },
1302
+ {
1303
+ "epoch": 3.976789555299885,
1304
+ "grad_norm": 7.8411149978637695,
1305
+ "learning_rate": 1.023210444700115e-05,
1306
+ "loss": 0.6023,
1307
+ "num_input_tokens_seen": 264314614,
1308
+ "step": 79500
1309
+ },
1310
+ {
1311
+ "epoch": 4.0,
1312
+ "eval_loss": 0.54269939661026,
1313
+ "eval_runtime": 96.9182,
1314
+ "eval_samples_per_second": 412.523,
1315
+ "eval_steps_per_second": 51.569,
1316
+ "num_input_tokens_seen": 265875340,
1317
+ "step": 79964
1318
+ },
1319
+ {
1320
+ "epoch": 4.0018008103646645,
1321
+ "grad_norm": 6.620047569274902,
1322
+ "learning_rate": 9.98199189635336e-06,
1323
+ "loss": 0.6268,
1324
+ "num_input_tokens_seen": 266010940,
1325
+ "step": 80000
1326
+ },
1327
+ {
1328
+ "epoch": 4.026812065429443,
1329
+ "grad_norm": 10.007366180419922,
1330
+ "learning_rate": 9.731879345705567e-06,
1331
+ "loss": 0.5924,
1332
+ "num_input_tokens_seen": 267660364,
1333
+ "step": 80500
1334
+ },
1335
+ {
1336
+ "epoch": 4.051823320494222,
1337
+ "grad_norm": 6.680395603179932,
1338
+ "learning_rate": 9.481766795057777e-06,
1339
+ "loss": 0.5786,
1340
+ "num_input_tokens_seen": 269338492,
1341
+ "step": 81000
1342
+ },
1343
+ {
1344
+ "epoch": 4.076834575559001,
1345
+ "grad_norm": 4.809377670288086,
1346
+ "learning_rate": 9.231654244409984e-06,
1347
+ "loss": 0.5942,
1348
+ "num_input_tokens_seen": 271024236,
1349
+ "step": 81500
1350
+ },
1351
+ {
1352
+ "epoch": 4.10184583062378,
1353
+ "grad_norm": 8.463695526123047,
1354
+ "learning_rate": 8.981541693762194e-06,
1355
+ "loss": 0.5796,
1356
+ "num_input_tokens_seen": 272672620,
1357
+ "step": 82000
1358
+ },
1359
+ {
1360
+ "epoch": 4.12685708568856,
1361
+ "grad_norm": 10.12741470336914,
1362
+ "learning_rate": 8.731429143114402e-06,
1363
+ "loss": 0.5879,
1364
+ "num_input_tokens_seen": 274353676,
1365
+ "step": 82500
1366
+ },
1367
+ {
1368
+ "epoch": 4.151868340753339,
1369
+ "grad_norm": 15.428593635559082,
1370
+ "learning_rate": 8.48131659246661e-06,
1371
+ "loss": 0.5977,
1372
+ "num_input_tokens_seen": 275998164,
1373
+ "step": 83000
1374
+ },
1375
+ {
1376
+ "epoch": 4.176879595818118,
1377
+ "grad_norm": 10.350814819335938,
1378
+ "learning_rate": 8.231204041818819e-06,
1379
+ "loss": 0.566,
1380
+ "num_input_tokens_seen": 277685356,
1381
+ "step": 83500
1382
+ },
1383
+ {
1384
+ "epoch": 4.201890850882897,
1385
+ "grad_norm": 11.962939262390137,
1386
+ "learning_rate": 7.981091491171027e-06,
1387
+ "loss": 0.5671,
1388
+ "num_input_tokens_seen": 279358548,
1389
+ "step": 84000
1390
+ },
1391
+ {
1392
+ "epoch": 4.226902105947676,
1393
+ "grad_norm": 10.32712459564209,
1394
+ "learning_rate": 7.730978940523236e-06,
1395
+ "loss": 0.5785,
1396
+ "num_input_tokens_seen": 280991044,
1397
+ "step": 84500
1398
+ },
1399
+ {
1400
+ "epoch": 4.2519133610124555,
1401
+ "grad_norm": 5.896986484527588,
1402
+ "learning_rate": 7.480866389875445e-06,
1403
+ "loss": 0.6051,
1404
+ "num_input_tokens_seen": 282646764,
1405
+ "step": 85000
1406
+ },
1407
+ {
1408
+ "epoch": 4.276924616077235,
1409
+ "grad_norm": 7.187685966491699,
1410
+ "learning_rate": 7.230753839227652e-06,
1411
+ "loss": 0.5943,
1412
+ "num_input_tokens_seen": 284342508,
1413
+ "step": 85500
1414
+ },
1415
+ {
1416
+ "epoch": 4.301935871142014,
1417
+ "grad_norm": 6.680044174194336,
1418
+ "learning_rate": 6.980641288579861e-06,
1419
+ "loss": 0.5765,
1420
+ "num_input_tokens_seen": 286036340,
1421
+ "step": 86000
1422
+ },
1423
+ {
1424
+ "epoch": 4.326947126206793,
1425
+ "grad_norm": 4.963362693786621,
1426
+ "learning_rate": 6.73052873793207e-06,
1427
+ "loss": 0.6137,
1428
+ "num_input_tokens_seen": 287681564,
1429
+ "step": 86500
1430
+ },
1431
+ {
1432
+ "epoch": 4.351958381271572,
1433
+ "grad_norm": 12.112903594970703,
1434
+ "learning_rate": 6.480416187284279e-06,
1435
+ "loss": 0.5983,
1436
+ "num_input_tokens_seen": 289353828,
1437
+ "step": 87000
1438
+ },
1439
+ {
1440
+ "epoch": 4.3769696363363515,
1441
+ "grad_norm": 5.938944339752197,
1442
+ "learning_rate": 6.230303636636486e-06,
1443
+ "loss": 0.6017,
1444
+ "num_input_tokens_seen": 291011668,
1445
+ "step": 87500
1446
+ },
1447
+ {
1448
+ "epoch": 4.401980891401131,
1449
+ "grad_norm": 4.485511302947998,
1450
+ "learning_rate": 5.980191085988695e-06,
1451
+ "loss": 0.5898,
1452
+ "num_input_tokens_seen": 292675092,
1453
+ "step": 88000
1454
+ },
1455
+ {
1456
+ "epoch": 4.42699214646591,
1457
+ "grad_norm": 9.15986442565918,
1458
+ "learning_rate": 5.730078535340903e-06,
1459
+ "loss": 0.5744,
1460
+ "num_input_tokens_seen": 294338212,
1461
+ "step": 88500
1462
+ },
1463
+ {
1464
+ "epoch": 4.452003401530689,
1465
+ "grad_norm": 3.6591997146606445,
1466
+ "learning_rate": 5.479965984693112e-06,
1467
+ "loss": 0.5948,
1468
+ "num_input_tokens_seen": 296004820,
1469
+ "step": 89000
1470
+ },
1471
+ {
1472
+ "epoch": 4.477014656595468,
1473
+ "grad_norm": 9.19853401184082,
1474
+ "learning_rate": 5.2298534340453205e-06,
1475
+ "loss": 0.5838,
1476
+ "num_input_tokens_seen": 297661964,
1477
+ "step": 89500
1478
+ },
1479
+ {
1480
+ "epoch": 4.5020259116602475,
1481
+ "grad_norm": 13.491796493530273,
1482
+ "learning_rate": 4.979740883397529e-06,
1483
+ "loss": 0.5726,
1484
+ "num_input_tokens_seen": 299312356,
1485
+ "step": 90000
1486
+ },
1487
+ {
1488
+ "epoch": 4.527037166725027,
1489
+ "grad_norm": 6.374147415161133,
1490
+ "learning_rate": 4.729628332749737e-06,
1491
+ "loss": 0.5728,
1492
+ "num_input_tokens_seen": 300978468,
1493
+ "step": 90500
1494
+ },
1495
+ {
1496
+ "epoch": 4.552048421789806,
1497
+ "grad_norm": 7.507421970367432,
1498
+ "learning_rate": 4.479515782101945e-06,
1499
+ "loss": 0.5903,
1500
+ "num_input_tokens_seen": 302639252,
1501
+ "step": 91000
1502
+ },
1503
+ {
1504
+ "epoch": 4.577059676854585,
1505
+ "grad_norm": 12.31728744506836,
1506
+ "learning_rate": 4.229403231454155e-06,
1507
+ "loss": 0.5916,
1508
+ "num_input_tokens_seen": 304289124,
1509
+ "step": 91500
1510
+ },
1511
+ {
1512
+ "epoch": 4.602070931919363,
1513
+ "grad_norm": 11.238248825073242,
1514
+ "learning_rate": 3.979290680806363e-06,
1515
+ "loss": 0.5617,
1516
+ "num_input_tokens_seen": 305968436,
1517
+ "step": 92000
1518
+ },
1519
+ {
1520
+ "epoch": 4.6270821869841425,
1521
+ "grad_norm": 6.74647331237793,
1522
+ "learning_rate": 3.7291781301585712e-06,
1523
+ "loss": 0.6249,
1524
+ "num_input_tokens_seen": 307616156,
1525
+ "step": 92500
1526
+ },
1527
+ {
1528
+ "epoch": 4.652093442048922,
1529
+ "grad_norm": 7.845546722412109,
1530
+ "learning_rate": 3.4790655795107795e-06,
1531
+ "loss": 0.6015,
1532
+ "num_input_tokens_seen": 309294188,
1533
+ "step": 93000
1534
+ },
1535
+ {
1536
+ "epoch": 4.677104697113701,
1537
+ "grad_norm": 5.631568431854248,
1538
+ "learning_rate": 3.2289530288629883e-06,
1539
+ "loss": 0.5747,
1540
+ "num_input_tokens_seen": 310930388,
1541
+ "step": 93500
1542
+ },
1543
+ {
1544
+ "epoch": 4.70211595217848,
1545
+ "grad_norm": 4.305506229400635,
1546
+ "learning_rate": 2.978840478215197e-06,
1547
+ "loss": 0.5957,
1548
+ "num_input_tokens_seen": 312600876,
1549
+ "step": 94000
1550
+ },
1551
+ {
1552
+ "epoch": 4.727127207243259,
1553
+ "grad_norm": 12.092133522033691,
1554
+ "learning_rate": 2.7287279275674053e-06,
1555
+ "loss": 0.5952,
1556
+ "num_input_tokens_seen": 314275796,
1557
+ "step": 94500
1558
+ },
1559
+ {
1560
+ "epoch": 4.7521384623080385,
1561
+ "grad_norm": 7.043518543243408,
1562
+ "learning_rate": 2.478615376919614e-06,
1563
+ "loss": 0.6013,
1564
+ "num_input_tokens_seen": 315945468,
1565
+ "step": 95000
1566
+ },
1567
+ {
1568
+ "epoch": 4.777149717372818,
1569
+ "grad_norm": 6.208098888397217,
1570
+ "learning_rate": 2.2285028262718224e-06,
1571
+ "loss": 0.591,
1572
+ "num_input_tokens_seen": 317595388,
1573
+ "step": 95500
1574
+ },
1575
+ {
1576
+ "epoch": 4.802160972437597,
1577
+ "grad_norm": 3.588547706604004,
1578
+ "learning_rate": 1.978390275624031e-06,
1579
+ "loss": 0.5846,
1580
+ "num_input_tokens_seen": 319229212,
1581
+ "step": 96000
1582
+ },
1583
+ {
1584
+ "epoch": 4.827172227502376,
1585
+ "grad_norm": 10.502739906311035,
1586
+ "learning_rate": 1.7282777249762395e-06,
1587
+ "loss": 0.5904,
1588
+ "num_input_tokens_seen": 320908604,
1589
+ "step": 96500
1590
+ },
1591
+ {
1592
+ "epoch": 4.852183482567155,
1593
+ "grad_norm": 8.170723915100098,
1594
+ "learning_rate": 1.4781651743284478e-06,
1595
+ "loss": 0.5925,
1596
+ "num_input_tokens_seen": 322558268,
1597
+ "step": 97000
1598
+ },
1599
+ {
1600
+ "epoch": 4.8771947376319345,
1601
+ "grad_norm": 10.083109855651855,
1602
+ "learning_rate": 1.2280526236806563e-06,
1603
+ "loss": 0.5977,
1604
+ "num_input_tokens_seen": 324205708,
1605
+ "step": 97500
1606
+ },
1607
+ {
1608
+ "epoch": 4.902205992696714,
1609
+ "grad_norm": 6.591386795043945,
1610
+ "learning_rate": 9.779400730328649e-07,
1611
+ "loss": 0.5633,
1612
+ "num_input_tokens_seen": 325850036,
1613
+ "step": 98000
1614
+ },
1615
+ {
1616
+ "epoch": 4.927217247761493,
1617
+ "grad_norm": 7.133991241455078,
1618
+ "learning_rate": 7.278275223850733e-07,
1619
+ "loss": 0.5786,
1620
+ "num_input_tokens_seen": 327509276,
1621
+ "step": 98500
1622
+ },
1623
+ {
1624
+ "epoch": 4.952228502826272,
1625
+ "grad_norm": 5.090227127075195,
1626
+ "learning_rate": 4.777149717372818e-07,
1627
+ "loss": 0.5886,
1628
+ "num_input_tokens_seen": 329175052,
1629
+ "step": 99000
1630
+ },
1631
+ {
1632
+ "epoch": 4.977239757891051,
1633
+ "grad_norm": 7.157599925994873,
1634
+ "learning_rate": 2.276024210894903e-07,
1635
+ "loss": 0.5879,
1636
+ "num_input_tokens_seen": 330819060,
1637
+ "step": 99500
1638
+ },
1639
+ {
1640
+ "epoch": 5.0,
1641
+ "eval_loss": 0.5393198132514954,
1642
+ "eval_runtime": 97.8527,
1643
+ "eval_samples_per_second": 408.584,
1644
+ "eval_steps_per_second": 51.077,
1645
+ "num_input_tokens_seen": 332318598,
1646
+ "step": 99955
1647
+ },
1648
+ {
1649
+ "epoch": 5.0,
1650
+ "num_input_tokens_seen": 332318598,
1651
+ "step": 99955,
1652
+ "total_flos": 1.2062750373789696e+17,
1653
+ "train_loss": 0.7601320325591829,
1654
+ "train_runtime": 7988.2275,
1655
+ "train_samples_per_second": 100.099,
1656
+ "train_steps_per_second": 12.513,
1657
+ "train_tokens_per_second": 41593.134
1658
+ }
1659
+ ],
1660
+ "logging_steps": 500,
1661
+ "max_steps": 99955,
1662
+ "num_input_tokens_seen": 332318598,
1663
+ "num_train_epochs": 5,
1664
+ "save_steps": 500,
1665
+ "stateful_callbacks": {
1666
+ "TrainerControl": {
1667
+ "args": {
1668
+ "should_epoch_stop": false,
1669
+ "should_evaluate": false,
1670
+ "should_log": false,
1671
+ "should_save": true,
1672
+ "should_training_stop": true
1673
+ },
1674
+ "attributes": {}
1675
+ }
1676
+ },
1677
+ "total_flos": 1.2062750373789696e+17,
1678
+ "train_batch_size": 8,
1679
+ "trial_name": null,
1680
+ "trial_params": null
1681
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6782a5ce63f15c3fe5f3758e612a2db9264d743242151ecb54c0e5cd4b930c9e
3
+ size 5496