agentlans commited on
Commit
7199504
·
verified ·
1 Parent(s): b54240b

Upload 12 files

Browse files
README.md CHANGED
@@ -1,3 +1,70 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: mit
4
+ base_model: avsolatorio/GIST-all-MiniLM-L6-v2
5
+ tags:
6
+ - generated_from_trainer
7
+ model-index:
8
+ - name: GIST-all-MiniLM-L6-v2-text-quality-v3
9
+ results: []
10
+ ---
11
+
12
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
+ should probably proofread and complete it, then remove this comment. -->
14
+
15
+ # GIST-all-MiniLM-L6-v2-text-quality-v3
16
+
17
+ This model is a fine-tuned version of [avsolatorio/GIST-all-MiniLM-L6-v2](https://huggingface.co/avsolatorio/GIST-all-MiniLM-L6-v2) on an unknown dataset.
18
+ It achieves the following results on the evaluation set:
19
+ - Loss: 0.1572
20
+ - Mse: 0.1572
21
+ - Combined Score: 0.1572
22
+ - Num Input Tokens Seen: 102398720
23
+
24
+ ## Model description
25
+
26
+ More information needed
27
+
28
+ ## Intended uses & limitations
29
+
30
+ More information needed
31
+
32
+ ## Training and evaluation data
33
+
34
+ More information needed
35
+
36
+ ## Training procedure
37
+
38
+ ### Training hyperparameters
39
+
40
+ The following hyperparameters were used during training:
41
+ - learning_rate: 5e-05
42
+ - train_batch_size: 8
43
+ - eval_batch_size: 8
44
+ - seed: 42
45
+ - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
46
+ - lr_scheduler_type: linear
47
+ - num_epochs: 10.0
48
+
49
+ ### Training results
50
+
51
+ | Training Loss | Epoch | Step | Validation Loss | Mse | Combined Score | Input Tokens Seen |
52
+ |:-------------:|:-----:|:------:|:---------------:|:------:|:--------------:|:-----------------:|
53
+ | 0.1777 | 1.0 | 10000 | 0.2354 | 0.2354 | 0.2354 | 10239872 |
54
+ | 0.1389 | 2.0 | 20000 | 0.1572 | 0.1572 | 0.1572 | 20479744 |
55
+ | 0.1 | 3.0 | 30000 | 0.1961 | 0.1961 | 0.1961 | 30719616 |
56
+ | 0.0687 | 4.0 | 40000 | 0.1596 | 0.1596 | 0.1596 | 40959488 |
57
+ | 0.0559 | 5.0 | 50000 | 0.1757 | 0.1757 | 0.1757 | 51199360 |
58
+ | 0.0409 | 6.0 | 60000 | 0.1677 | 0.1677 | 0.1677 | 61439232 |
59
+ | 0.0319 | 7.0 | 70000 | 0.1852 | 0.1852 | 0.1852 | 71679104 |
60
+ | 0.0266 | 8.0 | 80000 | 0.1840 | 0.1840 | 0.1840 | 81918976 |
61
+ | 0.0202 | 9.0 | 90000 | 0.1724 | 0.1724 | 0.1724 | 92158848 |
62
+ | 0.0172 | 10.0 | 100000 | 0.1731 | 0.1731 | 0.1731 | 102398720 |
63
+
64
+
65
+ ### Framework versions
66
+
67
+ - Transformers 4.51.3
68
+ - Pytorch 2.6.0+cu124
69
+ - Datasets 3.2.0
70
+ - Tokenizers 0.21.0
all_results.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 10.0,
3
+ "eval_combined_score": 0.15717529132199526,
4
+ "eval_loss": 0.15717528760433197,
5
+ "eval_mse": 0.15717529503965852,
6
+ "eval_runtime": 7.0327,
7
+ "eval_samples": 20000,
8
+ "eval_samples_per_second": 2843.868,
9
+ "eval_steps_per_second": 355.483,
10
+ "num_input_tokens_seen": 102398720,
11
+ "total_flos": 6632842477232640.0,
12
+ "train_loss": 0.06995858280181885,
13
+ "train_runtime": 1201.7215,
14
+ "train_samples": 79999,
15
+ "train_samples_per_second": 665.703,
16
+ "train_steps_per_second": 83.214,
17
+ "train_tokens_per_second": 85210.024
18
+ }
config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "BertForSequenceClassification"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.1,
6
+ "classifier_dropout": null,
7
+ "finetuning_task": "text-classification",
8
+ "gradient_checkpointing": false,
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": 6,
25
+ "pad_token_id": 0,
26
+ "position_embedding_type": "absolute",
27
+ "problem_type": "regression",
28
+ "torch_dtype": "float32",
29
+ "transformers_version": "4.51.3",
30
+ "type_vocab_size": 2,
31
+ "use_cache": true,
32
+ "vocab_size": 30522
33
+ }
eval_results.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 10.0,
3
+ "eval_combined_score": 0.15717529132199526,
4
+ "eval_loss": 0.15717528760433197,
5
+ "eval_mse": 0.15717529503965852,
6
+ "eval_runtime": 7.0327,
7
+ "eval_samples": 20000,
8
+ "eval_samples_per_second": 2843.868,
9
+ "eval_steps_per_second": 355.483,
10
+ "num_input_tokens_seen": 102398720
11
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e16431067f35e36cc24f4b6fef5e2979aa12b357aa33b5f8a011dda571cd4c32
3
+ size 90866412
special_tokens_map.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "cls_token": {
3
+ "content": "[CLS]",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "mask_token": {
10
+ "content": "[MASK]",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "[PAD]",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ },
23
+ "sep_token": {
24
+ "content": "[SEP]",
25
+ "lstrip": false,
26
+ "normalized": false,
27
+ "rstrip": false,
28
+ "single_word": false
29
+ },
30
+ "unk_token": {
31
+ "content": "[UNK]",
32
+ "lstrip": false,
33
+ "normalized": false,
34
+ "rstrip": false,
35
+ "single_word": false
36
+ }
37
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "[PAD]",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "100": {
12
+ "content": "[UNK]",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "101": {
20
+ "content": "[CLS]",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "102": {
28
+ "content": "[SEP]",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "103": {
36
+ "content": "[MASK]",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ }
43
+ },
44
+ "clean_up_tokenization_spaces": true,
45
+ "cls_token": "[CLS]",
46
+ "do_basic_tokenize": true,
47
+ "do_lower_case": true,
48
+ "extra_special_tokens": {},
49
+ "mask_token": "[MASK]",
50
+ "max_length": 128,
51
+ "model_max_length": 512,
52
+ "never_split": null,
53
+ "pad_to_multiple_of": null,
54
+ "pad_token": "[PAD]",
55
+ "pad_token_type_id": 0,
56
+ "padding_side": "right",
57
+ "sep_token": "[SEP]",
58
+ "stride": 0,
59
+ "strip_accents": null,
60
+ "tokenize_chinese_chars": true,
61
+ "tokenizer_class": "BertTokenizer",
62
+ "truncation_side": "right",
63
+ "truncation_strategy": "longest_first",
64
+ "unk_token": "[UNK]"
65
+ }
train_results.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 10.0,
3
+ "num_input_tokens_seen": 102398720,
4
+ "total_flos": 6632842477232640.0,
5
+ "train_loss": 0.06995858280181885,
6
+ "train_runtime": 1201.7215,
7
+ "train_samples": 79999,
8
+ "train_samples_per_second": 665.703,
9
+ "train_steps_per_second": 83.214,
10
+ "train_tokens_per_second": 85210.024
11
+ }
trainer_state.json ADDED
@@ -0,0 +1,1755 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 20000,
3
+ "best_metric": 0.15717528760433197,
4
+ "best_model_checkpoint": "/media/user/Expansion1/GIST-all-MiniLM-L6-v2-text-quality-v3/checkpoint-20000",
5
+ "epoch": 10.0,
6
+ "eval_steps": 500,
7
+ "global_step": 100000,
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": 10.430134773254395,
15
+ "learning_rate": 4.97505e-05,
16
+ "loss": 0.381,
17
+ "num_input_tokens_seen": 512000,
18
+ "step": 500
19
+ },
20
+ {
21
+ "epoch": 0.1,
22
+ "grad_norm": 6.132630348205566,
23
+ "learning_rate": 4.95005e-05,
24
+ "loss": 0.2502,
25
+ "num_input_tokens_seen": 1024000,
26
+ "step": 1000
27
+ },
28
+ {
29
+ "epoch": 0.15,
30
+ "grad_norm": 15.999980926513672,
31
+ "learning_rate": 4.9250500000000006e-05,
32
+ "loss": 0.2378,
33
+ "num_input_tokens_seen": 1536000,
34
+ "step": 1500
35
+ },
36
+ {
37
+ "epoch": 0.2,
38
+ "grad_norm": 4.05879545211792,
39
+ "learning_rate": 4.9000500000000006e-05,
40
+ "loss": 0.2275,
41
+ "num_input_tokens_seen": 2048000,
42
+ "step": 2000
43
+ },
44
+ {
45
+ "epoch": 0.25,
46
+ "grad_norm": 4.510811805725098,
47
+ "learning_rate": 4.875050000000001e-05,
48
+ "loss": 0.2184,
49
+ "num_input_tokens_seen": 2560000,
50
+ "step": 2500
51
+ },
52
+ {
53
+ "epoch": 0.3,
54
+ "grad_norm": 14.602776527404785,
55
+ "learning_rate": 4.85005e-05,
56
+ "loss": 0.2078,
57
+ "num_input_tokens_seen": 3072000,
58
+ "step": 3000
59
+ },
60
+ {
61
+ "epoch": 0.35,
62
+ "grad_norm": 5.921443939208984,
63
+ "learning_rate": 4.82505e-05,
64
+ "loss": 0.2032,
65
+ "num_input_tokens_seen": 3584000,
66
+ "step": 3500
67
+ },
68
+ {
69
+ "epoch": 0.4,
70
+ "grad_norm": 6.71400785446167,
71
+ "learning_rate": 4.80005e-05,
72
+ "loss": 0.2032,
73
+ "num_input_tokens_seen": 4096000,
74
+ "step": 4000
75
+ },
76
+ {
77
+ "epoch": 0.45,
78
+ "grad_norm": 7.379507541656494,
79
+ "learning_rate": 4.77505e-05,
80
+ "loss": 0.2099,
81
+ "num_input_tokens_seen": 4608000,
82
+ "step": 4500
83
+ },
84
+ {
85
+ "epoch": 0.5,
86
+ "grad_norm": 8.111740112304688,
87
+ "learning_rate": 4.7500500000000004e-05,
88
+ "loss": 0.1901,
89
+ "num_input_tokens_seen": 5120000,
90
+ "step": 5000
91
+ },
92
+ {
93
+ "epoch": 0.55,
94
+ "grad_norm": 3.8721070289611816,
95
+ "learning_rate": 4.7250500000000004e-05,
96
+ "loss": 0.1923,
97
+ "num_input_tokens_seen": 5632000,
98
+ "step": 5500
99
+ },
100
+ {
101
+ "epoch": 0.6,
102
+ "grad_norm": 4.166099548339844,
103
+ "learning_rate": 4.7000500000000005e-05,
104
+ "loss": 0.1807,
105
+ "num_input_tokens_seen": 6144000,
106
+ "step": 6000
107
+ },
108
+ {
109
+ "epoch": 0.65,
110
+ "grad_norm": 7.753055572509766,
111
+ "learning_rate": 4.6750500000000006e-05,
112
+ "loss": 0.1856,
113
+ "num_input_tokens_seen": 6656000,
114
+ "step": 6500
115
+ },
116
+ {
117
+ "epoch": 0.7,
118
+ "grad_norm": 9.00536823272705,
119
+ "learning_rate": 4.65005e-05,
120
+ "loss": 0.1864,
121
+ "num_input_tokens_seen": 7168000,
122
+ "step": 7000
123
+ },
124
+ {
125
+ "epoch": 0.75,
126
+ "grad_norm": 3.9977951049804688,
127
+ "learning_rate": 4.62505e-05,
128
+ "loss": 0.1739,
129
+ "num_input_tokens_seen": 7680000,
130
+ "step": 7500
131
+ },
132
+ {
133
+ "epoch": 0.8,
134
+ "grad_norm": 16.222850799560547,
135
+ "learning_rate": 4.60005e-05,
136
+ "loss": 0.183,
137
+ "num_input_tokens_seen": 8192000,
138
+ "step": 8000
139
+ },
140
+ {
141
+ "epoch": 0.85,
142
+ "grad_norm": 6.198567867279053,
143
+ "learning_rate": 4.57505e-05,
144
+ "loss": 0.176,
145
+ "num_input_tokens_seen": 8704000,
146
+ "step": 8500
147
+ },
148
+ {
149
+ "epoch": 0.9,
150
+ "grad_norm": 4.518280029296875,
151
+ "learning_rate": 4.55005e-05,
152
+ "loss": 0.1728,
153
+ "num_input_tokens_seen": 9216000,
154
+ "step": 9000
155
+ },
156
+ {
157
+ "epoch": 0.95,
158
+ "grad_norm": 3.101031541824341,
159
+ "learning_rate": 4.52505e-05,
160
+ "loss": 0.1737,
161
+ "num_input_tokens_seen": 9728000,
162
+ "step": 9500
163
+ },
164
+ {
165
+ "epoch": 1.0,
166
+ "grad_norm": 13.572478294372559,
167
+ "learning_rate": 4.5000500000000004e-05,
168
+ "loss": 0.1777,
169
+ "num_input_tokens_seen": 10239872,
170
+ "step": 10000
171
+ },
172
+ {
173
+ "epoch": 1.0,
174
+ "eval_combined_score": 0.23537112119271972,
175
+ "eval_loss": 0.23537111282348633,
176
+ "eval_mse": 0.2353711295619531,
177
+ "eval_runtime": 7.0429,
178
+ "eval_samples_per_second": 2839.722,
179
+ "eval_steps_per_second": 354.965,
180
+ "num_input_tokens_seen": 10239872,
181
+ "step": 10000
182
+ },
183
+ {
184
+ "epoch": 1.05,
185
+ "grad_norm": 3.3423728942871094,
186
+ "learning_rate": 4.47505e-05,
187
+ "loss": 0.1346,
188
+ "num_input_tokens_seen": 10751872,
189
+ "step": 10500
190
+ },
191
+ {
192
+ "epoch": 1.1,
193
+ "grad_norm": 2.3073153495788574,
194
+ "learning_rate": 4.45005e-05,
195
+ "loss": 0.132,
196
+ "num_input_tokens_seen": 11263872,
197
+ "step": 11000
198
+ },
199
+ {
200
+ "epoch": 1.15,
201
+ "grad_norm": 1.443601369857788,
202
+ "learning_rate": 4.42505e-05,
203
+ "loss": 0.133,
204
+ "num_input_tokens_seen": 11775872,
205
+ "step": 11500
206
+ },
207
+ {
208
+ "epoch": 1.2,
209
+ "grad_norm": 2.6792664527893066,
210
+ "learning_rate": 4.40005e-05,
211
+ "loss": 0.1327,
212
+ "num_input_tokens_seen": 12287872,
213
+ "step": 12000
214
+ },
215
+ {
216
+ "epoch": 1.25,
217
+ "grad_norm": 4.662805557250977,
218
+ "learning_rate": 4.37505e-05,
219
+ "loss": 0.1265,
220
+ "num_input_tokens_seen": 12799872,
221
+ "step": 12500
222
+ },
223
+ {
224
+ "epoch": 1.3,
225
+ "grad_norm": 0.8095790147781372,
226
+ "learning_rate": 4.35005e-05,
227
+ "loss": 0.1267,
228
+ "num_input_tokens_seen": 13311872,
229
+ "step": 13000
230
+ },
231
+ {
232
+ "epoch": 1.35,
233
+ "grad_norm": 2.9624879360198975,
234
+ "learning_rate": 4.32505e-05,
235
+ "loss": 0.1288,
236
+ "num_input_tokens_seen": 13823872,
237
+ "step": 13500
238
+ },
239
+ {
240
+ "epoch": 1.4,
241
+ "grad_norm": 3.0661849975585938,
242
+ "learning_rate": 4.30005e-05,
243
+ "loss": 0.1307,
244
+ "num_input_tokens_seen": 14335872,
245
+ "step": 14000
246
+ },
247
+ {
248
+ "epoch": 1.45,
249
+ "grad_norm": 2.763495683670044,
250
+ "learning_rate": 4.2750500000000003e-05,
251
+ "loss": 0.1387,
252
+ "num_input_tokens_seen": 14847872,
253
+ "step": 14500
254
+ },
255
+ {
256
+ "epoch": 1.5,
257
+ "grad_norm": 2.1672914028167725,
258
+ "learning_rate": 4.2500500000000004e-05,
259
+ "loss": 0.1332,
260
+ "num_input_tokens_seen": 15359872,
261
+ "step": 15000
262
+ },
263
+ {
264
+ "epoch": 1.55,
265
+ "grad_norm": 3.3547232151031494,
266
+ "learning_rate": 4.2250500000000005e-05,
267
+ "loss": 0.1387,
268
+ "num_input_tokens_seen": 15871872,
269
+ "step": 15500
270
+ },
271
+ {
272
+ "epoch": 1.6,
273
+ "grad_norm": 1.501448631286621,
274
+ "learning_rate": 4.2000500000000006e-05,
275
+ "loss": 0.1313,
276
+ "num_input_tokens_seen": 16383872,
277
+ "step": 16000
278
+ },
279
+ {
280
+ "epoch": 1.65,
281
+ "grad_norm": 3.581545114517212,
282
+ "learning_rate": 4.1750500000000006e-05,
283
+ "loss": 0.1377,
284
+ "num_input_tokens_seen": 16895872,
285
+ "step": 16500
286
+ },
287
+ {
288
+ "epoch": 1.7,
289
+ "grad_norm": 5.36037540435791,
290
+ "learning_rate": 4.15005e-05,
291
+ "loss": 0.1312,
292
+ "num_input_tokens_seen": 17407872,
293
+ "step": 17000
294
+ },
295
+ {
296
+ "epoch": 1.75,
297
+ "grad_norm": 3.6308796405792236,
298
+ "learning_rate": 4.12505e-05,
299
+ "loss": 0.1339,
300
+ "num_input_tokens_seen": 17919872,
301
+ "step": 17500
302
+ },
303
+ {
304
+ "epoch": 1.8,
305
+ "grad_norm": 5.437199592590332,
306
+ "learning_rate": 4.10005e-05,
307
+ "loss": 0.136,
308
+ "num_input_tokens_seen": 18431872,
309
+ "step": 18000
310
+ },
311
+ {
312
+ "epoch": 1.85,
313
+ "grad_norm": 4.052628993988037,
314
+ "learning_rate": 4.07505e-05,
315
+ "loss": 0.1356,
316
+ "num_input_tokens_seen": 18943872,
317
+ "step": 18500
318
+ },
319
+ {
320
+ "epoch": 1.9,
321
+ "grad_norm": 4.826311111450195,
322
+ "learning_rate": 4.05005e-05,
323
+ "loss": 0.1399,
324
+ "num_input_tokens_seen": 19455872,
325
+ "step": 19000
326
+ },
327
+ {
328
+ "epoch": 1.95,
329
+ "grad_norm": 4.320687294006348,
330
+ "learning_rate": 4.0250500000000004e-05,
331
+ "loss": 0.136,
332
+ "num_input_tokens_seen": 19967872,
333
+ "step": 19500
334
+ },
335
+ {
336
+ "epoch": 2.0,
337
+ "grad_norm": 3.1729485988616943,
338
+ "learning_rate": 4.0000500000000004e-05,
339
+ "loss": 0.1389,
340
+ "num_input_tokens_seen": 20479744,
341
+ "step": 20000
342
+ },
343
+ {
344
+ "epoch": 2.0,
345
+ "eval_combined_score": 0.15717529132199526,
346
+ "eval_loss": 0.15717528760433197,
347
+ "eval_mse": 0.15717529503965852,
348
+ "eval_runtime": 7.0647,
349
+ "eval_samples_per_second": 2830.974,
350
+ "eval_steps_per_second": 353.872,
351
+ "num_input_tokens_seen": 20479744,
352
+ "step": 20000
353
+ },
354
+ {
355
+ "epoch": 2.05,
356
+ "grad_norm": 1.8643279075622559,
357
+ "learning_rate": 3.97505e-05,
358
+ "loss": 0.0982,
359
+ "num_input_tokens_seen": 20991744,
360
+ "step": 20500
361
+ },
362
+ {
363
+ "epoch": 2.1,
364
+ "grad_norm": 4.278918743133545,
365
+ "learning_rate": 3.95005e-05,
366
+ "loss": 0.0919,
367
+ "num_input_tokens_seen": 21503744,
368
+ "step": 21000
369
+ },
370
+ {
371
+ "epoch": 2.15,
372
+ "grad_norm": 3.386025905609131,
373
+ "learning_rate": 3.92505e-05,
374
+ "loss": 0.0966,
375
+ "num_input_tokens_seen": 22015744,
376
+ "step": 21500
377
+ },
378
+ {
379
+ "epoch": 2.2,
380
+ "grad_norm": 1.091706395149231,
381
+ "learning_rate": 3.90005e-05,
382
+ "loss": 0.098,
383
+ "num_input_tokens_seen": 22527744,
384
+ "step": 22000
385
+ },
386
+ {
387
+ "epoch": 2.25,
388
+ "grad_norm": 7.324082374572754,
389
+ "learning_rate": 3.87505e-05,
390
+ "loss": 0.0977,
391
+ "num_input_tokens_seen": 23039744,
392
+ "step": 22500
393
+ },
394
+ {
395
+ "epoch": 2.3,
396
+ "grad_norm": 4.270073890686035,
397
+ "learning_rate": 3.85005e-05,
398
+ "loss": 0.098,
399
+ "num_input_tokens_seen": 23551744,
400
+ "step": 23000
401
+ },
402
+ {
403
+ "epoch": 2.35,
404
+ "grad_norm": 2.5364954471588135,
405
+ "learning_rate": 3.82505e-05,
406
+ "loss": 0.1023,
407
+ "num_input_tokens_seen": 24063744,
408
+ "step": 23500
409
+ },
410
+ {
411
+ "epoch": 2.4,
412
+ "grad_norm": 4.379617691040039,
413
+ "learning_rate": 3.80005e-05,
414
+ "loss": 0.1014,
415
+ "num_input_tokens_seen": 24575744,
416
+ "step": 24000
417
+ },
418
+ {
419
+ "epoch": 2.45,
420
+ "grad_norm": 2.539891242980957,
421
+ "learning_rate": 3.77505e-05,
422
+ "loss": 0.094,
423
+ "num_input_tokens_seen": 25087744,
424
+ "step": 24500
425
+ },
426
+ {
427
+ "epoch": 2.5,
428
+ "grad_norm": 6.501862049102783,
429
+ "learning_rate": 3.75005e-05,
430
+ "loss": 0.0944,
431
+ "num_input_tokens_seen": 25599744,
432
+ "step": 25000
433
+ },
434
+ {
435
+ "epoch": 2.55,
436
+ "grad_norm": 4.673591613769531,
437
+ "learning_rate": 3.72505e-05,
438
+ "loss": 0.0982,
439
+ "num_input_tokens_seen": 26111744,
440
+ "step": 25500
441
+ },
442
+ {
443
+ "epoch": 2.6,
444
+ "grad_norm": 3.8183979988098145,
445
+ "learning_rate": 3.70005e-05,
446
+ "loss": 0.0904,
447
+ "num_input_tokens_seen": 26623744,
448
+ "step": 26000
449
+ },
450
+ {
451
+ "epoch": 2.65,
452
+ "grad_norm": 4.115967273712158,
453
+ "learning_rate": 3.675050000000001e-05,
454
+ "loss": 0.0995,
455
+ "num_input_tokens_seen": 27135744,
456
+ "step": 26500
457
+ },
458
+ {
459
+ "epoch": 2.7,
460
+ "grad_norm": 2.9983203411102295,
461
+ "learning_rate": 3.650050000000001e-05,
462
+ "loss": 0.1038,
463
+ "num_input_tokens_seen": 27647744,
464
+ "step": 27000
465
+ },
466
+ {
467
+ "epoch": 2.75,
468
+ "grad_norm": 4.20590353012085,
469
+ "learning_rate": 3.62505e-05,
470
+ "loss": 0.1023,
471
+ "num_input_tokens_seen": 28159744,
472
+ "step": 27500
473
+ },
474
+ {
475
+ "epoch": 2.8,
476
+ "grad_norm": 3.6869823932647705,
477
+ "learning_rate": 3.60005e-05,
478
+ "loss": 0.1024,
479
+ "num_input_tokens_seen": 28671744,
480
+ "step": 28000
481
+ },
482
+ {
483
+ "epoch": 2.85,
484
+ "grad_norm": 2.130293607711792,
485
+ "learning_rate": 3.57505e-05,
486
+ "loss": 0.0952,
487
+ "num_input_tokens_seen": 29183744,
488
+ "step": 28500
489
+ },
490
+ {
491
+ "epoch": 2.9,
492
+ "grad_norm": 2.846696376800537,
493
+ "learning_rate": 3.5500500000000003e-05,
494
+ "loss": 0.0994,
495
+ "num_input_tokens_seen": 29695744,
496
+ "step": 29000
497
+ },
498
+ {
499
+ "epoch": 2.95,
500
+ "grad_norm": 6.687178134918213,
501
+ "learning_rate": 3.5250500000000004e-05,
502
+ "loss": 0.0981,
503
+ "num_input_tokens_seen": 30207744,
504
+ "step": 29500
505
+ },
506
+ {
507
+ "epoch": 3.0,
508
+ "grad_norm": 2.013150930404663,
509
+ "learning_rate": 3.5000500000000005e-05,
510
+ "loss": 0.1,
511
+ "num_input_tokens_seen": 30719616,
512
+ "step": 30000
513
+ },
514
+ {
515
+ "epoch": 3.0,
516
+ "eval_combined_score": 0.1961452062019333,
517
+ "eval_loss": 0.1961452215909958,
518
+ "eval_mse": 0.196145205714032,
519
+ "eval_runtime": 7.1175,
520
+ "eval_samples_per_second": 2809.967,
521
+ "eval_steps_per_second": 351.246,
522
+ "num_input_tokens_seen": 30719616,
523
+ "step": 30000
524
+ },
525
+ {
526
+ "epoch": 3.05,
527
+ "grad_norm": 1.9965280294418335,
528
+ "learning_rate": 3.4750500000000006e-05,
529
+ "loss": 0.0703,
530
+ "num_input_tokens_seen": 31231616,
531
+ "step": 30500
532
+ },
533
+ {
534
+ "epoch": 3.1,
535
+ "grad_norm": 5.204916954040527,
536
+ "learning_rate": 3.45005e-05,
537
+ "loss": 0.0713,
538
+ "num_input_tokens_seen": 31743616,
539
+ "step": 31000
540
+ },
541
+ {
542
+ "epoch": 3.15,
543
+ "grad_norm": 2.138298273086548,
544
+ "learning_rate": 3.42505e-05,
545
+ "loss": 0.0663,
546
+ "num_input_tokens_seen": 32255616,
547
+ "step": 31500
548
+ },
549
+ {
550
+ "epoch": 3.2,
551
+ "grad_norm": 2.778095245361328,
552
+ "learning_rate": 3.40005e-05,
553
+ "loss": 0.0711,
554
+ "num_input_tokens_seen": 32767616,
555
+ "step": 32000
556
+ },
557
+ {
558
+ "epoch": 3.25,
559
+ "grad_norm": 3.4932539463043213,
560
+ "learning_rate": 3.37505e-05,
561
+ "loss": 0.0723,
562
+ "num_input_tokens_seen": 33279616,
563
+ "step": 32500
564
+ },
565
+ {
566
+ "epoch": 3.3,
567
+ "grad_norm": 2.357044219970703,
568
+ "learning_rate": 3.35005e-05,
569
+ "loss": 0.0728,
570
+ "num_input_tokens_seen": 33791616,
571
+ "step": 33000
572
+ },
573
+ {
574
+ "epoch": 3.35,
575
+ "grad_norm": 5.320715427398682,
576
+ "learning_rate": 3.32505e-05,
577
+ "loss": 0.0693,
578
+ "num_input_tokens_seen": 34303616,
579
+ "step": 33500
580
+ },
581
+ {
582
+ "epoch": 3.4,
583
+ "grad_norm": 1.8415720462799072,
584
+ "learning_rate": 3.3000500000000004e-05,
585
+ "loss": 0.0723,
586
+ "num_input_tokens_seen": 34815616,
587
+ "step": 34000
588
+ },
589
+ {
590
+ "epoch": 3.45,
591
+ "grad_norm": 6.073017120361328,
592
+ "learning_rate": 3.27505e-05,
593
+ "loss": 0.0724,
594
+ "num_input_tokens_seen": 35327616,
595
+ "step": 34500
596
+ },
597
+ {
598
+ "epoch": 3.5,
599
+ "grad_norm": 2.4829020500183105,
600
+ "learning_rate": 3.25005e-05,
601
+ "loss": 0.073,
602
+ "num_input_tokens_seen": 35839616,
603
+ "step": 35000
604
+ },
605
+ {
606
+ "epoch": 3.55,
607
+ "grad_norm": 1.3849653005599976,
608
+ "learning_rate": 3.22505e-05,
609
+ "loss": 0.0674,
610
+ "num_input_tokens_seen": 36351616,
611
+ "step": 35500
612
+ },
613
+ {
614
+ "epoch": 3.6,
615
+ "grad_norm": 1.6798043251037598,
616
+ "learning_rate": 3.20005e-05,
617
+ "loss": 0.0739,
618
+ "num_input_tokens_seen": 36863616,
619
+ "step": 36000
620
+ },
621
+ {
622
+ "epoch": 3.65,
623
+ "grad_norm": 2.017993927001953,
624
+ "learning_rate": 3.17505e-05,
625
+ "loss": 0.071,
626
+ "num_input_tokens_seen": 37375616,
627
+ "step": 36500
628
+ },
629
+ {
630
+ "epoch": 3.7,
631
+ "grad_norm": 2.444216251373291,
632
+ "learning_rate": 3.15005e-05,
633
+ "loss": 0.0726,
634
+ "num_input_tokens_seen": 37887616,
635
+ "step": 37000
636
+ },
637
+ {
638
+ "epoch": 3.75,
639
+ "grad_norm": 6.782371520996094,
640
+ "learning_rate": 3.12505e-05,
641
+ "loss": 0.0734,
642
+ "num_input_tokens_seen": 38399616,
643
+ "step": 37500
644
+ },
645
+ {
646
+ "epoch": 3.8,
647
+ "grad_norm": 2.4976208209991455,
648
+ "learning_rate": 3.1000499999999996e-05,
649
+ "loss": 0.0712,
650
+ "num_input_tokens_seen": 38911616,
651
+ "step": 38000
652
+ },
653
+ {
654
+ "epoch": 3.85,
655
+ "grad_norm": 1.0673044919967651,
656
+ "learning_rate": 3.0750499999999996e-05,
657
+ "loss": 0.0697,
658
+ "num_input_tokens_seen": 39423616,
659
+ "step": 38500
660
+ },
661
+ {
662
+ "epoch": 3.9,
663
+ "grad_norm": 7.465501308441162,
664
+ "learning_rate": 3.0500500000000004e-05,
665
+ "loss": 0.0719,
666
+ "num_input_tokens_seen": 39935616,
667
+ "step": 39000
668
+ },
669
+ {
670
+ "epoch": 3.95,
671
+ "grad_norm": 3.7942514419555664,
672
+ "learning_rate": 3.0250500000000005e-05,
673
+ "loss": 0.0729,
674
+ "num_input_tokens_seen": 40447616,
675
+ "step": 39500
676
+ },
677
+ {
678
+ "epoch": 4.0,
679
+ "grad_norm": 4.152807235717773,
680
+ "learning_rate": 3.0000500000000005e-05,
681
+ "loss": 0.0687,
682
+ "num_input_tokens_seen": 40959488,
683
+ "step": 40000
684
+ },
685
+ {
686
+ "epoch": 4.0,
687
+ "eval_combined_score": 0.1595882055619099,
688
+ "eval_loss": 0.15958820283412933,
689
+ "eval_mse": 0.15958820828969045,
690
+ "eval_runtime": 7.0738,
691
+ "eval_samples_per_second": 2827.348,
692
+ "eval_steps_per_second": 353.418,
693
+ "num_input_tokens_seen": 40959488,
694
+ "step": 40000
695
+ },
696
+ {
697
+ "epoch": 4.05,
698
+ "grad_norm": 6.4203410148620605,
699
+ "learning_rate": 2.9750500000000003e-05,
700
+ "loss": 0.0499,
701
+ "num_input_tokens_seen": 41471488,
702
+ "step": 40500
703
+ },
704
+ {
705
+ "epoch": 4.1,
706
+ "grad_norm": 1.5940698385238647,
707
+ "learning_rate": 2.9500500000000003e-05,
708
+ "loss": 0.0494,
709
+ "num_input_tokens_seen": 41983488,
710
+ "step": 41000
711
+ },
712
+ {
713
+ "epoch": 4.15,
714
+ "grad_norm": 1.753504753112793,
715
+ "learning_rate": 2.9250500000000004e-05,
716
+ "loss": 0.0514,
717
+ "num_input_tokens_seen": 42495488,
718
+ "step": 41500
719
+ },
720
+ {
721
+ "epoch": 4.2,
722
+ "grad_norm": 1.8505773544311523,
723
+ "learning_rate": 2.90005e-05,
724
+ "loss": 0.0535,
725
+ "num_input_tokens_seen": 43007488,
726
+ "step": 42000
727
+ },
728
+ {
729
+ "epoch": 4.25,
730
+ "grad_norm": 4.076910018920898,
731
+ "learning_rate": 2.8750500000000002e-05,
732
+ "loss": 0.0513,
733
+ "num_input_tokens_seen": 43519488,
734
+ "step": 42500
735
+ },
736
+ {
737
+ "epoch": 4.3,
738
+ "grad_norm": 2.8389265537261963,
739
+ "learning_rate": 2.8500500000000003e-05,
740
+ "loss": 0.0509,
741
+ "num_input_tokens_seen": 44031488,
742
+ "step": 43000
743
+ },
744
+ {
745
+ "epoch": 4.35,
746
+ "grad_norm": 2.887282133102417,
747
+ "learning_rate": 2.8250500000000003e-05,
748
+ "loss": 0.0527,
749
+ "num_input_tokens_seen": 44543488,
750
+ "step": 43500
751
+ },
752
+ {
753
+ "epoch": 4.4,
754
+ "grad_norm": 2.474107027053833,
755
+ "learning_rate": 2.80005e-05,
756
+ "loss": 0.0522,
757
+ "num_input_tokens_seen": 45055488,
758
+ "step": 44000
759
+ },
760
+ {
761
+ "epoch": 4.45,
762
+ "grad_norm": 1.4676923751831055,
763
+ "learning_rate": 2.77505e-05,
764
+ "loss": 0.0541,
765
+ "num_input_tokens_seen": 45567488,
766
+ "step": 44500
767
+ },
768
+ {
769
+ "epoch": 4.5,
770
+ "grad_norm": 1.7855340242385864,
771
+ "learning_rate": 2.7500500000000002e-05,
772
+ "loss": 0.0524,
773
+ "num_input_tokens_seen": 46079488,
774
+ "step": 45000
775
+ },
776
+ {
777
+ "epoch": 4.55,
778
+ "grad_norm": 3.4113402366638184,
779
+ "learning_rate": 2.72505e-05,
780
+ "loss": 0.0523,
781
+ "num_input_tokens_seen": 46591488,
782
+ "step": 45500
783
+ },
784
+ {
785
+ "epoch": 4.6,
786
+ "grad_norm": 2.2944934368133545,
787
+ "learning_rate": 2.70005e-05,
788
+ "loss": 0.0536,
789
+ "num_input_tokens_seen": 47103488,
790
+ "step": 46000
791
+ },
792
+ {
793
+ "epoch": 4.65,
794
+ "grad_norm": 3.624361038208008,
795
+ "learning_rate": 2.67505e-05,
796
+ "loss": 0.0515,
797
+ "num_input_tokens_seen": 47615488,
798
+ "step": 46500
799
+ },
800
+ {
801
+ "epoch": 4.7,
802
+ "grad_norm": 1.4528284072875977,
803
+ "learning_rate": 2.65005e-05,
804
+ "loss": 0.0544,
805
+ "num_input_tokens_seen": 48127488,
806
+ "step": 47000
807
+ },
808
+ {
809
+ "epoch": 4.75,
810
+ "grad_norm": 1.9667190313339233,
811
+ "learning_rate": 2.62505e-05,
812
+ "loss": 0.0559,
813
+ "num_input_tokens_seen": 48639488,
814
+ "step": 47500
815
+ },
816
+ {
817
+ "epoch": 4.8,
818
+ "grad_norm": 3.1912217140197754,
819
+ "learning_rate": 2.60005e-05,
820
+ "loss": 0.0539,
821
+ "num_input_tokens_seen": 49151488,
822
+ "step": 48000
823
+ },
824
+ {
825
+ "epoch": 4.85,
826
+ "grad_norm": 1.3516061305999756,
827
+ "learning_rate": 2.57505e-05,
828
+ "loss": 0.0566,
829
+ "num_input_tokens_seen": 49663488,
830
+ "step": 48500
831
+ },
832
+ {
833
+ "epoch": 4.9,
834
+ "grad_norm": 3.7553634643554688,
835
+ "learning_rate": 2.55005e-05,
836
+ "loss": 0.0544,
837
+ "num_input_tokens_seen": 50175488,
838
+ "step": 49000
839
+ },
840
+ {
841
+ "epoch": 4.95,
842
+ "grad_norm": 0.9317166805267334,
843
+ "learning_rate": 2.5250499999999998e-05,
844
+ "loss": 0.0551,
845
+ "num_input_tokens_seen": 50687488,
846
+ "step": 49500
847
+ },
848
+ {
849
+ "epoch": 5.0,
850
+ "grad_norm": 3.4310598373413086,
851
+ "learning_rate": 2.50005e-05,
852
+ "loss": 0.0559,
853
+ "num_input_tokens_seen": 51199360,
854
+ "step": 50000
855
+ },
856
+ {
857
+ "epoch": 5.0,
858
+ "eval_combined_score": 0.17571219370952593,
859
+ "eval_loss": 0.1757121980190277,
860
+ "eval_mse": 0.17571218940002414,
861
+ "eval_runtime": 7.0238,
862
+ "eval_samples_per_second": 2847.448,
863
+ "eval_steps_per_second": 355.931,
864
+ "num_input_tokens_seen": 51199360,
865
+ "step": 50000
866
+ },
867
+ {
868
+ "epoch": 5.05,
869
+ "grad_norm": 1.6329584121704102,
870
+ "learning_rate": 2.4750500000000003e-05,
871
+ "loss": 0.0392,
872
+ "num_input_tokens_seen": 51711360,
873
+ "step": 50500
874
+ },
875
+ {
876
+ "epoch": 5.1,
877
+ "grad_norm": 4.350171089172363,
878
+ "learning_rate": 2.45005e-05,
879
+ "loss": 0.0381,
880
+ "num_input_tokens_seen": 52223360,
881
+ "step": 51000
882
+ },
883
+ {
884
+ "epoch": 5.15,
885
+ "grad_norm": 1.3488142490386963,
886
+ "learning_rate": 2.42505e-05,
887
+ "loss": 0.04,
888
+ "num_input_tokens_seen": 52735360,
889
+ "step": 51500
890
+ },
891
+ {
892
+ "epoch": 5.2,
893
+ "grad_norm": 2.9289724826812744,
894
+ "learning_rate": 2.4000500000000002e-05,
895
+ "loss": 0.0401,
896
+ "num_input_tokens_seen": 53247360,
897
+ "step": 52000
898
+ },
899
+ {
900
+ "epoch": 5.25,
901
+ "grad_norm": 2.1096582412719727,
902
+ "learning_rate": 2.37505e-05,
903
+ "loss": 0.0417,
904
+ "num_input_tokens_seen": 53759360,
905
+ "step": 52500
906
+ },
907
+ {
908
+ "epoch": 5.3,
909
+ "grad_norm": 3.2949271202087402,
910
+ "learning_rate": 2.35005e-05,
911
+ "loss": 0.0406,
912
+ "num_input_tokens_seen": 54271360,
913
+ "step": 53000
914
+ },
915
+ {
916
+ "epoch": 5.35,
917
+ "grad_norm": 1.4956310987472534,
918
+ "learning_rate": 2.32505e-05,
919
+ "loss": 0.0397,
920
+ "num_input_tokens_seen": 54783360,
921
+ "step": 53500
922
+ },
923
+ {
924
+ "epoch": 5.4,
925
+ "grad_norm": 3.9012367725372314,
926
+ "learning_rate": 2.30005e-05,
927
+ "loss": 0.0414,
928
+ "num_input_tokens_seen": 55295360,
929
+ "step": 54000
930
+ },
931
+ {
932
+ "epoch": 5.45,
933
+ "grad_norm": 4.583008289337158,
934
+ "learning_rate": 2.2750500000000002e-05,
935
+ "loss": 0.0411,
936
+ "num_input_tokens_seen": 55807360,
937
+ "step": 54500
938
+ },
939
+ {
940
+ "epoch": 5.5,
941
+ "grad_norm": 1.1253963708877563,
942
+ "learning_rate": 2.2500500000000003e-05,
943
+ "loss": 0.04,
944
+ "num_input_tokens_seen": 56319360,
945
+ "step": 55000
946
+ },
947
+ {
948
+ "epoch": 5.55,
949
+ "grad_norm": 2.3628132343292236,
950
+ "learning_rate": 2.2250500000000003e-05,
951
+ "loss": 0.0407,
952
+ "num_input_tokens_seen": 56831360,
953
+ "step": 55500
954
+ },
955
+ {
956
+ "epoch": 5.6,
957
+ "grad_norm": 2.3171374797821045,
958
+ "learning_rate": 2.20005e-05,
959
+ "loss": 0.0389,
960
+ "num_input_tokens_seen": 57343360,
961
+ "step": 56000
962
+ },
963
+ {
964
+ "epoch": 5.65,
965
+ "grad_norm": 1.1634831428527832,
966
+ "learning_rate": 2.17505e-05,
967
+ "loss": 0.0429,
968
+ "num_input_tokens_seen": 57855360,
969
+ "step": 56500
970
+ },
971
+ {
972
+ "epoch": 5.7,
973
+ "grad_norm": 2.303175926208496,
974
+ "learning_rate": 2.1500500000000002e-05,
975
+ "loss": 0.0428,
976
+ "num_input_tokens_seen": 58367360,
977
+ "step": 57000
978
+ },
979
+ {
980
+ "epoch": 5.75,
981
+ "grad_norm": 2.301100015640259,
982
+ "learning_rate": 2.1250500000000003e-05,
983
+ "loss": 0.0412,
984
+ "num_input_tokens_seen": 58879360,
985
+ "step": 57500
986
+ },
987
+ {
988
+ "epoch": 5.8,
989
+ "grad_norm": 3.2030017375946045,
990
+ "learning_rate": 2.10005e-05,
991
+ "loss": 0.0408,
992
+ "num_input_tokens_seen": 59391360,
993
+ "step": 58000
994
+ },
995
+ {
996
+ "epoch": 5.85,
997
+ "grad_norm": 2.3684282302856445,
998
+ "learning_rate": 2.07505e-05,
999
+ "loss": 0.0435,
1000
+ "num_input_tokens_seen": 59903360,
1001
+ "step": 58500
1002
+ },
1003
+ {
1004
+ "epoch": 5.9,
1005
+ "grad_norm": 2.874260902404785,
1006
+ "learning_rate": 2.05005e-05,
1007
+ "loss": 0.0412,
1008
+ "num_input_tokens_seen": 60415360,
1009
+ "step": 59000
1010
+ },
1011
+ {
1012
+ "epoch": 5.95,
1013
+ "grad_norm": 3.2977681159973145,
1014
+ "learning_rate": 2.02505e-05,
1015
+ "loss": 0.0411,
1016
+ "num_input_tokens_seen": 60927360,
1017
+ "step": 59500
1018
+ },
1019
+ {
1020
+ "epoch": 6.0,
1021
+ "grad_norm": 3.4521987438201904,
1022
+ "learning_rate": 2.00005e-05,
1023
+ "loss": 0.0409,
1024
+ "num_input_tokens_seen": 61439232,
1025
+ "step": 60000
1026
+ },
1027
+ {
1028
+ "epoch": 6.0,
1029
+ "eval_combined_score": 0.16771116720526497,
1030
+ "eval_loss": 0.16771116852760315,
1031
+ "eval_mse": 0.16771116588292675,
1032
+ "eval_runtime": 6.9877,
1033
+ "eval_samples_per_second": 2862.173,
1034
+ "eval_steps_per_second": 357.772,
1035
+ "num_input_tokens_seen": 61439232,
1036
+ "step": 60000
1037
+ },
1038
+ {
1039
+ "epoch": 6.05,
1040
+ "grad_norm": 3.089998960494995,
1041
+ "learning_rate": 1.97505e-05,
1042
+ "loss": 0.0281,
1043
+ "num_input_tokens_seen": 61951232,
1044
+ "step": 60500
1045
+ },
1046
+ {
1047
+ "epoch": 6.1,
1048
+ "grad_norm": 1.7373288869857788,
1049
+ "learning_rate": 1.95005e-05,
1050
+ "loss": 0.0304,
1051
+ "num_input_tokens_seen": 62463232,
1052
+ "step": 61000
1053
+ },
1054
+ {
1055
+ "epoch": 6.15,
1056
+ "grad_norm": 2.2447423934936523,
1057
+ "learning_rate": 1.92505e-05,
1058
+ "loss": 0.0305,
1059
+ "num_input_tokens_seen": 62975232,
1060
+ "step": 61500
1061
+ },
1062
+ {
1063
+ "epoch": 6.2,
1064
+ "grad_norm": 1.8519532680511475,
1065
+ "learning_rate": 1.9000500000000002e-05,
1066
+ "loss": 0.0324,
1067
+ "num_input_tokens_seen": 63487232,
1068
+ "step": 62000
1069
+ },
1070
+ {
1071
+ "epoch": 6.25,
1072
+ "grad_norm": 1.1486473083496094,
1073
+ "learning_rate": 1.8750500000000003e-05,
1074
+ "loss": 0.0313,
1075
+ "num_input_tokens_seen": 63999232,
1076
+ "step": 62500
1077
+ },
1078
+ {
1079
+ "epoch": 6.3,
1080
+ "grad_norm": 2.004425048828125,
1081
+ "learning_rate": 1.85005e-05,
1082
+ "loss": 0.0306,
1083
+ "num_input_tokens_seen": 64511232,
1084
+ "step": 63000
1085
+ },
1086
+ {
1087
+ "epoch": 6.35,
1088
+ "grad_norm": 3.8117074966430664,
1089
+ "learning_rate": 1.82505e-05,
1090
+ "loss": 0.0326,
1091
+ "num_input_tokens_seen": 65023232,
1092
+ "step": 63500
1093
+ },
1094
+ {
1095
+ "epoch": 6.4,
1096
+ "grad_norm": 1.6470508575439453,
1097
+ "learning_rate": 1.80005e-05,
1098
+ "loss": 0.0324,
1099
+ "num_input_tokens_seen": 65535232,
1100
+ "step": 64000
1101
+ },
1102
+ {
1103
+ "epoch": 6.45,
1104
+ "grad_norm": 2.5725181102752686,
1105
+ "learning_rate": 1.77505e-05,
1106
+ "loss": 0.0324,
1107
+ "num_input_tokens_seen": 66047232,
1108
+ "step": 64500
1109
+ },
1110
+ {
1111
+ "epoch": 6.5,
1112
+ "grad_norm": 2.1077489852905273,
1113
+ "learning_rate": 1.75005e-05,
1114
+ "loss": 0.0309,
1115
+ "num_input_tokens_seen": 66559232,
1116
+ "step": 65000
1117
+ },
1118
+ {
1119
+ "epoch": 6.55,
1120
+ "grad_norm": 1.4315509796142578,
1121
+ "learning_rate": 1.72505e-05,
1122
+ "loss": 0.033,
1123
+ "num_input_tokens_seen": 67071232,
1124
+ "step": 65500
1125
+ },
1126
+ {
1127
+ "epoch": 6.6,
1128
+ "grad_norm": 1.7525889873504639,
1129
+ "learning_rate": 1.70005e-05,
1130
+ "loss": 0.0315,
1131
+ "num_input_tokens_seen": 67583232,
1132
+ "step": 66000
1133
+ },
1134
+ {
1135
+ "epoch": 6.65,
1136
+ "grad_norm": 1.3063749074935913,
1137
+ "learning_rate": 1.6750499999999998e-05,
1138
+ "loss": 0.0317,
1139
+ "num_input_tokens_seen": 68095232,
1140
+ "step": 66500
1141
+ },
1142
+ {
1143
+ "epoch": 6.7,
1144
+ "grad_norm": 1.0604239702224731,
1145
+ "learning_rate": 1.6500500000000002e-05,
1146
+ "loss": 0.0317,
1147
+ "num_input_tokens_seen": 68607232,
1148
+ "step": 67000
1149
+ },
1150
+ {
1151
+ "epoch": 6.75,
1152
+ "grad_norm": 1.3383687734603882,
1153
+ "learning_rate": 1.6250500000000003e-05,
1154
+ "loss": 0.0319,
1155
+ "num_input_tokens_seen": 69119232,
1156
+ "step": 67500
1157
+ },
1158
+ {
1159
+ "epoch": 6.8,
1160
+ "grad_norm": 1.843859314918518,
1161
+ "learning_rate": 1.60005e-05,
1162
+ "loss": 0.0323,
1163
+ "num_input_tokens_seen": 69631232,
1164
+ "step": 68000
1165
+ },
1166
+ {
1167
+ "epoch": 6.85,
1168
+ "grad_norm": 1.956433653831482,
1169
+ "learning_rate": 1.57505e-05,
1170
+ "loss": 0.0317,
1171
+ "num_input_tokens_seen": 70143232,
1172
+ "step": 68500
1173
+ },
1174
+ {
1175
+ "epoch": 6.9,
1176
+ "grad_norm": 2.344168186187744,
1177
+ "learning_rate": 1.5500500000000002e-05,
1178
+ "loss": 0.0322,
1179
+ "num_input_tokens_seen": 70655232,
1180
+ "step": 69000
1181
+ },
1182
+ {
1183
+ "epoch": 6.95,
1184
+ "grad_norm": 1.5573519468307495,
1185
+ "learning_rate": 1.52505e-05,
1186
+ "loss": 0.0338,
1187
+ "num_input_tokens_seen": 71167232,
1188
+ "step": 69500
1189
+ },
1190
+ {
1191
+ "epoch": 7.0,
1192
+ "grad_norm": 1.1548386812210083,
1193
+ "learning_rate": 1.5000500000000001e-05,
1194
+ "loss": 0.0319,
1195
+ "num_input_tokens_seen": 71679104,
1196
+ "step": 70000
1197
+ },
1198
+ {
1199
+ "epoch": 7.0,
1200
+ "eval_combined_score": 0.18516436754953325,
1201
+ "eval_loss": 0.18516436219215393,
1202
+ "eval_mse": 0.1851643729069126,
1203
+ "eval_runtime": 7.0539,
1204
+ "eval_samples_per_second": 2835.299,
1205
+ "eval_steps_per_second": 354.412,
1206
+ "num_input_tokens_seen": 71679104,
1207
+ "step": 70000
1208
+ },
1209
+ {
1210
+ "epoch": 7.05,
1211
+ "grad_norm": 1.4887698888778687,
1212
+ "learning_rate": 1.47505e-05,
1213
+ "loss": 0.0273,
1214
+ "num_input_tokens_seen": 72191104,
1215
+ "step": 70500
1216
+ },
1217
+ {
1218
+ "epoch": 7.1,
1219
+ "grad_norm": 1.1770190000534058,
1220
+ "learning_rate": 1.45005e-05,
1221
+ "loss": 0.0257,
1222
+ "num_input_tokens_seen": 72703104,
1223
+ "step": 71000
1224
+ },
1225
+ {
1226
+ "epoch": 7.15,
1227
+ "grad_norm": 0.8513312339782715,
1228
+ "learning_rate": 1.42505e-05,
1229
+ "loss": 0.0248,
1230
+ "num_input_tokens_seen": 73215104,
1231
+ "step": 71500
1232
+ },
1233
+ {
1234
+ "epoch": 7.2,
1235
+ "grad_norm": 0.743748664855957,
1236
+ "learning_rate": 1.40005e-05,
1237
+ "loss": 0.0262,
1238
+ "num_input_tokens_seen": 73727104,
1239
+ "step": 72000
1240
+ },
1241
+ {
1242
+ "epoch": 7.25,
1243
+ "grad_norm": 2.1001296043395996,
1244
+ "learning_rate": 1.37505e-05,
1245
+ "loss": 0.0251,
1246
+ "num_input_tokens_seen": 74239104,
1247
+ "step": 72500
1248
+ },
1249
+ {
1250
+ "epoch": 7.3,
1251
+ "grad_norm": 2.2133898735046387,
1252
+ "learning_rate": 1.3500499999999999e-05,
1253
+ "loss": 0.0252,
1254
+ "num_input_tokens_seen": 74751104,
1255
+ "step": 73000
1256
+ },
1257
+ {
1258
+ "epoch": 7.35,
1259
+ "grad_norm": 1.5228677988052368,
1260
+ "learning_rate": 1.3250500000000001e-05,
1261
+ "loss": 0.026,
1262
+ "num_input_tokens_seen": 75263104,
1263
+ "step": 73500
1264
+ },
1265
+ {
1266
+ "epoch": 7.4,
1267
+ "grad_norm": 1.042326807975769,
1268
+ "learning_rate": 1.3000500000000002e-05,
1269
+ "loss": 0.0259,
1270
+ "num_input_tokens_seen": 75775104,
1271
+ "step": 74000
1272
+ },
1273
+ {
1274
+ "epoch": 7.45,
1275
+ "grad_norm": 2.952548027038574,
1276
+ "learning_rate": 1.2750500000000001e-05,
1277
+ "loss": 0.0266,
1278
+ "num_input_tokens_seen": 76287104,
1279
+ "step": 74500
1280
+ },
1281
+ {
1282
+ "epoch": 7.5,
1283
+ "grad_norm": 3.490576982498169,
1284
+ "learning_rate": 1.2500500000000002e-05,
1285
+ "loss": 0.0244,
1286
+ "num_input_tokens_seen": 76799104,
1287
+ "step": 75000
1288
+ },
1289
+ {
1290
+ "epoch": 7.55,
1291
+ "grad_norm": 1.804547905921936,
1292
+ "learning_rate": 1.22505e-05,
1293
+ "loss": 0.0252,
1294
+ "num_input_tokens_seen": 77311104,
1295
+ "step": 75500
1296
+ },
1297
+ {
1298
+ "epoch": 7.6,
1299
+ "grad_norm": 2.1144022941589355,
1300
+ "learning_rate": 1.2000500000000001e-05,
1301
+ "loss": 0.0254,
1302
+ "num_input_tokens_seen": 77823104,
1303
+ "step": 76000
1304
+ },
1305
+ {
1306
+ "epoch": 7.65,
1307
+ "grad_norm": 1.1987674236297607,
1308
+ "learning_rate": 1.17505e-05,
1309
+ "loss": 0.0254,
1310
+ "num_input_tokens_seen": 78335104,
1311
+ "step": 76500
1312
+ },
1313
+ {
1314
+ "epoch": 7.7,
1315
+ "grad_norm": 2.3036508560180664,
1316
+ "learning_rate": 1.15005e-05,
1317
+ "loss": 0.0246,
1318
+ "num_input_tokens_seen": 78847104,
1319
+ "step": 77000
1320
+ },
1321
+ {
1322
+ "epoch": 7.75,
1323
+ "grad_norm": 0.9033938050270081,
1324
+ "learning_rate": 1.12505e-05,
1325
+ "loss": 0.0253,
1326
+ "num_input_tokens_seen": 79359104,
1327
+ "step": 77500
1328
+ },
1329
+ {
1330
+ "epoch": 7.8,
1331
+ "grad_norm": 1.7020654678344727,
1332
+ "learning_rate": 1.10005e-05,
1333
+ "loss": 0.0249,
1334
+ "num_input_tokens_seen": 79871104,
1335
+ "step": 78000
1336
+ },
1337
+ {
1338
+ "epoch": 7.85,
1339
+ "grad_norm": 3.093097448348999,
1340
+ "learning_rate": 1.0750500000000002e-05,
1341
+ "loss": 0.0254,
1342
+ "num_input_tokens_seen": 80383104,
1343
+ "step": 78500
1344
+ },
1345
+ {
1346
+ "epoch": 7.9,
1347
+ "grad_norm": 1.5874762535095215,
1348
+ "learning_rate": 1.05005e-05,
1349
+ "loss": 0.0264,
1350
+ "num_input_tokens_seen": 80895104,
1351
+ "step": 79000
1352
+ },
1353
+ {
1354
+ "epoch": 7.95,
1355
+ "grad_norm": 1.178058385848999,
1356
+ "learning_rate": 1.0250500000000001e-05,
1357
+ "loss": 0.0252,
1358
+ "num_input_tokens_seen": 81407104,
1359
+ "step": 79500
1360
+ },
1361
+ {
1362
+ "epoch": 8.0,
1363
+ "grad_norm": 2.111694097518921,
1364
+ "learning_rate": 1.00005e-05,
1365
+ "loss": 0.0266,
1366
+ "num_input_tokens_seen": 81918976,
1367
+ "step": 80000
1368
+ },
1369
+ {
1370
+ "epoch": 8.0,
1371
+ "eval_combined_score": 0.18399447017391748,
1372
+ "eval_loss": 0.18399447202682495,
1373
+ "eval_mse": 0.18399446832101002,
1374
+ "eval_runtime": 7.0282,
1375
+ "eval_samples_per_second": 2845.677,
1376
+ "eval_steps_per_second": 355.71,
1377
+ "num_input_tokens_seen": 81918976,
1378
+ "step": 80000
1379
+ },
1380
+ {
1381
+ "epoch": 8.05,
1382
+ "grad_norm": 1.775572419166565,
1383
+ "learning_rate": 9.7505e-06,
1384
+ "loss": 0.0216,
1385
+ "num_input_tokens_seen": 82430976,
1386
+ "step": 80500
1387
+ },
1388
+ {
1389
+ "epoch": 8.1,
1390
+ "grad_norm": 2.352768659591675,
1391
+ "learning_rate": 9.500500000000002e-06,
1392
+ "loss": 0.0204,
1393
+ "num_input_tokens_seen": 82942976,
1394
+ "step": 81000
1395
+ },
1396
+ {
1397
+ "epoch": 8.15,
1398
+ "grad_norm": 1.497861623764038,
1399
+ "learning_rate": 9.2505e-06,
1400
+ "loss": 0.0214,
1401
+ "num_input_tokens_seen": 83454976,
1402
+ "step": 81500
1403
+ },
1404
+ {
1405
+ "epoch": 8.2,
1406
+ "grad_norm": 2.0861403942108154,
1407
+ "learning_rate": 9.000500000000001e-06,
1408
+ "loss": 0.0203,
1409
+ "num_input_tokens_seen": 83966976,
1410
+ "step": 82000
1411
+ },
1412
+ {
1413
+ "epoch": 8.25,
1414
+ "grad_norm": 2.6044728755950928,
1415
+ "learning_rate": 8.7505e-06,
1416
+ "loss": 0.0209,
1417
+ "num_input_tokens_seen": 84478976,
1418
+ "step": 82500
1419
+ },
1420
+ {
1421
+ "epoch": 8.3,
1422
+ "grad_norm": 1.7525115013122559,
1423
+ "learning_rate": 8.5005e-06,
1424
+ "loss": 0.0213,
1425
+ "num_input_tokens_seen": 84990976,
1426
+ "step": 83000
1427
+ },
1428
+ {
1429
+ "epoch": 8.35,
1430
+ "grad_norm": 5.014759540557861,
1431
+ "learning_rate": 8.2505e-06,
1432
+ "loss": 0.0215,
1433
+ "num_input_tokens_seen": 85502976,
1434
+ "step": 83500
1435
+ },
1436
+ {
1437
+ "epoch": 8.4,
1438
+ "grad_norm": 1.2806963920593262,
1439
+ "learning_rate": 8.0005e-06,
1440
+ "loss": 0.0208,
1441
+ "num_input_tokens_seen": 86014976,
1442
+ "step": 84000
1443
+ },
1444
+ {
1445
+ "epoch": 8.45,
1446
+ "grad_norm": 1.6682835817337036,
1447
+ "learning_rate": 7.750500000000001e-06,
1448
+ "loss": 0.0208,
1449
+ "num_input_tokens_seen": 86526976,
1450
+ "step": 84500
1451
+ },
1452
+ {
1453
+ "epoch": 8.5,
1454
+ "grad_norm": 2.586468458175659,
1455
+ "learning_rate": 7.5005000000000004e-06,
1456
+ "loss": 0.0213,
1457
+ "num_input_tokens_seen": 87038976,
1458
+ "step": 85000
1459
+ },
1460
+ {
1461
+ "epoch": 8.55,
1462
+ "grad_norm": 1.1163995265960693,
1463
+ "learning_rate": 7.2505e-06,
1464
+ "loss": 0.021,
1465
+ "num_input_tokens_seen": 87550976,
1466
+ "step": 85500
1467
+ },
1468
+ {
1469
+ "epoch": 8.6,
1470
+ "grad_norm": 1.3840800523757935,
1471
+ "learning_rate": 7.0005e-06,
1472
+ "loss": 0.0217,
1473
+ "num_input_tokens_seen": 88062976,
1474
+ "step": 86000
1475
+ },
1476
+ {
1477
+ "epoch": 8.65,
1478
+ "grad_norm": 2.503617763519287,
1479
+ "learning_rate": 6.7505e-06,
1480
+ "loss": 0.021,
1481
+ "num_input_tokens_seen": 88574976,
1482
+ "step": 86500
1483
+ },
1484
+ {
1485
+ "epoch": 8.7,
1486
+ "grad_norm": 0.9129466414451599,
1487
+ "learning_rate": 6.5005e-06,
1488
+ "loss": 0.0217,
1489
+ "num_input_tokens_seen": 89086976,
1490
+ "step": 87000
1491
+ },
1492
+ {
1493
+ "epoch": 8.75,
1494
+ "grad_norm": 0.9430443644523621,
1495
+ "learning_rate": 6.2505000000000005e-06,
1496
+ "loss": 0.0205,
1497
+ "num_input_tokens_seen": 89598976,
1498
+ "step": 87500
1499
+ },
1500
+ {
1501
+ "epoch": 8.8,
1502
+ "grad_norm": 1.8605977296829224,
1503
+ "learning_rate": 6.0005e-06,
1504
+ "loss": 0.0213,
1505
+ "num_input_tokens_seen": 90110976,
1506
+ "step": 88000
1507
+ },
1508
+ {
1509
+ "epoch": 8.85,
1510
+ "grad_norm": 1.7093037366867065,
1511
+ "learning_rate": 5.7505e-06,
1512
+ "loss": 0.0206,
1513
+ "num_input_tokens_seen": 90622976,
1514
+ "step": 88500
1515
+ },
1516
+ {
1517
+ "epoch": 8.9,
1518
+ "grad_norm": 1.1300668716430664,
1519
+ "learning_rate": 5.5005e-06,
1520
+ "loss": 0.0199,
1521
+ "num_input_tokens_seen": 91134976,
1522
+ "step": 89000
1523
+ },
1524
+ {
1525
+ "epoch": 8.95,
1526
+ "grad_norm": 1.529173493385315,
1527
+ "learning_rate": 5.250500000000001e-06,
1528
+ "loss": 0.0213,
1529
+ "num_input_tokens_seen": 91646976,
1530
+ "step": 89500
1531
+ },
1532
+ {
1533
+ "epoch": 9.0,
1534
+ "grad_norm": 2.0707693099975586,
1535
+ "learning_rate": 5.000500000000001e-06,
1536
+ "loss": 0.0202,
1537
+ "num_input_tokens_seen": 92158848,
1538
+ "step": 90000
1539
+ },
1540
+ {
1541
+ "epoch": 9.0,
1542
+ "eval_combined_score": 0.17236512670393594,
1543
+ "eval_loss": 0.17236514389514923,
1544
+ "eval_mse": 0.17236512441388385,
1545
+ "eval_runtime": 7.0087,
1546
+ "eval_samples_per_second": 2853.589,
1547
+ "eval_steps_per_second": 356.699,
1548
+ "num_input_tokens_seen": 92158848,
1549
+ "step": 90000
1550
+ },
1551
+ {
1552
+ "epoch": 9.05,
1553
+ "grad_norm": 2.7466800212860107,
1554
+ "learning_rate": 4.7505000000000005e-06,
1555
+ "loss": 0.0175,
1556
+ "num_input_tokens_seen": 92670848,
1557
+ "step": 90500
1558
+ },
1559
+ {
1560
+ "epoch": 9.1,
1561
+ "grad_norm": 2.3883421421051025,
1562
+ "learning_rate": 4.5005e-06,
1563
+ "loss": 0.0183,
1564
+ "num_input_tokens_seen": 93182848,
1565
+ "step": 91000
1566
+ },
1567
+ {
1568
+ "epoch": 9.15,
1569
+ "grad_norm": 2.6846847534179688,
1570
+ "learning_rate": 4.2505e-06,
1571
+ "loss": 0.0187,
1572
+ "num_input_tokens_seen": 93694848,
1573
+ "step": 91500
1574
+ },
1575
+ {
1576
+ "epoch": 9.2,
1577
+ "grad_norm": 1.8421474695205688,
1578
+ "learning_rate": 4.0005e-06,
1579
+ "loss": 0.018,
1580
+ "num_input_tokens_seen": 94206848,
1581
+ "step": 92000
1582
+ },
1583
+ {
1584
+ "epoch": 9.25,
1585
+ "grad_norm": 1.388693928718567,
1586
+ "learning_rate": 3.7505e-06,
1587
+ "loss": 0.0181,
1588
+ "num_input_tokens_seen": 94718848,
1589
+ "step": 92500
1590
+ },
1591
+ {
1592
+ "epoch": 9.3,
1593
+ "grad_norm": 1.4700050354003906,
1594
+ "learning_rate": 3.5005e-06,
1595
+ "loss": 0.0177,
1596
+ "num_input_tokens_seen": 95230848,
1597
+ "step": 93000
1598
+ },
1599
+ {
1600
+ "epoch": 9.35,
1601
+ "grad_norm": 0.908022940158844,
1602
+ "learning_rate": 3.2505e-06,
1603
+ "loss": 0.0182,
1604
+ "num_input_tokens_seen": 95742848,
1605
+ "step": 93500
1606
+ },
1607
+ {
1608
+ "epoch": 9.4,
1609
+ "grad_norm": 1.2627387046813965,
1610
+ "learning_rate": 3.0005000000000003e-06,
1611
+ "loss": 0.0178,
1612
+ "num_input_tokens_seen": 96254848,
1613
+ "step": 94000
1614
+ },
1615
+ {
1616
+ "epoch": 9.45,
1617
+ "grad_norm": 0.44914329051971436,
1618
+ "learning_rate": 2.7505e-06,
1619
+ "loss": 0.0173,
1620
+ "num_input_tokens_seen": 96766848,
1621
+ "step": 94500
1622
+ },
1623
+ {
1624
+ "epoch": 9.5,
1625
+ "grad_norm": 0.689420223236084,
1626
+ "learning_rate": 2.5005e-06,
1627
+ "loss": 0.0181,
1628
+ "num_input_tokens_seen": 97278848,
1629
+ "step": 95000
1630
+ },
1631
+ {
1632
+ "epoch": 9.55,
1633
+ "grad_norm": 1.2363290786743164,
1634
+ "learning_rate": 2.2505000000000003e-06,
1635
+ "loss": 0.0173,
1636
+ "num_input_tokens_seen": 97790848,
1637
+ "step": 95500
1638
+ },
1639
+ {
1640
+ "epoch": 9.6,
1641
+ "grad_norm": 0.676801323890686,
1642
+ "learning_rate": 2.0004999999999997e-06,
1643
+ "loss": 0.0178,
1644
+ "num_input_tokens_seen": 98302848,
1645
+ "step": 96000
1646
+ },
1647
+ {
1648
+ "epoch": 9.65,
1649
+ "grad_norm": 0.6555656790733337,
1650
+ "learning_rate": 1.7505e-06,
1651
+ "loss": 0.0179,
1652
+ "num_input_tokens_seen": 98814848,
1653
+ "step": 96500
1654
+ },
1655
+ {
1656
+ "epoch": 9.7,
1657
+ "grad_norm": 0.6341429948806763,
1658
+ "learning_rate": 1.5005e-06,
1659
+ "loss": 0.018,
1660
+ "num_input_tokens_seen": 99326848,
1661
+ "step": 97000
1662
+ },
1663
+ {
1664
+ "epoch": 9.75,
1665
+ "grad_norm": 0.9721384644508362,
1666
+ "learning_rate": 1.2505000000000001e-06,
1667
+ "loss": 0.0179,
1668
+ "num_input_tokens_seen": 99838848,
1669
+ "step": 97500
1670
+ },
1671
+ {
1672
+ "epoch": 9.8,
1673
+ "grad_norm": 1.3599611520767212,
1674
+ "learning_rate": 1.0005e-06,
1675
+ "loss": 0.0179,
1676
+ "num_input_tokens_seen": 100350848,
1677
+ "step": 98000
1678
+ },
1679
+ {
1680
+ "epoch": 9.85,
1681
+ "grad_norm": 1.1056464910507202,
1682
+ "learning_rate": 7.505000000000001e-07,
1683
+ "loss": 0.017,
1684
+ "num_input_tokens_seen": 100862848,
1685
+ "step": 98500
1686
+ },
1687
+ {
1688
+ "epoch": 9.9,
1689
+ "grad_norm": 1.6119426488876343,
1690
+ "learning_rate": 5.005e-07,
1691
+ "loss": 0.0184,
1692
+ "num_input_tokens_seen": 101374848,
1693
+ "step": 99000
1694
+ },
1695
+ {
1696
+ "epoch": 9.95,
1697
+ "grad_norm": 1.2073465585708618,
1698
+ "learning_rate": 2.5049999999999997e-07,
1699
+ "loss": 0.0185,
1700
+ "num_input_tokens_seen": 101886848,
1701
+ "step": 99500
1702
+ },
1703
+ {
1704
+ "epoch": 10.0,
1705
+ "grad_norm": 0.7742397785186768,
1706
+ "learning_rate": 5e-10,
1707
+ "loss": 0.0172,
1708
+ "num_input_tokens_seen": 102398720,
1709
+ "step": 100000
1710
+ },
1711
+ {
1712
+ "epoch": 10.0,
1713
+ "eval_combined_score": 0.17306020632561403,
1714
+ "eval_loss": 0.17306020855903625,
1715
+ "eval_mse": 0.1730602040921918,
1716
+ "eval_runtime": 7.0331,
1717
+ "eval_samples_per_second": 2843.716,
1718
+ "eval_steps_per_second": 355.465,
1719
+ "num_input_tokens_seen": 102398720,
1720
+ "step": 100000
1721
+ },
1722
+ {
1723
+ "epoch": 10.0,
1724
+ "num_input_tokens_seen": 102398720,
1725
+ "step": 100000,
1726
+ "total_flos": 6632842477232640.0,
1727
+ "train_loss": 0.06995858280181885,
1728
+ "train_runtime": 1201.7215,
1729
+ "train_samples_per_second": 665.703,
1730
+ "train_steps_per_second": 83.214,
1731
+ "train_tokens_per_second": 85210.024
1732
+ }
1733
+ ],
1734
+ "logging_steps": 500,
1735
+ "max_steps": 100000,
1736
+ "num_input_tokens_seen": 102398720,
1737
+ "num_train_epochs": 10,
1738
+ "save_steps": 500,
1739
+ "stateful_callbacks": {
1740
+ "TrainerControl": {
1741
+ "args": {
1742
+ "should_epoch_stop": false,
1743
+ "should_evaluate": false,
1744
+ "should_log": false,
1745
+ "should_save": true,
1746
+ "should_training_stop": true
1747
+ },
1748
+ "attributes": {}
1749
+ }
1750
+ },
1751
+ "total_flos": 6632842477232640.0,
1752
+ "train_batch_size": 8,
1753
+ "trial_name": null,
1754
+ "trial_params": null
1755
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f18daaeb0283bb83b1225fbc87dd8f7828a3b5524a0331d6e110ca4f9f22158
3
+ size 5432
vocab.txt ADDED
The diff for this file is too large to render. See raw diff