Text Classification
Transformers
PyTorch
English
electra
reward-model
reward_model
RLHF
Inference Endpoints
theblackcat102 commited on
Commit
ae6162c
1 Parent(s): 5e83f9b

Upload 12 files

Browse files
config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/electra-large-discriminator",
3
+ "architectures": [
4
+ "ElectraForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "classifier_dropout": null,
8
+ "embedding_size": 1024,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.1,
11
+ "hidden_size": 1024,
12
+ "id2label": {
13
+ "0": "REWARD"
14
+ },
15
+ "initializer_range": 0.02,
16
+ "intermediate_size": 4096,
17
+ "label2id": {
18
+ "REWARD": 0
19
+ },
20
+ "layer_norm_eps": 1e-12,
21
+ "max_position_embeddings": 512,
22
+ "model_type": "electra",
23
+ "num_attention_heads": 16,
24
+ "num_hidden_layers": 24,
25
+ "pad_token_id": 0,
26
+ "position_embedding_type": "absolute",
27
+ "problem_type": "regression",
28
+ "summary_activation": "gelu",
29
+ "summary_last_dropout": 0.1,
30
+ "summary_type": "first",
31
+ "summary_use_proj": true,
32
+ "torch_dtype": "float32",
33
+ "transformers_version": "4.25.1",
34
+ "type_vocab_size": 2,
35
+ "use_cache": true,
36
+ "vocab_size": 30522
37
+ }
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e7e4dc7571cc17c1f3682817d0f6297d9e0f55ebd67a0d8aca5659a24b2b6a59
3
+ size 2681374085
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:00fffef793100af475612223b133b0a85ad8acad886e8cf02a57348da1ab8b43
3
+ size 1340711669
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a78052beadb01ea9132e5f501fed142e67e1b427c11888cab7b6d2eea4386adf
3
+ size 15597
scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:80582f4346a0c51af79b5d61f3bc54485b7bf7ebf0705dff919e88a2659194b3
3
+ size 557
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:67987ee0bd16058a18c5fe7c0db5728da0deeb21df656c8f2e55fb6b2edb3b4e
3
+ size 563
special_tokens_map.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "cls_token": "[CLS]",
3
+ "mask_token": "[MASK]",
4
+ "pad_token": "[PAD]",
5
+ "sep_token": "[SEP]",
6
+ "unk_token": "[UNK]"
7
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "cls_token": "[CLS]",
3
+ "do_lower_case": true,
4
+ "mask_token": "[MASK]",
5
+ "model_max_length": 512,
6
+ "name_or_path": "google/electra-large-discriminator",
7
+ "pad_token": "[PAD]",
8
+ "sep_token": "[SEP]",
9
+ "special_tokens_map_file": null,
10
+ "strip_accents": null,
11
+ "tokenize_chinese_chars": true,
12
+ "tokenizer_class": "ElectraTokenizer",
13
+ "unk_token": "[UNK]"
14
+ }
trainer_state.json ADDED
@@ -0,0 +1,2848 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 1.8507242605603262,
5
+ "global_step": 4000,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.0,
12
+ "learning_rate": 1.62e-07,
13
+ "loss": 0.6957,
14
+ "step": 10
15
+ },
16
+ {
17
+ "epoch": 0.01,
18
+ "learning_rate": 3.42e-07,
19
+ "loss": 0.6958,
20
+ "step": 20
21
+ },
22
+ {
23
+ "epoch": 0.01,
24
+ "learning_rate": 5.22e-07,
25
+ "loss": 0.6916,
26
+ "step": 30
27
+ },
28
+ {
29
+ "epoch": 0.02,
30
+ "learning_rate": 7.02e-07,
31
+ "loss": 0.6959,
32
+ "step": 40
33
+ },
34
+ {
35
+ "epoch": 0.02,
36
+ "learning_rate": 8.820000000000001e-07,
37
+ "loss": 0.6927,
38
+ "step": 50
39
+ },
40
+ {
41
+ "epoch": 0.03,
42
+ "learning_rate": 1.062e-06,
43
+ "loss": 0.6921,
44
+ "step": 60
45
+ },
46
+ {
47
+ "epoch": 0.03,
48
+ "learning_rate": 1.2420000000000001e-06,
49
+ "loss": 0.691,
50
+ "step": 70
51
+ },
52
+ {
53
+ "epoch": 0.04,
54
+ "learning_rate": 1.4220000000000001e-06,
55
+ "loss": 0.6892,
56
+ "step": 80
57
+ },
58
+ {
59
+ "epoch": 0.04,
60
+ "learning_rate": 1.6019999999999999e-06,
61
+ "loss": 0.6881,
62
+ "step": 90
63
+ },
64
+ {
65
+ "epoch": 0.05,
66
+ "learning_rate": 1.782e-06,
67
+ "loss": 0.6857,
68
+ "step": 100
69
+ },
70
+ {
71
+ "epoch": 0.05,
72
+ "learning_rate": 1.962e-06,
73
+ "loss": 0.6836,
74
+ "step": 110
75
+ },
76
+ {
77
+ "epoch": 0.06,
78
+ "learning_rate": 2.142e-06,
79
+ "loss": 0.6802,
80
+ "step": 120
81
+ },
82
+ {
83
+ "epoch": 0.06,
84
+ "learning_rate": 2.322e-06,
85
+ "loss": 0.6714,
86
+ "step": 130
87
+ },
88
+ {
89
+ "epoch": 0.06,
90
+ "learning_rate": 2.4840000000000003e-06,
91
+ "loss": 0.6661,
92
+ "step": 140
93
+ },
94
+ {
95
+ "epoch": 0.07,
96
+ "learning_rate": 2.664e-06,
97
+ "loss": 0.6599,
98
+ "step": 150
99
+ },
100
+ {
101
+ "epoch": 0.07,
102
+ "learning_rate": 2.826e-06,
103
+ "loss": 0.6524,
104
+ "step": 160
105
+ },
106
+ {
107
+ "epoch": 0.08,
108
+ "learning_rate": 3.006e-06,
109
+ "loss": 0.6362,
110
+ "step": 170
111
+ },
112
+ {
113
+ "epoch": 0.08,
114
+ "learning_rate": 3.168e-06,
115
+ "loss": 0.632,
116
+ "step": 180
117
+ },
118
+ {
119
+ "epoch": 0.09,
120
+ "learning_rate": 3.348e-06,
121
+ "loss": 0.6301,
122
+ "step": 190
123
+ },
124
+ {
125
+ "epoch": 0.09,
126
+ "learning_rate": 3.5280000000000004e-06,
127
+ "loss": 0.6272,
128
+ "step": 200
129
+ },
130
+ {
131
+ "epoch": 0.1,
132
+ "learning_rate": 3.708e-06,
133
+ "loss": 0.6172,
134
+ "step": 210
135
+ },
136
+ {
137
+ "epoch": 0.1,
138
+ "learning_rate": 3.87e-06,
139
+ "loss": 0.6146,
140
+ "step": 220
141
+ },
142
+ {
143
+ "epoch": 0.11,
144
+ "learning_rate": 4.05e-06,
145
+ "loss": 0.606,
146
+ "step": 230
147
+ },
148
+ {
149
+ "epoch": 0.11,
150
+ "learning_rate": 4.23e-06,
151
+ "loss": 0.6194,
152
+ "step": 240
153
+ },
154
+ {
155
+ "epoch": 0.12,
156
+ "learning_rate": 4.392e-06,
157
+ "loss": 0.5863,
158
+ "step": 250
159
+ },
160
+ {
161
+ "epoch": 0.12,
162
+ "eval_webgpt_accuracy": 0.547752808988764,
163
+ "eval_webgpt_loss": 0.7199520468711853,
164
+ "eval_webgpt_runtime": 152.9037,
165
+ "eval_webgpt_samples_per_second": 25.611,
166
+ "eval_webgpt_steps_per_second": 2.564,
167
+ "step": 250
168
+ },
169
+ {
170
+ "epoch": 0.12,
171
+ "eval_hfsummary_accuracy": 0.630263277211861,
172
+ "eval_hfsummary_loss": 0.6598580479621887,
173
+ "eval_hfsummary_runtime": 2471.5424,
174
+ "eval_hfsummary_samples_per_second": 13.386,
175
+ "eval_hfsummary_steps_per_second": 1.339,
176
+ "step": 250
177
+ },
178
+ {
179
+ "epoch": 0.12,
180
+ "eval_gptsynthetic_accuracy": 0.9978883861236802,
181
+ "eval_gptsynthetic_loss": 0.09524902701377869,
182
+ "eval_gptsynthetic_runtime": 116.7014,
183
+ "eval_gptsynthetic_samples_per_second": 28.406,
184
+ "eval_gptsynthetic_steps_per_second": 2.845,
185
+ "step": 250
186
+ },
187
+ {
188
+ "epoch": 0.12,
189
+ "learning_rate": 4.5720000000000004e-06,
190
+ "loss": 0.5818,
191
+ "step": 260
192
+ },
193
+ {
194
+ "epoch": 0.12,
195
+ "learning_rate": 4.752e-06,
196
+ "loss": 0.5817,
197
+ "step": 270
198
+ },
199
+ {
200
+ "epoch": 0.13,
201
+ "learning_rate": 4.932e-06,
202
+ "loss": 0.5713,
203
+ "step": 280
204
+ },
205
+ {
206
+ "epoch": 0.13,
207
+ "learning_rate": 5.1119999999999995e-06,
208
+ "loss": 0.5787,
209
+ "step": 290
210
+ },
211
+ {
212
+ "epoch": 0.14,
213
+ "learning_rate": 5.2919999999999995e-06,
214
+ "loss": 0.566,
215
+ "step": 300
216
+ },
217
+ {
218
+ "epoch": 0.14,
219
+ "learning_rate": 5.472e-06,
220
+ "loss": 0.5765,
221
+ "step": 310
222
+ },
223
+ {
224
+ "epoch": 0.15,
225
+ "learning_rate": 5.652e-06,
226
+ "loss": 0.5286,
227
+ "step": 320
228
+ },
229
+ {
230
+ "epoch": 0.15,
231
+ "learning_rate": 5.814000000000001e-06,
232
+ "loss": 0.5644,
233
+ "step": 330
234
+ },
235
+ {
236
+ "epoch": 0.16,
237
+ "learning_rate": 5.9940000000000005e-06,
238
+ "loss": 0.5278,
239
+ "step": 340
240
+ },
241
+ {
242
+ "epoch": 0.16,
243
+ "learning_rate": 6.1740000000000005e-06,
244
+ "loss": 0.5376,
245
+ "step": 350
246
+ },
247
+ {
248
+ "epoch": 0.17,
249
+ "learning_rate": 6.354e-06,
250
+ "loss": 0.5396,
251
+ "step": 360
252
+ },
253
+ {
254
+ "epoch": 0.17,
255
+ "learning_rate": 6.534e-06,
256
+ "loss": 0.517,
257
+ "step": 370
258
+ },
259
+ {
260
+ "epoch": 0.18,
261
+ "learning_rate": 6.7140000000000004e-06,
262
+ "loss": 0.5001,
263
+ "step": 380
264
+ },
265
+ {
266
+ "epoch": 0.18,
267
+ "learning_rate": 6.894e-06,
268
+ "loss": 0.5069,
269
+ "step": 390
270
+ },
271
+ {
272
+ "epoch": 0.19,
273
+ "learning_rate": 7.074e-06,
274
+ "loss": 0.5102,
275
+ "step": 400
276
+ },
277
+ {
278
+ "epoch": 0.19,
279
+ "learning_rate": 7.254e-06,
280
+ "loss": 0.5105,
281
+ "step": 410
282
+ },
283
+ {
284
+ "epoch": 0.19,
285
+ "learning_rate": 7.4339999999999995e-06,
286
+ "loss": 0.5332,
287
+ "step": 420
288
+ },
289
+ {
290
+ "epoch": 0.2,
291
+ "learning_rate": 7.614e-06,
292
+ "loss": 0.5076,
293
+ "step": 430
294
+ },
295
+ {
296
+ "epoch": 0.2,
297
+ "learning_rate": 7.794e-06,
298
+ "loss": 0.5088,
299
+ "step": 440
300
+ },
301
+ {
302
+ "epoch": 0.21,
303
+ "learning_rate": 7.974e-06,
304
+ "loss": 0.4968,
305
+ "step": 450
306
+ },
307
+ {
308
+ "epoch": 0.21,
309
+ "learning_rate": 8.154e-06,
310
+ "loss": 0.4902,
311
+ "step": 460
312
+ },
313
+ {
314
+ "epoch": 0.22,
315
+ "learning_rate": 8.334e-06,
316
+ "loss": 0.5097,
317
+ "step": 470
318
+ },
319
+ {
320
+ "epoch": 0.22,
321
+ "learning_rate": 8.514e-06,
322
+ "loss": 0.5038,
323
+ "step": 480
324
+ },
325
+ {
326
+ "epoch": 0.23,
327
+ "learning_rate": 8.694e-06,
328
+ "loss": 0.5082,
329
+ "step": 490
330
+ },
331
+ {
332
+ "epoch": 0.23,
333
+ "learning_rate": 8.874e-06,
334
+ "loss": 0.4887,
335
+ "step": 500
336
+ },
337
+ {
338
+ "epoch": 0.23,
339
+ "eval_webgpt_accuracy": 0.5623084780388151,
340
+ "eval_webgpt_loss": 0.7031316161155701,
341
+ "eval_webgpt_runtime": 152.8509,
342
+ "eval_webgpt_samples_per_second": 25.62,
343
+ "eval_webgpt_steps_per_second": 2.565,
344
+ "step": 500
345
+ },
346
+ {
347
+ "epoch": 0.23,
348
+ "eval_hfsummary_accuracy": 0.6744249312335641,
349
+ "eval_hfsummary_loss": 0.6107630729675293,
350
+ "eval_hfsummary_runtime": 2471.6908,
351
+ "eval_hfsummary_samples_per_second": 13.385,
352
+ "eval_hfsummary_steps_per_second": 1.339,
353
+ "step": 500
354
+ },
355
+ {
356
+ "epoch": 0.23,
357
+ "eval_gptsynthetic_accuracy": 0.9987933634992459,
358
+ "eval_gptsynthetic_loss": 0.023942505940794945,
359
+ "eval_gptsynthetic_runtime": 116.8138,
360
+ "eval_gptsynthetic_samples_per_second": 28.379,
361
+ "eval_gptsynthetic_steps_per_second": 2.842,
362
+ "step": 500
363
+ },
364
+ {
365
+ "epoch": 0.24,
366
+ "learning_rate": 8.99293563579278e-06,
367
+ "loss": 0.4875,
368
+ "step": 510
369
+ },
370
+ {
371
+ "epoch": 0.24,
372
+ "learning_rate": 8.969387755102041e-06,
373
+ "loss": 0.483,
374
+ "step": 520
375
+ },
376
+ {
377
+ "epoch": 0.25,
378
+ "learning_rate": 8.945839874411304e-06,
379
+ "loss": 0.5057,
380
+ "step": 530
381
+ },
382
+ {
383
+ "epoch": 0.25,
384
+ "learning_rate": 8.922291993720566e-06,
385
+ "loss": 0.4923,
386
+ "step": 540
387
+ },
388
+ {
389
+ "epoch": 0.25,
390
+ "learning_rate": 8.898744113029827e-06,
391
+ "loss": 0.4623,
392
+ "step": 550
393
+ },
394
+ {
395
+ "epoch": 0.26,
396
+ "learning_rate": 8.87519623233909e-06,
397
+ "loss": 0.4735,
398
+ "step": 560
399
+ },
400
+ {
401
+ "epoch": 0.26,
402
+ "learning_rate": 8.851648351648352e-06,
403
+ "loss": 0.4516,
404
+ "step": 570
405
+ },
406
+ {
407
+ "epoch": 0.27,
408
+ "learning_rate": 8.828100470957614e-06,
409
+ "loss": 0.4542,
410
+ "step": 580
411
+ },
412
+ {
413
+ "epoch": 0.27,
414
+ "learning_rate": 8.804552590266875e-06,
415
+ "loss": 0.465,
416
+ "step": 590
417
+ },
418
+ {
419
+ "epoch": 0.28,
420
+ "learning_rate": 8.781004709576138e-06,
421
+ "loss": 0.4577,
422
+ "step": 600
423
+ },
424
+ {
425
+ "epoch": 0.28,
426
+ "learning_rate": 8.7574568288854e-06,
427
+ "loss": 0.4994,
428
+ "step": 610
429
+ },
430
+ {
431
+ "epoch": 0.29,
432
+ "learning_rate": 8.733908948194663e-06,
433
+ "loss": 0.4565,
434
+ "step": 620
435
+ },
436
+ {
437
+ "epoch": 0.29,
438
+ "learning_rate": 8.710361067503925e-06,
439
+ "loss": 0.4617,
440
+ "step": 630
441
+ },
442
+ {
443
+ "epoch": 0.3,
444
+ "learning_rate": 8.686813186813188e-06,
445
+ "loss": 0.4598,
446
+ "step": 640
447
+ },
448
+ {
449
+ "epoch": 0.3,
450
+ "learning_rate": 8.663265306122449e-06,
451
+ "loss": 0.4441,
452
+ "step": 650
453
+ },
454
+ {
455
+ "epoch": 0.31,
456
+ "learning_rate": 8.639717425431711e-06,
457
+ "loss": 0.4935,
458
+ "step": 660
459
+ },
460
+ {
461
+ "epoch": 0.31,
462
+ "learning_rate": 8.616169544740974e-06,
463
+ "loss": 0.4646,
464
+ "step": 670
465
+ },
466
+ {
467
+ "epoch": 0.31,
468
+ "learning_rate": 8.592621664050236e-06,
469
+ "loss": 0.4771,
470
+ "step": 680
471
+ },
472
+ {
473
+ "epoch": 0.32,
474
+ "learning_rate": 8.569073783359497e-06,
475
+ "loss": 0.4754,
476
+ "step": 690
477
+ },
478
+ {
479
+ "epoch": 0.32,
480
+ "learning_rate": 8.54552590266876e-06,
481
+ "loss": 0.4684,
482
+ "step": 700
483
+ },
484
+ {
485
+ "epoch": 0.33,
486
+ "learning_rate": 8.521978021978022e-06,
487
+ "loss": 0.4691,
488
+ "step": 710
489
+ },
490
+ {
491
+ "epoch": 0.33,
492
+ "learning_rate": 8.498430141287285e-06,
493
+ "loss": 0.4807,
494
+ "step": 720
495
+ },
496
+ {
497
+ "epoch": 0.34,
498
+ "learning_rate": 8.474882260596547e-06,
499
+ "loss": 0.4597,
500
+ "step": 730
501
+ },
502
+ {
503
+ "epoch": 0.34,
504
+ "learning_rate": 8.45133437990581e-06,
505
+ "loss": 0.455,
506
+ "step": 740
507
+ },
508
+ {
509
+ "epoch": 0.35,
510
+ "learning_rate": 8.42778649921507e-06,
511
+ "loss": 0.4561,
512
+ "step": 750
513
+ },
514
+ {
515
+ "epoch": 0.35,
516
+ "eval_webgpt_accuracy": 0.5702247191011236,
517
+ "eval_webgpt_loss": 0.6994287371635437,
518
+ "eval_webgpt_runtime": 152.9386,
519
+ "eval_webgpt_samples_per_second": 25.605,
520
+ "eval_webgpt_steps_per_second": 2.563,
521
+ "step": 750
522
+ },
523
+ {
524
+ "epoch": 0.35,
525
+ "eval_hfsummary_accuracy": 0.6827373575552399,
526
+ "eval_hfsummary_loss": 0.60725337266922,
527
+ "eval_hfsummary_runtime": 2470.4498,
528
+ "eval_hfsummary_samples_per_second": 13.391,
529
+ "eval_hfsummary_steps_per_second": 1.339,
530
+ "step": 750
531
+ },
532
+ {
533
+ "epoch": 0.35,
534
+ "eval_gptsynthetic_accuracy": 0.9975867269984917,
535
+ "eval_gptsynthetic_loss": 0.01110268384218216,
536
+ "eval_gptsynthetic_runtime": 116.8749,
537
+ "eval_gptsynthetic_samples_per_second": 28.364,
538
+ "eval_gptsynthetic_steps_per_second": 2.841,
539
+ "step": 750
540
+ },
541
+ {
542
+ "epoch": 0.35,
543
+ "learning_rate": 8.404238618524333e-06,
544
+ "loss": 0.4578,
545
+ "step": 760
546
+ },
547
+ {
548
+ "epoch": 0.36,
549
+ "learning_rate": 8.380690737833595e-06,
550
+ "loss": 0.4798,
551
+ "step": 770
552
+ },
553
+ {
554
+ "epoch": 0.36,
555
+ "learning_rate": 8.359497645211932e-06,
556
+ "loss": 0.4402,
557
+ "step": 780
558
+ },
559
+ {
560
+ "epoch": 0.37,
561
+ "learning_rate": 8.335949764521193e-06,
562
+ "loss": 0.4632,
563
+ "step": 790
564
+ },
565
+ {
566
+ "epoch": 0.37,
567
+ "learning_rate": 8.312401883830455e-06,
568
+ "loss": 0.4505,
569
+ "step": 800
570
+ },
571
+ {
572
+ "epoch": 0.37,
573
+ "learning_rate": 8.288854003139718e-06,
574
+ "loss": 0.4641,
575
+ "step": 810
576
+ },
577
+ {
578
+ "epoch": 0.38,
579
+ "learning_rate": 8.26530612244898e-06,
580
+ "loss": 0.4373,
581
+ "step": 820
582
+ },
583
+ {
584
+ "epoch": 0.38,
585
+ "learning_rate": 8.241758241758241e-06,
586
+ "loss": 0.4192,
587
+ "step": 830
588
+ },
589
+ {
590
+ "epoch": 0.39,
591
+ "learning_rate": 8.218210361067504e-06,
592
+ "loss": 0.4671,
593
+ "step": 840
594
+ },
595
+ {
596
+ "epoch": 0.39,
597
+ "learning_rate": 8.194662480376766e-06,
598
+ "loss": 0.4481,
599
+ "step": 850
600
+ },
601
+ {
602
+ "epoch": 0.4,
603
+ "learning_rate": 8.171114599686029e-06,
604
+ "loss": 0.4481,
605
+ "step": 860
606
+ },
607
+ {
608
+ "epoch": 0.4,
609
+ "learning_rate": 8.147566718995291e-06,
610
+ "loss": 0.4317,
611
+ "step": 870
612
+ },
613
+ {
614
+ "epoch": 0.41,
615
+ "learning_rate": 8.124018838304554e-06,
616
+ "loss": 0.4517,
617
+ "step": 880
618
+ },
619
+ {
620
+ "epoch": 0.41,
621
+ "learning_rate": 8.100470957613814e-06,
622
+ "loss": 0.4282,
623
+ "step": 890
624
+ },
625
+ {
626
+ "epoch": 0.42,
627
+ "learning_rate": 8.076923076923077e-06,
628
+ "loss": 0.4543,
629
+ "step": 900
630
+ },
631
+ {
632
+ "epoch": 0.42,
633
+ "learning_rate": 8.05337519623234e-06,
634
+ "loss": 0.4303,
635
+ "step": 910
636
+ },
637
+ {
638
+ "epoch": 0.43,
639
+ "learning_rate": 8.029827315541602e-06,
640
+ "loss": 0.4423,
641
+ "step": 920
642
+ },
643
+ {
644
+ "epoch": 0.43,
645
+ "learning_rate": 8.006279434850863e-06,
646
+ "loss": 0.4624,
647
+ "step": 930
648
+ },
649
+ {
650
+ "epoch": 0.43,
651
+ "learning_rate": 7.982731554160125e-06,
652
+ "loss": 0.4318,
653
+ "step": 940
654
+ },
655
+ {
656
+ "epoch": 0.44,
657
+ "learning_rate": 7.959183673469388e-06,
658
+ "loss": 0.4374,
659
+ "step": 950
660
+ },
661
+ {
662
+ "epoch": 0.44,
663
+ "learning_rate": 7.93563579277865e-06,
664
+ "loss": 0.4271,
665
+ "step": 960
666
+ },
667
+ {
668
+ "epoch": 0.45,
669
+ "learning_rate": 7.912087912087913e-06,
670
+ "loss": 0.4535,
671
+ "step": 970
672
+ },
673
+ {
674
+ "epoch": 0.45,
675
+ "learning_rate": 7.888540031397175e-06,
676
+ "loss": 0.4393,
677
+ "step": 980
678
+ },
679
+ {
680
+ "epoch": 0.46,
681
+ "learning_rate": 7.864992150706436e-06,
682
+ "loss": 0.4513,
683
+ "step": 990
684
+ },
685
+ {
686
+ "epoch": 0.46,
687
+ "learning_rate": 7.841444270015699e-06,
688
+ "loss": 0.4259,
689
+ "step": 1000
690
+ },
691
+ {
692
+ "epoch": 0.46,
693
+ "eval_webgpt_accuracy": 0.5720122574055159,
694
+ "eval_webgpt_loss": 0.6880369782447815,
695
+ "eval_webgpt_runtime": 153.0376,
696
+ "eval_webgpt_samples_per_second": 25.588,
697
+ "eval_webgpt_steps_per_second": 2.561,
698
+ "step": 1000
699
+ },
700
+ {
701
+ "epoch": 0.46,
702
+ "eval_hfsummary_accuracy": 0.6844905238339933,
703
+ "eval_hfsummary_loss": 0.6170333027839661,
704
+ "eval_hfsummary_runtime": 2474.9308,
705
+ "eval_hfsummary_samples_per_second": 13.367,
706
+ "eval_hfsummary_steps_per_second": 1.337,
707
+ "step": 1000
708
+ },
709
+ {
710
+ "epoch": 0.46,
711
+ "eval_gptsynthetic_accuracy": 0.9978883861236802,
712
+ "eval_gptsynthetic_loss": 0.014577150344848633,
713
+ "eval_gptsynthetic_runtime": 116.8198,
714
+ "eval_gptsynthetic_samples_per_second": 28.377,
715
+ "eval_gptsynthetic_steps_per_second": 2.842,
716
+ "step": 1000
717
+ },
718
+ {
719
+ "epoch": 0.47,
720
+ "learning_rate": 7.817896389324961e-06,
721
+ "loss": 0.449,
722
+ "step": 1010
723
+ },
724
+ {
725
+ "epoch": 0.47,
726
+ "learning_rate": 7.794348508634224e-06,
727
+ "loss": 0.4274,
728
+ "step": 1020
729
+ },
730
+ {
731
+ "epoch": 0.48,
732
+ "learning_rate": 7.770800627943485e-06,
733
+ "loss": 0.4014,
734
+ "step": 1030
735
+ },
736
+ {
737
+ "epoch": 0.48,
738
+ "learning_rate": 7.747252747252747e-06,
739
+ "loss": 0.4356,
740
+ "step": 1040
741
+ },
742
+ {
743
+ "epoch": 0.49,
744
+ "learning_rate": 7.72370486656201e-06,
745
+ "loss": 0.4378,
746
+ "step": 1050
747
+ },
748
+ {
749
+ "epoch": 0.49,
750
+ "learning_rate": 7.700156985871272e-06,
751
+ "loss": 0.4298,
752
+ "step": 1060
753
+ },
754
+ {
755
+ "epoch": 0.49,
756
+ "learning_rate": 7.676609105180535e-06,
757
+ "loss": 0.4514,
758
+ "step": 1070
759
+ },
760
+ {
761
+ "epoch": 0.5,
762
+ "learning_rate": 7.653061224489796e-06,
763
+ "loss": 0.4526,
764
+ "step": 1080
765
+ },
766
+ {
767
+ "epoch": 0.5,
768
+ "learning_rate": 7.629513343799058e-06,
769
+ "loss": 0.4192,
770
+ "step": 1090
771
+ },
772
+ {
773
+ "epoch": 0.51,
774
+ "learning_rate": 7.60596546310832e-06,
775
+ "loss": 0.4433,
776
+ "step": 1100
777
+ },
778
+ {
779
+ "epoch": 0.51,
780
+ "learning_rate": 7.582417582417582e-06,
781
+ "loss": 0.4199,
782
+ "step": 1110
783
+ },
784
+ {
785
+ "epoch": 0.52,
786
+ "learning_rate": 7.558869701726845e-06,
787
+ "loss": 0.4419,
788
+ "step": 1120
789
+ },
790
+ {
791
+ "epoch": 0.52,
792
+ "learning_rate": 7.535321821036107e-06,
793
+ "loss": 0.4163,
794
+ "step": 1130
795
+ },
796
+ {
797
+ "epoch": 0.53,
798
+ "learning_rate": 7.511773940345369e-06,
799
+ "loss": 0.411,
800
+ "step": 1140
801
+ },
802
+ {
803
+ "epoch": 0.53,
804
+ "learning_rate": 7.4882260596546314e-06,
805
+ "loss": 0.4492,
806
+ "step": 1150
807
+ },
808
+ {
809
+ "epoch": 0.54,
810
+ "learning_rate": 7.464678178963893e-06,
811
+ "loss": 0.454,
812
+ "step": 1160
813
+ },
814
+ {
815
+ "epoch": 0.54,
816
+ "learning_rate": 7.441130298273156e-06,
817
+ "loss": 0.4311,
818
+ "step": 1170
819
+ },
820
+ {
821
+ "epoch": 0.55,
822
+ "learning_rate": 7.417582417582417e-06,
823
+ "loss": 0.4368,
824
+ "step": 1180
825
+ },
826
+ {
827
+ "epoch": 0.55,
828
+ "learning_rate": 7.39403453689168e-06,
829
+ "loss": 0.4264,
830
+ "step": 1190
831
+ },
832
+ {
833
+ "epoch": 0.56,
834
+ "learning_rate": 7.3704866562009415e-06,
835
+ "loss": 0.4163,
836
+ "step": 1200
837
+ },
838
+ {
839
+ "epoch": 0.56,
840
+ "learning_rate": 7.346938775510204e-06,
841
+ "loss": 0.4214,
842
+ "step": 1210
843
+ },
844
+ {
845
+ "epoch": 0.56,
846
+ "learning_rate": 7.3233908948194665e-06,
847
+ "loss": 0.4263,
848
+ "step": 1220
849
+ },
850
+ {
851
+ "epoch": 0.57,
852
+ "learning_rate": 7.299843014128729e-06,
853
+ "loss": 0.4233,
854
+ "step": 1230
855
+ },
856
+ {
857
+ "epoch": 0.57,
858
+ "learning_rate": 7.276295133437991e-06,
859
+ "loss": 0.4331,
860
+ "step": 1240
861
+ },
862
+ {
863
+ "epoch": 0.58,
864
+ "learning_rate": 7.252747252747253e-06,
865
+ "loss": 0.4241,
866
+ "step": 1250
867
+ },
868
+ {
869
+ "epoch": 0.58,
870
+ "eval_webgpt_accuracy": 0.5806945863125639,
871
+ "eval_webgpt_loss": 0.6766389608383179,
872
+ "eval_webgpt_runtime": 153.0189,
873
+ "eval_webgpt_samples_per_second": 25.592,
874
+ "eval_webgpt_steps_per_second": 2.562,
875
+ "step": 1250
876
+ },
877
+ {
878
+ "epoch": 0.58,
879
+ "eval_hfsummary_accuracy": 0.6927122691412508,
880
+ "eval_hfsummary_loss": 0.6117472648620605,
881
+ "eval_hfsummary_runtime": 2473.4013,
882
+ "eval_hfsummary_samples_per_second": 13.376,
883
+ "eval_hfsummary_steps_per_second": 1.338,
884
+ "step": 1250
885
+ },
886
+ {
887
+ "epoch": 0.58,
888
+ "eval_gptsynthetic_accuracy": 0.9987933634992459,
889
+ "eval_gptsynthetic_loss": 0.007306403946131468,
890
+ "eval_gptsynthetic_runtime": 116.8034,
891
+ "eval_gptsynthetic_samples_per_second": 28.381,
892
+ "eval_gptsynthetic_steps_per_second": 2.842,
893
+ "step": 1250
894
+ },
895
+ {
896
+ "epoch": 0.58,
897
+ "learning_rate": 7.229199372056515e-06,
898
+ "loss": 0.3976,
899
+ "step": 1260
900
+ },
901
+ {
902
+ "epoch": 0.59,
903
+ "learning_rate": 7.205651491365777e-06,
904
+ "loss": 0.4221,
905
+ "step": 1270
906
+ },
907
+ {
908
+ "epoch": 0.59,
909
+ "learning_rate": 7.182103610675039e-06,
910
+ "loss": 0.4225,
911
+ "step": 1280
912
+ },
913
+ {
914
+ "epoch": 0.6,
915
+ "learning_rate": 7.158555729984302e-06,
916
+ "loss": 0.3961,
917
+ "step": 1290
918
+ },
919
+ {
920
+ "epoch": 0.6,
921
+ "learning_rate": 7.135007849293563e-06,
922
+ "loss": 0.4348,
923
+ "step": 1300
924
+ },
925
+ {
926
+ "epoch": 0.61,
927
+ "learning_rate": 7.111459968602826e-06,
928
+ "loss": 0.4146,
929
+ "step": 1310
930
+ },
931
+ {
932
+ "epoch": 0.61,
933
+ "learning_rate": 7.087912087912088e-06,
934
+ "loss": 0.4122,
935
+ "step": 1320
936
+ },
937
+ {
938
+ "epoch": 0.62,
939
+ "learning_rate": 7.064364207221351e-06,
940
+ "loss": 0.4126,
941
+ "step": 1330
942
+ },
943
+ {
944
+ "epoch": 0.62,
945
+ "learning_rate": 7.0408163265306125e-06,
946
+ "loss": 0.4,
947
+ "step": 1340
948
+ },
949
+ {
950
+ "epoch": 0.62,
951
+ "learning_rate": 7.017268445839875e-06,
952
+ "loss": 0.3939,
953
+ "step": 1350
954
+ },
955
+ {
956
+ "epoch": 0.63,
957
+ "learning_rate": 6.993720565149137e-06,
958
+ "loss": 0.4144,
959
+ "step": 1360
960
+ },
961
+ {
962
+ "epoch": 0.63,
963
+ "learning_rate": 6.970172684458399e-06,
964
+ "loss": 0.3895,
965
+ "step": 1370
966
+ },
967
+ {
968
+ "epoch": 0.64,
969
+ "learning_rate": 6.946624803767661e-06,
970
+ "loss": 0.431,
971
+ "step": 1380
972
+ },
973
+ {
974
+ "epoch": 0.64,
975
+ "learning_rate": 6.923076923076923e-06,
976
+ "loss": 0.4156,
977
+ "step": 1390
978
+ },
979
+ {
980
+ "epoch": 0.65,
981
+ "learning_rate": 6.899529042386185e-06,
982
+ "loss": 0.4317,
983
+ "step": 1400
984
+ },
985
+ {
986
+ "epoch": 0.65,
987
+ "learning_rate": 6.8759811616954476e-06,
988
+ "loss": 0.4142,
989
+ "step": 1410
990
+ },
991
+ {
992
+ "epoch": 0.66,
993
+ "learning_rate": 6.85243328100471e-06,
994
+ "loss": 0.393,
995
+ "step": 1420
996
+ },
997
+ {
998
+ "epoch": 0.66,
999
+ "learning_rate": 6.828885400313973e-06,
1000
+ "loss": 0.4156,
1001
+ "step": 1430
1002
+ },
1003
+ {
1004
+ "epoch": 0.67,
1005
+ "learning_rate": 6.805337519623234e-06,
1006
+ "loss": 0.3995,
1007
+ "step": 1440
1008
+ },
1009
+ {
1010
+ "epoch": 0.67,
1011
+ "learning_rate": 6.781789638932497e-06,
1012
+ "loss": 0.4084,
1013
+ "step": 1450
1014
+ },
1015
+ {
1016
+ "epoch": 0.68,
1017
+ "learning_rate": 6.7582417582417585e-06,
1018
+ "loss": 0.4162,
1019
+ "step": 1460
1020
+ },
1021
+ {
1022
+ "epoch": 0.68,
1023
+ "learning_rate": 6.734693877551021e-06,
1024
+ "loss": 0.4167,
1025
+ "step": 1470
1026
+ },
1027
+ {
1028
+ "epoch": 0.68,
1029
+ "learning_rate": 6.711145996860283e-06,
1030
+ "loss": 0.3878,
1031
+ "step": 1480
1032
+ },
1033
+ {
1034
+ "epoch": 0.69,
1035
+ "learning_rate": 6.687598116169544e-06,
1036
+ "loss": 0.403,
1037
+ "step": 1490
1038
+ },
1039
+ {
1040
+ "epoch": 0.69,
1041
+ "learning_rate": 6.664050235478807e-06,
1042
+ "loss": 0.3988,
1043
+ "step": 1500
1044
+ },
1045
+ {
1046
+ "epoch": 0.69,
1047
+ "eval_webgpt_accuracy": 0.5840143003064351,
1048
+ "eval_webgpt_loss": 0.6755279302597046,
1049
+ "eval_webgpt_runtime": 152.9789,
1050
+ "eval_webgpt_samples_per_second": 25.598,
1051
+ "eval_webgpt_steps_per_second": 2.562,
1052
+ "step": 1500
1053
+ },
1054
+ {
1055
+ "epoch": 0.69,
1056
+ "eval_hfsummary_accuracy": 0.6905661518000181,
1057
+ "eval_hfsummary_loss": 0.6140123605728149,
1058
+ "eval_hfsummary_runtime": 2473.83,
1059
+ "eval_hfsummary_samples_per_second": 13.373,
1060
+ "eval_hfsummary_steps_per_second": 1.338,
1061
+ "step": 1500
1062
+ },
1063
+ {
1064
+ "epoch": 0.69,
1065
+ "eval_gptsynthetic_accuracy": 0.9978883861236802,
1066
+ "eval_gptsynthetic_loss": 0.008705263957381248,
1067
+ "eval_gptsynthetic_runtime": 116.82,
1068
+ "eval_gptsynthetic_samples_per_second": 28.377,
1069
+ "eval_gptsynthetic_steps_per_second": 2.842,
1070
+ "step": 1500
1071
+ },
1072
+ {
1073
+ "epoch": 0.7,
1074
+ "learning_rate": 6.6405023547880685e-06,
1075
+ "loss": 0.4004,
1076
+ "step": 1510
1077
+ },
1078
+ {
1079
+ "epoch": 0.7,
1080
+ "learning_rate": 6.616954474097332e-06,
1081
+ "loss": 0.3856,
1082
+ "step": 1520
1083
+ },
1084
+ {
1085
+ "epoch": 0.71,
1086
+ "learning_rate": 6.5934065934065935e-06,
1087
+ "loss": 0.4168,
1088
+ "step": 1530
1089
+ },
1090
+ {
1091
+ "epoch": 0.71,
1092
+ "learning_rate": 6.569858712715856e-06,
1093
+ "loss": 0.401,
1094
+ "step": 1540
1095
+ },
1096
+ {
1097
+ "epoch": 0.72,
1098
+ "learning_rate": 6.546310832025118e-06,
1099
+ "loss": 0.4038,
1100
+ "step": 1550
1101
+ },
1102
+ {
1103
+ "epoch": 0.72,
1104
+ "learning_rate": 6.52276295133438e-06,
1105
+ "loss": 0.3804,
1106
+ "step": 1560
1107
+ },
1108
+ {
1109
+ "epoch": 0.73,
1110
+ "learning_rate": 6.499215070643642e-06,
1111
+ "loss": 0.3787,
1112
+ "step": 1570
1113
+ },
1114
+ {
1115
+ "epoch": 0.73,
1116
+ "learning_rate": 6.4756671899529044e-06,
1117
+ "loss": 0.4003,
1118
+ "step": 1580
1119
+ },
1120
+ {
1121
+ "epoch": 0.74,
1122
+ "learning_rate": 6.452119309262166e-06,
1123
+ "loss": 0.3954,
1124
+ "step": 1590
1125
+ },
1126
+ {
1127
+ "epoch": 0.74,
1128
+ "learning_rate": 6.428571428571429e-06,
1129
+ "loss": 0.4011,
1130
+ "step": 1600
1131
+ },
1132
+ {
1133
+ "epoch": 0.74,
1134
+ "learning_rate": 6.40502354788069e-06,
1135
+ "loss": 0.3882,
1136
+ "step": 1610
1137
+ },
1138
+ {
1139
+ "epoch": 0.75,
1140
+ "learning_rate": 6.381475667189954e-06,
1141
+ "loss": 0.3954,
1142
+ "step": 1620
1143
+ },
1144
+ {
1145
+ "epoch": 0.75,
1146
+ "learning_rate": 6.357927786499215e-06,
1147
+ "loss": 0.3788,
1148
+ "step": 1630
1149
+ },
1150
+ {
1151
+ "epoch": 0.76,
1152
+ "learning_rate": 6.334379905808478e-06,
1153
+ "loss": 0.3921,
1154
+ "step": 1640
1155
+ },
1156
+ {
1157
+ "epoch": 0.76,
1158
+ "learning_rate": 6.3108320251177395e-06,
1159
+ "loss": 0.39,
1160
+ "step": 1650
1161
+ },
1162
+ {
1163
+ "epoch": 0.77,
1164
+ "learning_rate": 6.287284144427002e-06,
1165
+ "loss": 0.4137,
1166
+ "step": 1660
1167
+ },
1168
+ {
1169
+ "epoch": 0.77,
1170
+ "learning_rate": 6.263736263736264e-06,
1171
+ "loss": 0.4032,
1172
+ "step": 1670
1173
+ },
1174
+ {
1175
+ "epoch": 0.78,
1176
+ "learning_rate": 6.240188383045526e-06,
1177
+ "loss": 0.3865,
1178
+ "step": 1680
1179
+ },
1180
+ {
1181
+ "epoch": 0.78,
1182
+ "learning_rate": 6.216640502354788e-06,
1183
+ "loss": 0.3971,
1184
+ "step": 1690
1185
+ },
1186
+ {
1187
+ "epoch": 0.79,
1188
+ "learning_rate": 6.19309262166405e-06,
1189
+ "loss": 0.3585,
1190
+ "step": 1700
1191
+ },
1192
+ {
1193
+ "epoch": 0.79,
1194
+ "learning_rate": 6.169544740973312e-06,
1195
+ "loss": 0.3933,
1196
+ "step": 1710
1197
+ },
1198
+ {
1199
+ "epoch": 0.8,
1200
+ "learning_rate": 6.1459968602825754e-06,
1201
+ "loss": 0.4003,
1202
+ "step": 1720
1203
+ },
1204
+ {
1205
+ "epoch": 0.8,
1206
+ "learning_rate": 6.122448979591837e-06,
1207
+ "loss": 0.3931,
1208
+ "step": 1730
1209
+ },
1210
+ {
1211
+ "epoch": 0.8,
1212
+ "learning_rate": 6.0989010989011e-06,
1213
+ "loss": 0.3751,
1214
+ "step": 1740
1215
+ },
1216
+ {
1217
+ "epoch": 0.81,
1218
+ "learning_rate": 6.075353218210361e-06,
1219
+ "loss": 0.3793,
1220
+ "step": 1750
1221
+ },
1222
+ {
1223
+ "epoch": 0.81,
1224
+ "eval_webgpt_accuracy": 0.593973442288049,
1225
+ "eval_webgpt_loss": 0.6751279830932617,
1226
+ "eval_webgpt_runtime": 152.9792,
1227
+ "eval_webgpt_samples_per_second": 25.598,
1228
+ "eval_webgpt_steps_per_second": 2.562,
1229
+ "step": 1750
1230
+ },
1231
+ {
1232
+ "epoch": 0.81,
1233
+ "eval_hfsummary_accuracy": 0.6901127467279268,
1234
+ "eval_hfsummary_loss": 0.655200183391571,
1235
+ "eval_hfsummary_runtime": 2476.3756,
1236
+ "eval_hfsummary_samples_per_second": 13.359,
1237
+ "eval_hfsummary_steps_per_second": 1.336,
1238
+ "step": 1750
1239
+ },
1240
+ {
1241
+ "epoch": 0.81,
1242
+ "eval_gptsynthetic_accuracy": 0.9978883861236802,
1243
+ "eval_gptsynthetic_loss": 0.007934121415019035,
1244
+ "eval_gptsynthetic_runtime": 116.9338,
1245
+ "eval_gptsynthetic_samples_per_second": 28.349,
1246
+ "eval_gptsynthetic_steps_per_second": 2.839,
1247
+ "step": 1750
1248
+ },
1249
+ {
1250
+ "epoch": 0.81,
1251
+ "learning_rate": 6.051805337519624e-06,
1252
+ "loss": 0.3883,
1253
+ "step": 1760
1254
+ },
1255
+ {
1256
+ "epoch": 0.82,
1257
+ "learning_rate": 6.0282574568288855e-06,
1258
+ "loss": 0.3918,
1259
+ "step": 1770
1260
+ },
1261
+ {
1262
+ "epoch": 0.82,
1263
+ "learning_rate": 6.004709576138148e-06,
1264
+ "loss": 0.4086,
1265
+ "step": 1780
1266
+ },
1267
+ {
1268
+ "epoch": 0.83,
1269
+ "learning_rate": 5.98116169544741e-06,
1270
+ "loss": 0.3876,
1271
+ "step": 1790
1272
+ },
1273
+ {
1274
+ "epoch": 0.83,
1275
+ "learning_rate": 5.957613814756672e-06,
1276
+ "loss": 0.3613,
1277
+ "step": 1800
1278
+ },
1279
+ {
1280
+ "epoch": 0.84,
1281
+ "learning_rate": 5.934065934065934e-06,
1282
+ "loss": 0.3715,
1283
+ "step": 1810
1284
+ },
1285
+ {
1286
+ "epoch": 0.84,
1287
+ "learning_rate": 5.910518053375197e-06,
1288
+ "loss": 0.3879,
1289
+ "step": 1820
1290
+ },
1291
+ {
1292
+ "epoch": 0.85,
1293
+ "learning_rate": 5.886970172684459e-06,
1294
+ "loss": 0.3893,
1295
+ "step": 1830
1296
+ },
1297
+ {
1298
+ "epoch": 0.85,
1299
+ "learning_rate": 5.863422291993721e-06,
1300
+ "loss": 0.34,
1301
+ "step": 1840
1302
+ },
1303
+ {
1304
+ "epoch": 0.86,
1305
+ "learning_rate": 5.839874411302983e-06,
1306
+ "loss": 0.3938,
1307
+ "step": 1850
1308
+ },
1309
+ {
1310
+ "epoch": 0.86,
1311
+ "learning_rate": 5.816326530612246e-06,
1312
+ "loss": 0.3802,
1313
+ "step": 1860
1314
+ },
1315
+ {
1316
+ "epoch": 0.87,
1317
+ "learning_rate": 5.792778649921507e-06,
1318
+ "loss": 0.3869,
1319
+ "step": 1870
1320
+ },
1321
+ {
1322
+ "epoch": 0.87,
1323
+ "learning_rate": 5.76923076923077e-06,
1324
+ "loss": 0.4023,
1325
+ "step": 1880
1326
+ },
1327
+ {
1328
+ "epoch": 0.87,
1329
+ "learning_rate": 5.7456828885400314e-06,
1330
+ "loss": 0.3694,
1331
+ "step": 1890
1332
+ },
1333
+ {
1334
+ "epoch": 0.88,
1335
+ "learning_rate": 5.722135007849293e-06,
1336
+ "loss": 0.358,
1337
+ "step": 1900
1338
+ },
1339
+ {
1340
+ "epoch": 0.88,
1341
+ "learning_rate": 5.698587127158556e-06,
1342
+ "loss": 0.3939,
1343
+ "step": 1910
1344
+ },
1345
+ {
1346
+ "epoch": 0.89,
1347
+ "learning_rate": 5.675039246467817e-06,
1348
+ "loss": 0.3805,
1349
+ "step": 1920
1350
+ },
1351
+ {
1352
+ "epoch": 0.89,
1353
+ "learning_rate": 5.651491365777081e-06,
1354
+ "loss": 0.3754,
1355
+ "step": 1930
1356
+ },
1357
+ {
1358
+ "epoch": 0.9,
1359
+ "learning_rate": 5.627943485086342e-06,
1360
+ "loss": 0.3454,
1361
+ "step": 1940
1362
+ },
1363
+ {
1364
+ "epoch": 0.9,
1365
+ "learning_rate": 5.604395604395605e-06,
1366
+ "loss": 0.4109,
1367
+ "step": 1950
1368
+ },
1369
+ {
1370
+ "epoch": 0.91,
1371
+ "learning_rate": 5.5808477237048665e-06,
1372
+ "loss": 0.364,
1373
+ "step": 1960
1374
+ },
1375
+ {
1376
+ "epoch": 0.91,
1377
+ "learning_rate": 5.557299843014129e-06,
1378
+ "loss": 0.3869,
1379
+ "step": 1970
1380
+ },
1381
+ {
1382
+ "epoch": 0.92,
1383
+ "learning_rate": 5.533751962323391e-06,
1384
+ "loss": 0.3858,
1385
+ "step": 1980
1386
+ },
1387
+ {
1388
+ "epoch": 0.92,
1389
+ "learning_rate": 5.510204081632653e-06,
1390
+ "loss": 0.3634,
1391
+ "step": 1990
1392
+ },
1393
+ {
1394
+ "epoch": 0.93,
1395
+ "learning_rate": 5.486656200941915e-06,
1396
+ "loss": 0.3444,
1397
+ "step": 2000
1398
+ },
1399
+ {
1400
+ "epoch": 0.93,
1401
+ "eval_webgpt_accuracy": 0.5916751787538305,
1402
+ "eval_webgpt_loss": 0.6755998134613037,
1403
+ "eval_webgpt_runtime": 153.0377,
1404
+ "eval_webgpt_samples_per_second": 25.588,
1405
+ "eval_webgpt_steps_per_second": 2.561,
1406
+ "step": 2000
1407
+ },
1408
+ {
1409
+ "epoch": 0.93,
1410
+ "eval_hfsummary_accuracy": 0.6894477526221927,
1411
+ "eval_hfsummary_loss": 0.6544569730758667,
1412
+ "eval_hfsummary_runtime": 2482.2617,
1413
+ "eval_hfsummary_samples_per_second": 13.328,
1414
+ "eval_hfsummary_steps_per_second": 1.333,
1415
+ "step": 2000
1416
+ },
1417
+ {
1418
+ "epoch": 0.93,
1419
+ "eval_gptsynthetic_accuracy": 0.9975867269984917,
1420
+ "eval_gptsynthetic_loss": 0.00854982528835535,
1421
+ "eval_gptsynthetic_runtime": 117.0015,
1422
+ "eval_gptsynthetic_samples_per_second": 28.333,
1423
+ "eval_gptsynthetic_steps_per_second": 2.838,
1424
+ "step": 2000
1425
+ },
1426
+ {
1427
+ "epoch": 0.93,
1428
+ "learning_rate": 5.463108320251177e-06,
1429
+ "loss": 0.3548,
1430
+ "step": 2010
1431
+ },
1432
+ {
1433
+ "epoch": 0.93,
1434
+ "learning_rate": 5.439560439560439e-06,
1435
+ "loss": 0.3849,
1436
+ "step": 2020
1437
+ },
1438
+ {
1439
+ "epoch": 0.94,
1440
+ "learning_rate": 5.4160125588697024e-06,
1441
+ "loss": 0.3732,
1442
+ "step": 2030
1443
+ },
1444
+ {
1445
+ "epoch": 0.94,
1446
+ "learning_rate": 5.392464678178964e-06,
1447
+ "loss": 0.3589,
1448
+ "step": 2040
1449
+ },
1450
+ {
1451
+ "epoch": 0.95,
1452
+ "learning_rate": 5.368916797488227e-06,
1453
+ "loss": 0.3595,
1454
+ "step": 2050
1455
+ },
1456
+ {
1457
+ "epoch": 0.95,
1458
+ "learning_rate": 5.345368916797488e-06,
1459
+ "loss": 0.3704,
1460
+ "step": 2060
1461
+ },
1462
+ {
1463
+ "epoch": 0.96,
1464
+ "learning_rate": 5.324175824175825e-06,
1465
+ "loss": 0.3846,
1466
+ "step": 2070
1467
+ },
1468
+ {
1469
+ "epoch": 0.96,
1470
+ "learning_rate": 5.300627943485086e-06,
1471
+ "loss": 0.3862,
1472
+ "step": 2080
1473
+ },
1474
+ {
1475
+ "epoch": 0.97,
1476
+ "learning_rate": 5.277080062794349e-06,
1477
+ "loss": 0.3956,
1478
+ "step": 2090
1479
+ },
1480
+ {
1481
+ "epoch": 0.97,
1482
+ "learning_rate": 5.253532182103611e-06,
1483
+ "loss": 0.3629,
1484
+ "step": 2100
1485
+ },
1486
+ {
1487
+ "epoch": 0.98,
1488
+ "learning_rate": 5.229984301412873e-06,
1489
+ "loss": 0.3674,
1490
+ "step": 2110
1491
+ },
1492
+ {
1493
+ "epoch": 0.98,
1494
+ "learning_rate": 5.206436420722135e-06,
1495
+ "loss": 0.3827,
1496
+ "step": 2120
1497
+ },
1498
+ {
1499
+ "epoch": 0.99,
1500
+ "learning_rate": 5.182888540031397e-06,
1501
+ "loss": 0.3751,
1502
+ "step": 2130
1503
+ },
1504
+ {
1505
+ "epoch": 0.99,
1506
+ "learning_rate": 5.159340659340659e-06,
1507
+ "loss": 0.3834,
1508
+ "step": 2140
1509
+ },
1510
+ {
1511
+ "epoch": 0.99,
1512
+ "learning_rate": 5.1357927786499215e-06,
1513
+ "loss": 0.3425,
1514
+ "step": 2150
1515
+ },
1516
+ {
1517
+ "epoch": 1.0,
1518
+ "learning_rate": 5.112244897959183e-06,
1519
+ "loss": 0.3731,
1520
+ "step": 2160
1521
+ },
1522
+ {
1523
+ "epoch": 1.0,
1524
+ "learning_rate": 5.0886970172684465e-06,
1525
+ "loss": 0.3724,
1526
+ "step": 2170
1527
+ },
1528
+ {
1529
+ "epoch": 1.01,
1530
+ "learning_rate": 5.065149136577708e-06,
1531
+ "loss": 0.3842,
1532
+ "step": 2180
1533
+ },
1534
+ {
1535
+ "epoch": 1.01,
1536
+ "learning_rate": 5.041601255886971e-06,
1537
+ "loss": 0.3661,
1538
+ "step": 2190
1539
+ },
1540
+ {
1541
+ "epoch": 1.02,
1542
+ "learning_rate": 5.018053375196232e-06,
1543
+ "loss": 0.3795,
1544
+ "step": 2200
1545
+ },
1546
+ {
1547
+ "epoch": 1.02,
1548
+ "learning_rate": 4.994505494505495e-06,
1549
+ "loss": 0.3766,
1550
+ "step": 2210
1551
+ },
1552
+ {
1553
+ "epoch": 1.03,
1554
+ "learning_rate": 4.9709576138147566e-06,
1555
+ "loss": 0.3641,
1556
+ "step": 2220
1557
+ },
1558
+ {
1559
+ "epoch": 1.03,
1560
+ "learning_rate": 4.947409733124019e-06,
1561
+ "loss": 0.355,
1562
+ "step": 2230
1563
+ },
1564
+ {
1565
+ "epoch": 1.04,
1566
+ "learning_rate": 4.923861852433281e-06,
1567
+ "loss": 0.3481,
1568
+ "step": 2240
1569
+ },
1570
+ {
1571
+ "epoch": 1.04,
1572
+ "learning_rate": 4.900313971742543e-06,
1573
+ "loss": 0.312,
1574
+ "step": 2250
1575
+ },
1576
+ {
1577
+ "epoch": 1.04,
1578
+ "eval_webgpt_accuracy": 0.5970377936670072,
1579
+ "eval_webgpt_loss": 0.6743035912513733,
1580
+ "eval_webgpt_runtime": 153.349,
1581
+ "eval_webgpt_samples_per_second": 25.537,
1582
+ "eval_webgpt_steps_per_second": 2.556,
1583
+ "step": 2250
1584
+ },
1585
+ {
1586
+ "epoch": 1.04,
1587
+ "eval_hfsummary_accuracy": 0.6958558776410846,
1588
+ "eval_hfsummary_loss": 0.6904739141464233,
1589
+ "eval_hfsummary_runtime": 2483.5335,
1590
+ "eval_hfsummary_samples_per_second": 13.321,
1591
+ "eval_hfsummary_steps_per_second": 1.332,
1592
+ "step": 2250
1593
+ },
1594
+ {
1595
+ "epoch": 1.04,
1596
+ "eval_gptsynthetic_accuracy": 0.9972850678733032,
1597
+ "eval_gptsynthetic_loss": 0.011290821246802807,
1598
+ "eval_gptsynthetic_runtime": 117.0341,
1599
+ "eval_gptsynthetic_samples_per_second": 28.325,
1600
+ "eval_gptsynthetic_steps_per_second": 2.837,
1601
+ "step": 2250
1602
+ },
1603
+ {
1604
+ "epoch": 1.05,
1605
+ "learning_rate": 4.876766091051805e-06,
1606
+ "loss": 0.3593,
1607
+ "step": 2260
1608
+ },
1609
+ {
1610
+ "epoch": 1.05,
1611
+ "learning_rate": 4.853218210361068e-06,
1612
+ "loss": 0.3378,
1613
+ "step": 2270
1614
+ },
1615
+ {
1616
+ "epoch": 1.06,
1617
+ "learning_rate": 4.82967032967033e-06,
1618
+ "loss": 0.3522,
1619
+ "step": 2280
1620
+ },
1621
+ {
1622
+ "epoch": 1.06,
1623
+ "learning_rate": 4.8061224489795925e-06,
1624
+ "loss": 0.3732,
1625
+ "step": 2290
1626
+ },
1627
+ {
1628
+ "epoch": 1.06,
1629
+ "learning_rate": 4.782574568288854e-06,
1630
+ "loss": 0.3616,
1631
+ "step": 2300
1632
+ },
1633
+ {
1634
+ "epoch": 1.07,
1635
+ "learning_rate": 4.759026687598117e-06,
1636
+ "loss": 0.3478,
1637
+ "step": 2310
1638
+ },
1639
+ {
1640
+ "epoch": 1.07,
1641
+ "learning_rate": 4.735478806907378e-06,
1642
+ "loss": 0.3721,
1643
+ "step": 2320
1644
+ },
1645
+ {
1646
+ "epoch": 1.08,
1647
+ "learning_rate": 4.711930926216641e-06,
1648
+ "loss": 0.3314,
1649
+ "step": 2330
1650
+ },
1651
+ {
1652
+ "epoch": 1.08,
1653
+ "learning_rate": 4.6883830455259025e-06,
1654
+ "loss": 0.3413,
1655
+ "step": 2340
1656
+ },
1657
+ {
1658
+ "epoch": 1.09,
1659
+ "learning_rate": 4.664835164835165e-06,
1660
+ "loss": 0.3266,
1661
+ "step": 2350
1662
+ },
1663
+ {
1664
+ "epoch": 1.09,
1665
+ "learning_rate": 4.641287284144427e-06,
1666
+ "loss": 0.3579,
1667
+ "step": 2360
1668
+ },
1669
+ {
1670
+ "epoch": 1.1,
1671
+ "learning_rate": 4.61773940345369e-06,
1672
+ "loss": 0.3549,
1673
+ "step": 2370
1674
+ },
1675
+ {
1676
+ "epoch": 1.1,
1677
+ "learning_rate": 4.594191522762952e-06,
1678
+ "loss": 0.347,
1679
+ "step": 2380
1680
+ },
1681
+ {
1682
+ "epoch": 1.11,
1683
+ "learning_rate": 4.570643642072214e-06,
1684
+ "loss": 0.3453,
1685
+ "step": 2390
1686
+ },
1687
+ {
1688
+ "epoch": 1.11,
1689
+ "learning_rate": 4.547095761381476e-06,
1690
+ "loss": 0.3411,
1691
+ "step": 2400
1692
+ },
1693
+ {
1694
+ "epoch": 1.12,
1695
+ "learning_rate": 4.5235478806907384e-06,
1696
+ "loss": 0.3411,
1697
+ "step": 2410
1698
+ },
1699
+ {
1700
+ "epoch": 1.12,
1701
+ "learning_rate": 4.5e-06,
1702
+ "loss": 0.361,
1703
+ "step": 2420
1704
+ },
1705
+ {
1706
+ "epoch": 1.12,
1707
+ "learning_rate": 4.476452119309263e-06,
1708
+ "loss": 0.3576,
1709
+ "step": 2430
1710
+ },
1711
+ {
1712
+ "epoch": 1.13,
1713
+ "learning_rate": 4.452904238618524e-06,
1714
+ "loss": 0.3476,
1715
+ "step": 2440
1716
+ },
1717
+ {
1718
+ "epoch": 1.13,
1719
+ "learning_rate": 4.429356357927787e-06,
1720
+ "loss": 0.3382,
1721
+ "step": 2450
1722
+ },
1723
+ {
1724
+ "epoch": 1.14,
1725
+ "learning_rate": 4.4058084772370485e-06,
1726
+ "loss": 0.3485,
1727
+ "step": 2460
1728
+ },
1729
+ {
1730
+ "epoch": 1.14,
1731
+ "learning_rate": 4.382260596546311e-06,
1732
+ "loss": 0.3237,
1733
+ "step": 2470
1734
+ },
1735
+ {
1736
+ "epoch": 1.15,
1737
+ "learning_rate": 4.3587127158555735e-06,
1738
+ "loss": 0.3385,
1739
+ "step": 2480
1740
+ },
1741
+ {
1742
+ "epoch": 1.15,
1743
+ "learning_rate": 4.335164835164835e-06,
1744
+ "loss": 0.3124,
1745
+ "step": 2490
1746
+ },
1747
+ {
1748
+ "epoch": 1.16,
1749
+ "learning_rate": 4.311616954474098e-06,
1750
+ "loss": 0.3385,
1751
+ "step": 2500
1752
+ },
1753
+ {
1754
+ "epoch": 1.16,
1755
+ "eval_webgpt_accuracy": 0.589632277834525,
1756
+ "eval_webgpt_loss": 0.6730017066001892,
1757
+ "eval_webgpt_runtime": 153.2835,
1758
+ "eval_webgpt_samples_per_second": 25.547,
1759
+ "eval_webgpt_steps_per_second": 2.557,
1760
+ "step": 2500
1761
+ },
1762
+ {
1763
+ "epoch": 1.16,
1764
+ "eval_hfsummary_accuracy": 0.6939515763383006,
1765
+ "eval_hfsummary_loss": 0.6676614880561829,
1766
+ "eval_hfsummary_runtime": 2484.0871,
1767
+ "eval_hfsummary_samples_per_second": 13.318,
1768
+ "eval_hfsummary_steps_per_second": 1.332,
1769
+ "step": 2500
1770
+ },
1771
+ {
1772
+ "epoch": 1.16,
1773
+ "eval_gptsynthetic_accuracy": 0.9987933634992459,
1774
+ "eval_gptsynthetic_loss": 0.008312013931572437,
1775
+ "eval_gptsynthetic_runtime": 117.0521,
1776
+ "eval_gptsynthetic_samples_per_second": 28.321,
1777
+ "eval_gptsynthetic_steps_per_second": 2.836,
1778
+ "step": 2500
1779
+ },
1780
+ {
1781
+ "epoch": 1.16,
1782
+ "learning_rate": 4.288069073783359e-06,
1783
+ "loss": 0.3575,
1784
+ "step": 2510
1785
+ },
1786
+ {
1787
+ "epoch": 1.17,
1788
+ "learning_rate": 4.264521193092622e-06,
1789
+ "loss": 0.3562,
1790
+ "step": 2520
1791
+ },
1792
+ {
1793
+ "epoch": 1.17,
1794
+ "learning_rate": 4.240973312401884e-06,
1795
+ "loss": 0.3296,
1796
+ "step": 2530
1797
+ },
1798
+ {
1799
+ "epoch": 1.18,
1800
+ "learning_rate": 4.217425431711146e-06,
1801
+ "loss": 0.3447,
1802
+ "step": 2540
1803
+ },
1804
+ {
1805
+ "epoch": 1.18,
1806
+ "learning_rate": 4.193877551020409e-06,
1807
+ "loss": 0.3373,
1808
+ "step": 2550
1809
+ },
1810
+ {
1811
+ "epoch": 1.18,
1812
+ "learning_rate": 4.17032967032967e-06,
1813
+ "loss": 0.3394,
1814
+ "step": 2560
1815
+ },
1816
+ {
1817
+ "epoch": 1.19,
1818
+ "learning_rate": 4.146781789638933e-06,
1819
+ "loss": 0.3294,
1820
+ "step": 2570
1821
+ },
1822
+ {
1823
+ "epoch": 1.19,
1824
+ "learning_rate": 4.123233908948195e-06,
1825
+ "loss": 0.3715,
1826
+ "step": 2580
1827
+ },
1828
+ {
1829
+ "epoch": 1.2,
1830
+ "learning_rate": 4.099686028257457e-06,
1831
+ "loss": 0.3356,
1832
+ "step": 2590
1833
+ },
1834
+ {
1835
+ "epoch": 1.2,
1836
+ "learning_rate": 4.0761381475667195e-06,
1837
+ "loss": 0.3497,
1838
+ "step": 2600
1839
+ },
1840
+ {
1841
+ "epoch": 1.21,
1842
+ "learning_rate": 4.052590266875981e-06,
1843
+ "loss": 0.3189,
1844
+ "step": 2610
1845
+ },
1846
+ {
1847
+ "epoch": 1.21,
1848
+ "learning_rate": 4.029042386185244e-06,
1849
+ "loss": 0.3473,
1850
+ "step": 2620
1851
+ },
1852
+ {
1853
+ "epoch": 1.22,
1854
+ "learning_rate": 4.005494505494505e-06,
1855
+ "loss": 0.3259,
1856
+ "step": 2630
1857
+ },
1858
+ {
1859
+ "epoch": 1.22,
1860
+ "learning_rate": 3.981946624803768e-06,
1861
+ "loss": 0.3294,
1862
+ "step": 2640
1863
+ },
1864
+ {
1865
+ "epoch": 1.23,
1866
+ "learning_rate": 3.9583987441130295e-06,
1867
+ "loss": 0.3294,
1868
+ "step": 2650
1869
+ },
1870
+ {
1871
+ "epoch": 1.23,
1872
+ "learning_rate": 3.934850863422292e-06,
1873
+ "loss": 0.3221,
1874
+ "step": 2660
1875
+ },
1876
+ {
1877
+ "epoch": 1.24,
1878
+ "learning_rate": 3.911302982731554e-06,
1879
+ "loss": 0.3698,
1880
+ "step": 2670
1881
+ },
1882
+ {
1883
+ "epoch": 1.24,
1884
+ "learning_rate": 3.887755102040816e-06,
1885
+ "loss": 0.3514,
1886
+ "step": 2680
1887
+ },
1888
+ {
1889
+ "epoch": 1.24,
1890
+ "learning_rate": 3.864207221350079e-06,
1891
+ "loss": 0.3461,
1892
+ "step": 2690
1893
+ },
1894
+ {
1895
+ "epoch": 1.25,
1896
+ "learning_rate": 3.84065934065934e-06,
1897
+ "loss": 0.3321,
1898
+ "step": 2700
1899
+ },
1900
+ {
1901
+ "epoch": 1.25,
1902
+ "learning_rate": 3.817111459968603e-06,
1903
+ "loss": 0.3241,
1904
+ "step": 2710
1905
+ },
1906
+ {
1907
+ "epoch": 1.26,
1908
+ "learning_rate": 3.793563579277865e-06,
1909
+ "loss": 0.3466,
1910
+ "step": 2720
1911
+ },
1912
+ {
1913
+ "epoch": 1.26,
1914
+ "learning_rate": 3.770015698587127e-06,
1915
+ "loss": 0.3188,
1916
+ "step": 2730
1917
+ },
1918
+ {
1919
+ "epoch": 1.27,
1920
+ "learning_rate": 3.7464678178963892e-06,
1921
+ "loss": 0.32,
1922
+ "step": 2740
1923
+ },
1924
+ {
1925
+ "epoch": 1.27,
1926
+ "learning_rate": 3.7229199372056513e-06,
1927
+ "loss": 0.3383,
1928
+ "step": 2750
1929
+ },
1930
+ {
1931
+ "epoch": 1.27,
1932
+ "eval_webgpt_accuracy": 0.5776302349336058,
1933
+ "eval_webgpt_loss": 0.6770792603492737,
1934
+ "eval_webgpt_runtime": 153.2414,
1935
+ "eval_webgpt_samples_per_second": 25.554,
1936
+ "eval_webgpt_steps_per_second": 2.558,
1937
+ "step": 2750
1938
+ },
1939
+ {
1940
+ "epoch": 1.27,
1941
+ "eval_hfsummary_accuracy": 0.6819212284254753,
1942
+ "eval_hfsummary_loss": 0.7081930041313171,
1943
+ "eval_hfsummary_runtime": 2484.801,
1944
+ "eval_hfsummary_samples_per_second": 13.314,
1945
+ "eval_hfsummary_steps_per_second": 1.332,
1946
+ "step": 2750
1947
+ },
1948
+ {
1949
+ "epoch": 1.27,
1950
+ "eval_gptsynthetic_accuracy": 0.9984917043740573,
1951
+ "eval_gptsynthetic_loss": 0.006183410994708538,
1952
+ "eval_gptsynthetic_runtime": 117.1145,
1953
+ "eval_gptsynthetic_samples_per_second": 28.306,
1954
+ "eval_gptsynthetic_steps_per_second": 2.835,
1955
+ "step": 2750
1956
+ },
1957
+ {
1958
+ "epoch": 1.28,
1959
+ "learning_rate": 3.699372056514914e-06,
1960
+ "loss": 0.3136,
1961
+ "step": 2760
1962
+ },
1963
+ {
1964
+ "epoch": 1.28,
1965
+ "learning_rate": 3.675824175824176e-06,
1966
+ "loss": 0.3358,
1967
+ "step": 2770
1968
+ },
1969
+ {
1970
+ "epoch": 1.29,
1971
+ "learning_rate": 3.652276295133438e-06,
1972
+ "loss": 0.3612,
1973
+ "step": 2780
1974
+ },
1975
+ {
1976
+ "epoch": 1.29,
1977
+ "learning_rate": 3.6287284144427e-06,
1978
+ "loss": 0.319,
1979
+ "step": 2790
1980
+ },
1981
+ {
1982
+ "epoch": 1.3,
1983
+ "learning_rate": 3.605180533751962e-06,
1984
+ "loss": 0.3039,
1985
+ "step": 2800
1986
+ },
1987
+ {
1988
+ "epoch": 1.3,
1989
+ "learning_rate": 3.5816326530612247e-06,
1990
+ "loss": 0.3383,
1991
+ "step": 2810
1992
+ },
1993
+ {
1994
+ "epoch": 1.3,
1995
+ "learning_rate": 3.558084772370487e-06,
1996
+ "loss": 0.3096,
1997
+ "step": 2820
1998
+ },
1999
+ {
2000
+ "epoch": 1.31,
2001
+ "learning_rate": 3.534536891679749e-06,
2002
+ "loss": 0.3377,
2003
+ "step": 2830
2004
+ },
2005
+ {
2006
+ "epoch": 1.31,
2007
+ "learning_rate": 3.510989010989011e-06,
2008
+ "loss": 0.314,
2009
+ "step": 2840
2010
+ },
2011
+ {
2012
+ "epoch": 1.32,
2013
+ "learning_rate": 3.487441130298273e-06,
2014
+ "loss": 0.344,
2015
+ "step": 2850
2016
+ },
2017
+ {
2018
+ "epoch": 1.32,
2019
+ "learning_rate": 3.4638932496075356e-06,
2020
+ "loss": 0.3158,
2021
+ "step": 2860
2022
+ },
2023
+ {
2024
+ "epoch": 1.33,
2025
+ "learning_rate": 3.4403453689167977e-06,
2026
+ "loss": 0.3277,
2027
+ "step": 2870
2028
+ },
2029
+ {
2030
+ "epoch": 1.33,
2031
+ "learning_rate": 3.41679748822606e-06,
2032
+ "loss": 0.3292,
2033
+ "step": 2880
2034
+ },
2035
+ {
2036
+ "epoch": 1.34,
2037
+ "learning_rate": 3.393249607535322e-06,
2038
+ "loss": 0.3479,
2039
+ "step": 2890
2040
+ },
2041
+ {
2042
+ "epoch": 1.34,
2043
+ "learning_rate": 3.369701726844584e-06,
2044
+ "loss": 0.338,
2045
+ "step": 2900
2046
+ },
2047
+ {
2048
+ "epoch": 1.35,
2049
+ "learning_rate": 3.3461538461538465e-06,
2050
+ "loss": 0.3388,
2051
+ "step": 2910
2052
+ },
2053
+ {
2054
+ "epoch": 1.35,
2055
+ "learning_rate": 3.3226059654631086e-06,
2056
+ "loss": 0.2873,
2057
+ "step": 2920
2058
+ },
2059
+ {
2060
+ "epoch": 1.36,
2061
+ "learning_rate": 3.2990580847723707e-06,
2062
+ "loss": 0.3147,
2063
+ "step": 2930
2064
+ },
2065
+ {
2066
+ "epoch": 1.36,
2067
+ "learning_rate": 3.2755102040816328e-06,
2068
+ "loss": 0.3423,
2069
+ "step": 2940
2070
+ },
2071
+ {
2072
+ "epoch": 1.36,
2073
+ "learning_rate": 3.251962323390895e-06,
2074
+ "loss": 0.3052,
2075
+ "step": 2950
2076
+ },
2077
+ {
2078
+ "epoch": 1.37,
2079
+ "learning_rate": 3.2284144427001574e-06,
2080
+ "loss": 0.3308,
2081
+ "step": 2960
2082
+ },
2083
+ {
2084
+ "epoch": 1.37,
2085
+ "learning_rate": 3.2048665620094195e-06,
2086
+ "loss": 0.3484,
2087
+ "step": 2970
2088
+ },
2089
+ {
2090
+ "epoch": 1.38,
2091
+ "learning_rate": 3.1813186813186816e-06,
2092
+ "loss": 0.3287,
2093
+ "step": 2980
2094
+ },
2095
+ {
2096
+ "epoch": 1.38,
2097
+ "learning_rate": 3.1577708006279437e-06,
2098
+ "loss": 0.2954,
2099
+ "step": 2990
2100
+ },
2101
+ {
2102
+ "epoch": 1.39,
2103
+ "learning_rate": 3.1342229199372058e-06,
2104
+ "loss": 0.3284,
2105
+ "step": 3000
2106
+ },
2107
+ {
2108
+ "epoch": 1.39,
2109
+ "eval_webgpt_accuracy": 0.5832482124616956,
2110
+ "eval_webgpt_loss": 0.6810373067855835,
2111
+ "eval_webgpt_runtime": 153.3795,
2112
+ "eval_webgpt_samples_per_second": 25.531,
2113
+ "eval_webgpt_steps_per_second": 2.556,
2114
+ "step": 3000
2115
+ },
2116
+ {
2117
+ "epoch": 1.39,
2118
+ "eval_hfsummary_accuracy": 0.6889641205452952,
2119
+ "eval_hfsummary_loss": 0.7132372856140137,
2120
+ "eval_hfsummary_runtime": 2481.1392,
2121
+ "eval_hfsummary_samples_per_second": 13.334,
2122
+ "eval_hfsummary_steps_per_second": 1.334,
2123
+ "step": 3000
2124
+ },
2125
+ {
2126
+ "epoch": 1.39,
2127
+ "eval_gptsynthetic_accuracy": 0.9987933634992459,
2128
+ "eval_gptsynthetic_loss": 0.005898704752326012,
2129
+ "eval_gptsynthetic_runtime": 116.9777,
2130
+ "eval_gptsynthetic_samples_per_second": 28.339,
2131
+ "eval_gptsynthetic_steps_per_second": 2.838,
2132
+ "step": 3000
2133
+ },
2134
+ {
2135
+ "epoch": 1.39,
2136
+ "learning_rate": 3.1106750392464683e-06,
2137
+ "loss": 0.3138,
2138
+ "step": 3010
2139
+ },
2140
+ {
2141
+ "epoch": 1.4,
2142
+ "learning_rate": 3.0871271585557304e-06,
2143
+ "loss": 0.3364,
2144
+ "step": 3020
2145
+ },
2146
+ {
2147
+ "epoch": 1.4,
2148
+ "learning_rate": 3.063579277864992e-06,
2149
+ "loss": 0.3439,
2150
+ "step": 3030
2151
+ },
2152
+ {
2153
+ "epoch": 1.41,
2154
+ "learning_rate": 3.040031397174254e-06,
2155
+ "loss": 0.3489,
2156
+ "step": 3040
2157
+ },
2158
+ {
2159
+ "epoch": 1.41,
2160
+ "learning_rate": 3.0164835164835162e-06,
2161
+ "loss": 0.3419,
2162
+ "step": 3050
2163
+ },
2164
+ {
2165
+ "epoch": 1.42,
2166
+ "learning_rate": 2.9929356357927783e-06,
2167
+ "loss": 0.3052,
2168
+ "step": 3060
2169
+ },
2170
+ {
2171
+ "epoch": 1.42,
2172
+ "learning_rate": 2.969387755102041e-06,
2173
+ "loss": 0.3126,
2174
+ "step": 3070
2175
+ },
2176
+ {
2177
+ "epoch": 1.43,
2178
+ "learning_rate": 2.945839874411303e-06,
2179
+ "loss": 0.3475,
2180
+ "step": 3080
2181
+ },
2182
+ {
2183
+ "epoch": 1.43,
2184
+ "learning_rate": 2.922291993720565e-06,
2185
+ "loss": 0.3444,
2186
+ "step": 3090
2187
+ },
2188
+ {
2189
+ "epoch": 1.43,
2190
+ "learning_rate": 2.898744113029827e-06,
2191
+ "loss": 0.3279,
2192
+ "step": 3100
2193
+ },
2194
+ {
2195
+ "epoch": 1.44,
2196
+ "learning_rate": 2.8751962323390892e-06,
2197
+ "loss": 0.3315,
2198
+ "step": 3110
2199
+ },
2200
+ {
2201
+ "epoch": 1.44,
2202
+ "learning_rate": 2.8516483516483517e-06,
2203
+ "loss": 0.3238,
2204
+ "step": 3120
2205
+ },
2206
+ {
2207
+ "epoch": 1.45,
2208
+ "learning_rate": 2.828100470957614e-06,
2209
+ "loss": 0.3291,
2210
+ "step": 3130
2211
+ },
2212
+ {
2213
+ "epoch": 1.45,
2214
+ "learning_rate": 2.804552590266876e-06,
2215
+ "loss": 0.3014,
2216
+ "step": 3140
2217
+ },
2218
+ {
2219
+ "epoch": 1.46,
2220
+ "learning_rate": 2.781004709576138e-06,
2221
+ "loss": 0.3397,
2222
+ "step": 3150
2223
+ },
2224
+ {
2225
+ "epoch": 1.46,
2226
+ "learning_rate": 2.7574568288854e-06,
2227
+ "loss": 0.3143,
2228
+ "step": 3160
2229
+ },
2230
+ {
2231
+ "epoch": 1.47,
2232
+ "learning_rate": 2.7339089481946626e-06,
2233
+ "loss": 0.3317,
2234
+ "step": 3170
2235
+ },
2236
+ {
2237
+ "epoch": 1.47,
2238
+ "learning_rate": 2.7103610675039247e-06,
2239
+ "loss": 0.3522,
2240
+ "step": 3180
2241
+ },
2242
+ {
2243
+ "epoch": 1.48,
2244
+ "learning_rate": 2.686813186813187e-06,
2245
+ "loss": 0.3324,
2246
+ "step": 3190
2247
+ },
2248
+ {
2249
+ "epoch": 1.48,
2250
+ "learning_rate": 2.663265306122449e-06,
2251
+ "loss": 0.3043,
2252
+ "step": 3200
2253
+ },
2254
+ {
2255
+ "epoch": 1.49,
2256
+ "learning_rate": 2.639717425431711e-06,
2257
+ "loss": 0.3424,
2258
+ "step": 3210
2259
+ },
2260
+ {
2261
+ "epoch": 1.49,
2262
+ "learning_rate": 2.6161695447409735e-06,
2263
+ "loss": 0.3461,
2264
+ "step": 3220
2265
+ },
2266
+ {
2267
+ "epoch": 1.49,
2268
+ "learning_rate": 2.5926216640502356e-06,
2269
+ "loss": 0.3289,
2270
+ "step": 3230
2271
+ },
2272
+ {
2273
+ "epoch": 1.5,
2274
+ "learning_rate": 2.5690737833594977e-06,
2275
+ "loss": 0.3057,
2276
+ "step": 3240
2277
+ },
2278
+ {
2279
+ "epoch": 1.5,
2280
+ "learning_rate": 2.54552590266876e-06,
2281
+ "loss": 0.3118,
2282
+ "step": 3250
2283
+ },
2284
+ {
2285
+ "epoch": 1.5,
2286
+ "eval_webgpt_accuracy": 0.5847803881511746,
2287
+ "eval_webgpt_loss": 0.6773096919059753,
2288
+ "eval_webgpt_runtime": 153.2012,
2289
+ "eval_webgpt_samples_per_second": 25.561,
2290
+ "eval_webgpt_steps_per_second": 2.559,
2291
+ "step": 3250
2292
+ },
2293
+ {
2294
+ "epoch": 1.5,
2295
+ "eval_hfsummary_accuracy": 0.6834628056705861,
2296
+ "eval_hfsummary_loss": 0.7264418005943298,
2297
+ "eval_hfsummary_runtime": 2480.0981,
2298
+ "eval_hfsummary_samples_per_second": 13.339,
2299
+ "eval_hfsummary_steps_per_second": 1.334,
2300
+ "step": 3250
2301
+ },
2302
+ {
2303
+ "epoch": 1.5,
2304
+ "eval_gptsynthetic_accuracy": 0.9984917043740573,
2305
+ "eval_gptsynthetic_loss": 0.008459103293716908,
2306
+ "eval_gptsynthetic_runtime": 116.9721,
2307
+ "eval_gptsynthetic_samples_per_second": 28.34,
2308
+ "eval_gptsynthetic_steps_per_second": 2.838,
2309
+ "step": 3250
2310
+ },
2311
+ {
2312
+ "epoch": 1.51,
2313
+ "learning_rate": 2.521978021978022e-06,
2314
+ "loss": 0.2776,
2315
+ "step": 3260
2316
+ },
2317
+ {
2318
+ "epoch": 1.51,
2319
+ "learning_rate": 2.4984301412872844e-06,
2320
+ "loss": 0.3286,
2321
+ "step": 3270
2322
+ },
2323
+ {
2324
+ "epoch": 1.52,
2325
+ "learning_rate": 2.4748822605965465e-06,
2326
+ "loss": 0.3118,
2327
+ "step": 3280
2328
+ },
2329
+ {
2330
+ "epoch": 1.52,
2331
+ "learning_rate": 2.4513343799058086e-06,
2332
+ "loss": 0.3444,
2333
+ "step": 3290
2334
+ },
2335
+ {
2336
+ "epoch": 1.53,
2337
+ "learning_rate": 2.4277864992150707e-06,
2338
+ "loss": 0.3078,
2339
+ "step": 3300
2340
+ },
2341
+ {
2342
+ "epoch": 1.53,
2343
+ "learning_rate": 2.4042386185243328e-06,
2344
+ "loss": 0.3139,
2345
+ "step": 3310
2346
+ },
2347
+ {
2348
+ "epoch": 1.54,
2349
+ "learning_rate": 2.3806907378335953e-06,
2350
+ "loss": 0.32,
2351
+ "step": 3320
2352
+ },
2353
+ {
2354
+ "epoch": 1.54,
2355
+ "learning_rate": 2.3571428571428574e-06,
2356
+ "loss": 0.3211,
2357
+ "step": 3330
2358
+ },
2359
+ {
2360
+ "epoch": 1.55,
2361
+ "learning_rate": 2.3335949764521195e-06,
2362
+ "loss": 0.306,
2363
+ "step": 3340
2364
+ },
2365
+ {
2366
+ "epoch": 1.55,
2367
+ "learning_rate": 2.3100470957613816e-06,
2368
+ "loss": 0.3045,
2369
+ "step": 3350
2370
+ },
2371
+ {
2372
+ "epoch": 1.55,
2373
+ "learning_rate": 2.2864992150706437e-06,
2374
+ "loss": 0.3123,
2375
+ "step": 3360
2376
+ },
2377
+ {
2378
+ "epoch": 1.56,
2379
+ "learning_rate": 2.262951334379906e-06,
2380
+ "loss": 0.3073,
2381
+ "step": 3370
2382
+ },
2383
+ {
2384
+ "epoch": 1.56,
2385
+ "learning_rate": 2.239403453689168e-06,
2386
+ "loss": 0.3061,
2387
+ "step": 3380
2388
+ },
2389
+ {
2390
+ "epoch": 1.57,
2391
+ "learning_rate": 2.21585557299843e-06,
2392
+ "loss": 0.297,
2393
+ "step": 3390
2394
+ },
2395
+ {
2396
+ "epoch": 1.57,
2397
+ "learning_rate": 2.1923076923076925e-06,
2398
+ "loss": 0.3467,
2399
+ "step": 3400
2400
+ },
2401
+ {
2402
+ "epoch": 1.58,
2403
+ "learning_rate": 2.1687598116169546e-06,
2404
+ "loss": 0.3183,
2405
+ "step": 3410
2406
+ },
2407
+ {
2408
+ "epoch": 1.58,
2409
+ "learning_rate": 2.1452119309262166e-06,
2410
+ "loss": 0.3063,
2411
+ "step": 3420
2412
+ },
2413
+ {
2414
+ "epoch": 1.59,
2415
+ "learning_rate": 2.1216640502354787e-06,
2416
+ "loss": 0.286,
2417
+ "step": 3430
2418
+ },
2419
+ {
2420
+ "epoch": 1.59,
2421
+ "learning_rate": 2.098116169544741e-06,
2422
+ "loss": 0.2963,
2423
+ "step": 3440
2424
+ },
2425
+ {
2426
+ "epoch": 1.6,
2427
+ "learning_rate": 2.0745682888540034e-06,
2428
+ "loss": 0.3189,
2429
+ "step": 3450
2430
+ },
2431
+ {
2432
+ "epoch": 1.6,
2433
+ "learning_rate": 2.0510204081632654e-06,
2434
+ "loss": 0.3128,
2435
+ "step": 3460
2436
+ },
2437
+ {
2438
+ "epoch": 1.61,
2439
+ "learning_rate": 2.0274725274725275e-06,
2440
+ "loss": 0.3463,
2441
+ "step": 3470
2442
+ },
2443
+ {
2444
+ "epoch": 1.61,
2445
+ "learning_rate": 2.0039246467817896e-06,
2446
+ "loss": 0.2914,
2447
+ "step": 3480
2448
+ },
2449
+ {
2450
+ "epoch": 1.61,
2451
+ "learning_rate": 1.9803767660910517e-06,
2452
+ "loss": 0.3021,
2453
+ "step": 3490
2454
+ },
2455
+ {
2456
+ "epoch": 1.62,
2457
+ "learning_rate": 1.9568288854003142e-06,
2458
+ "loss": 0.3168,
2459
+ "step": 3500
2460
+ },
2461
+ {
2462
+ "epoch": 1.62,
2463
+ "eval_webgpt_accuracy": 0.5870786516853933,
2464
+ "eval_webgpt_loss": 0.6775321364402771,
2465
+ "eval_webgpt_runtime": 153.1117,
2466
+ "eval_webgpt_samples_per_second": 25.576,
2467
+ "eval_webgpt_steps_per_second": 2.56,
2468
+ "step": 3500
2469
+ },
2470
+ {
2471
+ "epoch": 1.62,
2472
+ "eval_hfsummary_accuracy": 0.6833116706465556,
2473
+ "eval_hfsummary_loss": 0.7361535429954529,
2474
+ "eval_hfsummary_runtime": 2478.5992,
2475
+ "eval_hfsummary_samples_per_second": 13.347,
2476
+ "eval_hfsummary_steps_per_second": 1.335,
2477
+ "step": 3500
2478
+ },
2479
+ {
2480
+ "epoch": 1.62,
2481
+ "eval_gptsynthetic_accuracy": 0.9984917043740573,
2482
+ "eval_gptsynthetic_loss": 0.005919534247368574,
2483
+ "eval_gptsynthetic_runtime": 116.9876,
2484
+ "eval_gptsynthetic_samples_per_second": 28.336,
2485
+ "eval_gptsynthetic_steps_per_second": 2.838,
2486
+ "step": 3500
2487
+ },
2488
+ {
2489
+ "epoch": 1.62,
2490
+ "learning_rate": 1.9332810047095763e-06,
2491
+ "loss": 0.3267,
2492
+ "step": 3510
2493
+ },
2494
+ {
2495
+ "epoch": 1.63,
2496
+ "learning_rate": 1.9097331240188384e-06,
2497
+ "loss": 0.3299,
2498
+ "step": 3520
2499
+ },
2500
+ {
2501
+ "epoch": 1.63,
2502
+ "learning_rate": 1.8861852433281007e-06,
2503
+ "loss": 0.3035,
2504
+ "step": 3530
2505
+ },
2506
+ {
2507
+ "epoch": 1.64,
2508
+ "learning_rate": 1.8626373626373626e-06,
2509
+ "loss": 0.2943,
2510
+ "step": 3540
2511
+ },
2512
+ {
2513
+ "epoch": 1.64,
2514
+ "learning_rate": 1.8390894819466247e-06,
2515
+ "loss": 0.287,
2516
+ "step": 3550
2517
+ },
2518
+ {
2519
+ "epoch": 1.65,
2520
+ "learning_rate": 1.815541601255887e-06,
2521
+ "loss": 0.3217,
2522
+ "step": 3560
2523
+ },
2524
+ {
2525
+ "epoch": 1.65,
2526
+ "learning_rate": 1.7919937205651491e-06,
2527
+ "loss": 0.3173,
2528
+ "step": 3570
2529
+ },
2530
+ {
2531
+ "epoch": 1.66,
2532
+ "learning_rate": 1.7684458398744112e-06,
2533
+ "loss": 0.319,
2534
+ "step": 3580
2535
+ },
2536
+ {
2537
+ "epoch": 1.66,
2538
+ "learning_rate": 1.7448979591836735e-06,
2539
+ "loss": 0.3236,
2540
+ "step": 3590
2541
+ },
2542
+ {
2543
+ "epoch": 1.67,
2544
+ "learning_rate": 1.7213500784929356e-06,
2545
+ "loss": 0.3055,
2546
+ "step": 3600
2547
+ },
2548
+ {
2549
+ "epoch": 1.67,
2550
+ "learning_rate": 1.697802197802198e-06,
2551
+ "loss": 0.3119,
2552
+ "step": 3610
2553
+ },
2554
+ {
2555
+ "epoch": 1.67,
2556
+ "learning_rate": 1.67425431711146e-06,
2557
+ "loss": 0.3349,
2558
+ "step": 3620
2559
+ },
2560
+ {
2561
+ "epoch": 1.68,
2562
+ "learning_rate": 1.650706436420722e-06,
2563
+ "loss": 0.2947,
2564
+ "step": 3630
2565
+ },
2566
+ {
2567
+ "epoch": 1.68,
2568
+ "learning_rate": 1.6271585557299844e-06,
2569
+ "loss": 0.3285,
2570
+ "step": 3640
2571
+ },
2572
+ {
2573
+ "epoch": 1.69,
2574
+ "learning_rate": 1.6036106750392465e-06,
2575
+ "loss": 0.3137,
2576
+ "step": 3650
2577
+ },
2578
+ {
2579
+ "epoch": 1.69,
2580
+ "learning_rate": 1.5800627943485088e-06,
2581
+ "loss": 0.2999,
2582
+ "step": 3660
2583
+ },
2584
+ {
2585
+ "epoch": 1.7,
2586
+ "learning_rate": 1.5565149136577709e-06,
2587
+ "loss": 0.3313,
2588
+ "step": 3670
2589
+ },
2590
+ {
2591
+ "epoch": 1.7,
2592
+ "learning_rate": 1.532967032967033e-06,
2593
+ "loss": 0.3295,
2594
+ "step": 3680
2595
+ },
2596
+ {
2597
+ "epoch": 1.71,
2598
+ "learning_rate": 1.5094191522762953e-06,
2599
+ "loss": 0.2978,
2600
+ "step": 3690
2601
+ },
2602
+ {
2603
+ "epoch": 1.71,
2604
+ "learning_rate": 1.4858712715855574e-06,
2605
+ "loss": 0.3036,
2606
+ "step": 3700
2607
+ },
2608
+ {
2609
+ "epoch": 1.72,
2610
+ "learning_rate": 1.4623233908948197e-06,
2611
+ "loss": 0.3072,
2612
+ "step": 3710
2613
+ },
2614
+ {
2615
+ "epoch": 1.72,
2616
+ "learning_rate": 1.4387755102040818e-06,
2617
+ "loss": 0.3102,
2618
+ "step": 3720
2619
+ },
2620
+ {
2621
+ "epoch": 1.73,
2622
+ "learning_rate": 1.4152276295133439e-06,
2623
+ "loss": 0.3215,
2624
+ "step": 3730
2625
+ },
2626
+ {
2627
+ "epoch": 1.73,
2628
+ "learning_rate": 1.3916797488226062e-06,
2629
+ "loss": 0.3018,
2630
+ "step": 3740
2631
+ },
2632
+ {
2633
+ "epoch": 1.74,
2634
+ "learning_rate": 1.368131868131868e-06,
2635
+ "loss": 0.312,
2636
+ "step": 3750
2637
+ },
2638
+ {
2639
+ "epoch": 1.74,
2640
+ "eval_webgpt_accuracy": 0.5873340143003064,
2641
+ "eval_webgpt_loss": 0.6725783348083496,
2642
+ "eval_webgpt_runtime": 153.1463,
2643
+ "eval_webgpt_samples_per_second": 25.57,
2644
+ "eval_webgpt_steps_per_second": 2.56,
2645
+ "step": 3750
2646
+ },
2647
+ {
2648
+ "epoch": 1.74,
2649
+ "eval_hfsummary_accuracy": 0.6811957803101291,
2650
+ "eval_hfsummary_loss": 0.7275147438049316,
2651
+ "eval_hfsummary_runtime": 2480.3805,
2652
+ "eval_hfsummary_samples_per_second": 13.338,
2653
+ "eval_hfsummary_steps_per_second": 1.334,
2654
+ "step": 3750
2655
+ },
2656
+ {
2657
+ "epoch": 1.74,
2658
+ "eval_gptsynthetic_accuracy": 0.9984917043740573,
2659
+ "eval_gptsynthetic_loss": 0.0056294966489076614,
2660
+ "eval_gptsynthetic_runtime": 116.9973,
2661
+ "eval_gptsynthetic_samples_per_second": 28.334,
2662
+ "eval_gptsynthetic_steps_per_second": 2.838,
2663
+ "step": 3750
2664
+ },
2665
+ {
2666
+ "epoch": 1.74,
2667
+ "learning_rate": 1.3445839874411302e-06,
2668
+ "loss": 0.3263,
2669
+ "step": 3760
2670
+ },
2671
+ {
2672
+ "epoch": 1.74,
2673
+ "learning_rate": 1.3210361067503925e-06,
2674
+ "loss": 0.3066,
2675
+ "step": 3770
2676
+ },
2677
+ {
2678
+ "epoch": 1.75,
2679
+ "learning_rate": 1.2974882260596545e-06,
2680
+ "loss": 0.2967,
2681
+ "step": 3780
2682
+ },
2683
+ {
2684
+ "epoch": 1.75,
2685
+ "learning_rate": 1.2739403453689169e-06,
2686
+ "loss": 0.3409,
2687
+ "step": 3790
2688
+ },
2689
+ {
2690
+ "epoch": 1.76,
2691
+ "learning_rate": 1.250392464678179e-06,
2692
+ "loss": 0.3158,
2693
+ "step": 3800
2694
+ },
2695
+ {
2696
+ "epoch": 1.76,
2697
+ "learning_rate": 1.226844583987441e-06,
2698
+ "loss": 0.3123,
2699
+ "step": 3810
2700
+ },
2701
+ {
2702
+ "epoch": 1.77,
2703
+ "learning_rate": 1.2032967032967033e-06,
2704
+ "loss": 0.3299,
2705
+ "step": 3820
2706
+ },
2707
+ {
2708
+ "epoch": 1.77,
2709
+ "learning_rate": 1.1797488226059654e-06,
2710
+ "loss": 0.2974,
2711
+ "step": 3830
2712
+ },
2713
+ {
2714
+ "epoch": 1.78,
2715
+ "learning_rate": 1.1562009419152277e-06,
2716
+ "loss": 0.3047,
2717
+ "step": 3840
2718
+ },
2719
+ {
2720
+ "epoch": 1.78,
2721
+ "learning_rate": 1.1326530612244898e-06,
2722
+ "loss": 0.2854,
2723
+ "step": 3850
2724
+ },
2725
+ {
2726
+ "epoch": 1.79,
2727
+ "learning_rate": 1.109105180533752e-06,
2728
+ "loss": 0.3379,
2729
+ "step": 3860
2730
+ },
2731
+ {
2732
+ "epoch": 1.79,
2733
+ "learning_rate": 1.0855572998430142e-06,
2734
+ "loss": 0.3147,
2735
+ "step": 3870
2736
+ },
2737
+ {
2738
+ "epoch": 1.8,
2739
+ "learning_rate": 1.0620094191522763e-06,
2740
+ "loss": 0.2994,
2741
+ "step": 3880
2742
+ },
2743
+ {
2744
+ "epoch": 1.8,
2745
+ "learning_rate": 1.0384615384615386e-06,
2746
+ "loss": 0.3146,
2747
+ "step": 3890
2748
+ },
2749
+ {
2750
+ "epoch": 1.8,
2751
+ "learning_rate": 1.0149136577708005e-06,
2752
+ "loss": 0.3412,
2753
+ "step": 3900
2754
+ },
2755
+ {
2756
+ "epoch": 1.81,
2757
+ "learning_rate": 9.913657770800628e-07,
2758
+ "loss": 0.285,
2759
+ "step": 3910
2760
+ },
2761
+ {
2762
+ "epoch": 1.81,
2763
+ "learning_rate": 9.67817896389325e-07,
2764
+ "loss": 0.3162,
2765
+ "step": 3920
2766
+ },
2767
+ {
2768
+ "epoch": 1.82,
2769
+ "learning_rate": 9.442700156985871e-07,
2770
+ "loss": 0.3123,
2771
+ "step": 3930
2772
+ },
2773
+ {
2774
+ "epoch": 1.82,
2775
+ "learning_rate": 9.207221350078493e-07,
2776
+ "loss": 0.3323,
2777
+ "step": 3940
2778
+ },
2779
+ {
2780
+ "epoch": 1.83,
2781
+ "learning_rate": 8.971742543171115e-07,
2782
+ "loss": 0.3072,
2783
+ "step": 3950
2784
+ },
2785
+ {
2786
+ "epoch": 1.83,
2787
+ "learning_rate": 8.736263736263737e-07,
2788
+ "loss": 0.3104,
2789
+ "step": 3960
2790
+ },
2791
+ {
2792
+ "epoch": 1.84,
2793
+ "learning_rate": 8.500784929356358e-07,
2794
+ "loss": 0.3,
2795
+ "step": 3970
2796
+ },
2797
+ {
2798
+ "epoch": 1.84,
2799
+ "learning_rate": 8.26530612244898e-07,
2800
+ "loss": 0.3235,
2801
+ "step": 3980
2802
+ },
2803
+ {
2804
+ "epoch": 1.85,
2805
+ "learning_rate": 8.029827315541602e-07,
2806
+ "loss": 0.3237,
2807
+ "step": 3990
2808
+ },
2809
+ {
2810
+ "epoch": 1.85,
2811
+ "learning_rate": 7.794348508634223e-07,
2812
+ "loss": 0.3086,
2813
+ "step": 4000
2814
+ },
2815
+ {
2816
+ "epoch": 1.85,
2817
+ "eval_webgpt_accuracy": 0.5875893769152196,
2818
+ "eval_webgpt_loss": 0.6754750609397888,
2819
+ "eval_webgpt_runtime": 153.2384,
2820
+ "eval_webgpt_samples_per_second": 25.555,
2821
+ "eval_webgpt_steps_per_second": 2.558,
2822
+ "step": 4000
2823
+ },
2824
+ {
2825
+ "epoch": 1.85,
2826
+ "eval_hfsummary_accuracy": 0.6865761871656137,
2827
+ "eval_hfsummary_loss": 0.7391700744628906,
2828
+ "eval_hfsummary_runtime": 2482.99,
2829
+ "eval_hfsummary_samples_per_second": 13.324,
2830
+ "eval_hfsummary_steps_per_second": 1.333,
2831
+ "step": 4000
2832
+ },
2833
+ {
2834
+ "epoch": 1.85,
2835
+ "eval_gptsynthetic_accuracy": 0.9984917043740573,
2836
+ "eval_gptsynthetic_loss": 0.006686400156468153,
2837
+ "eval_gptsynthetic_runtime": 116.9831,
2838
+ "eval_gptsynthetic_samples_per_second": 28.337,
2839
+ "eval_gptsynthetic_steps_per_second": 2.838,
2840
+ "step": 4000
2841
+ }
2842
+ ],
2843
+ "max_steps": 4322,
2844
+ "num_train_epochs": 2,
2845
+ "total_flos": 7.590150027767578e+17,
2846
+ "trial_name": null,
2847
+ "trial_params": null
2848
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:321aaabef7568e517bca0a26f23adae1622dcc40047f59d6454d24ef4335f240
3
+ size 3451
vocab.txt ADDED
The diff for this file is too large to render. See raw diff