agentlans commited on
Commit
5000b17
·
verified ·
1 Parent(s): 86eabe2

Upload 12 files

Browse files
README.md CHANGED
@@ -1,3 +1,65 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: EleutherAI/pythia-70m-deduped
5
+ tags:
6
+ - generated_from_trainer
7
+ metrics:
8
+ - accuracy
9
+ model-index:
10
+ - name: lmsys-prompts
11
+ results: []
12
+ ---
13
+
14
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
+ should probably proofread and complete it, then remove this comment. -->
16
+
17
+ # lmsys-prompts
18
+
19
+ This model is a fine-tuned version of [EleutherAI/pythia-70m-deduped](https://huggingface.co/EleutherAI/pythia-70m-deduped) on an unknown dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 2.6662
22
+ - Accuracy: 0.5068
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: 5.0
48
+
49
+ ### Training results
50
+
51
+ | Training Loss | Epoch | Step | Accuracy | Validation Loss |
52
+ |:-------------:|:-----:|:-----:|:--------:|:---------------:|
53
+ | 3.3254 | 1.0 | 4963 | 0.4213 | 3.2209 |
54
+ | 2.9236 | 2.0 | 9926 | 0.4686 | 2.9025 |
55
+ | 2.7526 | 3.0 | 14889 | 0.4861 | 2.7927 |
56
+ | 2.683 | 4.0 | 19852 | 2.7131 | 0.4999 |
57
+ | 2.6099 | 5.0 | 24815 | 2.6662 | 0.5068 |
58
+
59
+
60
+ ### Framework versions
61
+
62
+ - Transformers 4.51.3
63
+ - Pytorch 2.6.0+cu124
64
+ - Datasets 3.2.0
65
+ - Tokenizers 0.21.0
all_results.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 5.0,
3
+ "eval_accuracy": 0.5068343659501033,
4
+ "eval_loss": 2.6661746501922607,
5
+ "eval_runtime": 16.5477,
6
+ "eval_samples": 2191,
7
+ "eval_samples_per_second": 132.405,
8
+ "eval_steps_per_second": 16.558,
9
+ "perplexity": 14.384836777046193,
10
+ "total_flos": 5.448549676351488e+16,
11
+ "train_loss": 1.0647943758421696,
12
+ "train_runtime": 2230.6964,
13
+ "train_samples": 39704,
14
+ "train_samples_per_second": 88.995,
15
+ "train_steps_per_second": 11.124
16
+ }
config.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "GPTNeoXForCausalLM"
4
+ ],
5
+ "attention_bias": true,
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 0,
8
+ "classifier_dropout": 0.1,
9
+ "eos_token_id": 0,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout": 0.0,
12
+ "hidden_size": 512,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 2048,
15
+ "layer_norm_eps": 1e-05,
16
+ "max_position_embeddings": 2048,
17
+ "model_type": "gpt_neox",
18
+ "num_attention_heads": 8,
19
+ "num_hidden_layers": 6,
20
+ "partial_rotary_factor": 0.25,
21
+ "rope_scaling": null,
22
+ "rope_theta": 10000,
23
+ "rotary_emb_base": 10000,
24
+ "rotary_pct": 0.25,
25
+ "tie_word_embeddings": false,
26
+ "torch_dtype": "float32",
27
+ "transformers_version": "4.51.3",
28
+ "use_cache": true,
29
+ "use_parallel_residual": true,
30
+ "vocab_size": 50304
31
+ }
eval_results.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 5.0,
3
+ "eval_accuracy": 0.5068343659501033,
4
+ "eval_loss": 2.6661746501922607,
5
+ "eval_runtime": 16.5477,
6
+ "eval_samples": 2191,
7
+ "eval_samples_per_second": 132.405,
8
+ "eval_steps_per_second": 16.558,
9
+ "perplexity": 14.384836777046193
10
+ }
generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 0,
4
+ "eos_token_id": 0,
5
+ "transformers_version": "4.51.3"
6
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:77774ec91f7b86789c4c420f154a4fc6d5b3db78f4e5db43e5ea2b35de60c6df
3
+ size 281715176
special_tokens_map.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<|endoftext|>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "<|endoftext|>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "unk_token": {
17
+ "content": "<|endoftext|>",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ }
23
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,215 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": false,
3
+ "add_eos_token": false,
4
+ "add_prefix_space": false,
5
+ "added_tokens_decoder": {
6
+ "0": {
7
+ "content": "<|endoftext|>",
8
+ "lstrip": false,
9
+ "normalized": false,
10
+ "rstrip": false,
11
+ "single_word": false,
12
+ "special": true
13
+ },
14
+ "1": {
15
+ "content": "<|padding|>",
16
+ "lstrip": false,
17
+ "normalized": false,
18
+ "rstrip": false,
19
+ "single_word": false,
20
+ "special": true
21
+ },
22
+ "50254": {
23
+ "content": " ",
24
+ "lstrip": false,
25
+ "normalized": true,
26
+ "rstrip": false,
27
+ "single_word": false,
28
+ "special": false
29
+ },
30
+ "50255": {
31
+ "content": " ",
32
+ "lstrip": false,
33
+ "normalized": true,
34
+ "rstrip": false,
35
+ "single_word": false,
36
+ "special": false
37
+ },
38
+ "50256": {
39
+ "content": " ",
40
+ "lstrip": false,
41
+ "normalized": true,
42
+ "rstrip": false,
43
+ "single_word": false,
44
+ "special": false
45
+ },
46
+ "50257": {
47
+ "content": " ",
48
+ "lstrip": false,
49
+ "normalized": true,
50
+ "rstrip": false,
51
+ "single_word": false,
52
+ "special": false
53
+ },
54
+ "50258": {
55
+ "content": " ",
56
+ "lstrip": false,
57
+ "normalized": true,
58
+ "rstrip": false,
59
+ "single_word": false,
60
+ "special": false
61
+ },
62
+ "50259": {
63
+ "content": " ",
64
+ "lstrip": false,
65
+ "normalized": true,
66
+ "rstrip": false,
67
+ "single_word": false,
68
+ "special": false
69
+ },
70
+ "50260": {
71
+ "content": " ",
72
+ "lstrip": false,
73
+ "normalized": true,
74
+ "rstrip": false,
75
+ "single_word": false,
76
+ "special": false
77
+ },
78
+ "50261": {
79
+ "content": " ",
80
+ "lstrip": false,
81
+ "normalized": true,
82
+ "rstrip": false,
83
+ "single_word": false,
84
+ "special": false
85
+ },
86
+ "50262": {
87
+ "content": " ",
88
+ "lstrip": false,
89
+ "normalized": true,
90
+ "rstrip": false,
91
+ "single_word": false,
92
+ "special": false
93
+ },
94
+ "50263": {
95
+ "content": " ",
96
+ "lstrip": false,
97
+ "normalized": true,
98
+ "rstrip": false,
99
+ "single_word": false,
100
+ "special": false
101
+ },
102
+ "50264": {
103
+ "content": " ",
104
+ "lstrip": false,
105
+ "normalized": true,
106
+ "rstrip": false,
107
+ "single_word": false,
108
+ "special": false
109
+ },
110
+ "50265": {
111
+ "content": " ",
112
+ "lstrip": false,
113
+ "normalized": true,
114
+ "rstrip": false,
115
+ "single_word": false,
116
+ "special": false
117
+ },
118
+ "50266": {
119
+ "content": " ",
120
+ "lstrip": false,
121
+ "normalized": true,
122
+ "rstrip": false,
123
+ "single_word": false,
124
+ "special": false
125
+ },
126
+ "50267": {
127
+ "content": " ",
128
+ "lstrip": false,
129
+ "normalized": true,
130
+ "rstrip": false,
131
+ "single_word": false,
132
+ "special": false
133
+ },
134
+ "50268": {
135
+ "content": " ",
136
+ "lstrip": false,
137
+ "normalized": true,
138
+ "rstrip": false,
139
+ "single_word": false,
140
+ "special": false
141
+ },
142
+ "50269": {
143
+ "content": " ",
144
+ "lstrip": false,
145
+ "normalized": true,
146
+ "rstrip": false,
147
+ "single_word": false,
148
+ "special": false
149
+ },
150
+ "50270": {
151
+ "content": " ",
152
+ "lstrip": false,
153
+ "normalized": true,
154
+ "rstrip": false,
155
+ "single_word": false,
156
+ "special": false
157
+ },
158
+ "50271": {
159
+ "content": " ",
160
+ "lstrip": false,
161
+ "normalized": true,
162
+ "rstrip": false,
163
+ "single_word": false,
164
+ "special": false
165
+ },
166
+ "50272": {
167
+ "content": " ",
168
+ "lstrip": false,
169
+ "normalized": true,
170
+ "rstrip": false,
171
+ "single_word": false,
172
+ "special": false
173
+ },
174
+ "50273": {
175
+ "content": " ",
176
+ "lstrip": false,
177
+ "normalized": true,
178
+ "rstrip": false,
179
+ "single_word": false,
180
+ "special": false
181
+ },
182
+ "50274": {
183
+ "content": " ",
184
+ "lstrip": false,
185
+ "normalized": true,
186
+ "rstrip": false,
187
+ "single_word": false,
188
+ "special": false
189
+ },
190
+ "50275": {
191
+ "content": " ",
192
+ "lstrip": false,
193
+ "normalized": true,
194
+ "rstrip": false,
195
+ "single_word": false,
196
+ "special": false
197
+ },
198
+ "50276": {
199
+ "content": " ",
200
+ "lstrip": false,
201
+ "normalized": true,
202
+ "rstrip": false,
203
+ "single_word": false,
204
+ "special": false
205
+ }
206
+ },
207
+ "bos_token": "<|endoftext|>",
208
+ "clean_up_tokenization_spaces": false,
209
+ "eos_token": "<|endoftext|>",
210
+ "extra_special_tokens": {},
211
+ "model_max_length": 1000000000000000019884624838656,
212
+ "pad_token": null,
213
+ "tokenizer_class": "GPTNeoXTokenizer",
214
+ "unk_token": "<|endoftext|>"
215
+ }
train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 5.0,
3
+ "total_flos": 5.448549676351488e+16,
4
+ "train_loss": 1.0647943758421696,
5
+ "train_runtime": 2230.6964,
6
+ "train_samples": 39704,
7
+ "train_samples_per_second": 88.995,
8
+ "train_steps_per_second": 11.124
9
+ }
trainer_state.json ADDED
@@ -0,0 +1,431 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 24815,
3
+ "best_metric": 2.6661746501922607,
4
+ "best_model_checkpoint": "lmsys-prompts/checkpoint-24815",
5
+ "epoch": 5.0,
6
+ "eval_steps": 500,
7
+ "global_step": 24815,
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.10074551682450131,
14
+ "grad_norm": 216.5968017578125,
15
+ "learning_rate": 4.832426623681913e-05,
16
+ "loss": 5.5953,
17
+ "step": 500
18
+ },
19
+ {
20
+ "epoch": 0.20149103364900262,
21
+ "grad_norm": 90.29191589355469,
22
+ "learning_rate": 4.664517428974411e-05,
23
+ "loss": 4.1114,
24
+ "step": 1000
25
+ },
26
+ {
27
+ "epoch": 0.30223655047350395,
28
+ "grad_norm": 27.48844337463379,
29
+ "learning_rate": 4.496608234266908e-05,
30
+ "loss": 3.9607,
31
+ "step": 1500
32
+ },
33
+ {
34
+ "epoch": 0.40298206729800523,
35
+ "grad_norm": 29.192981719970703,
36
+ "learning_rate": 4.328699039559406e-05,
37
+ "loss": 3.7888,
38
+ "step": 2000
39
+ },
40
+ {
41
+ "epoch": 0.5037275841225065,
42
+ "grad_norm": 27.68057632446289,
43
+ "learning_rate": 4.160789844851904e-05,
44
+ "loss": 3.6636,
45
+ "step": 2500
46
+ },
47
+ {
48
+ "epoch": 0.6044731009470079,
49
+ "grad_norm": 26.73339080810547,
50
+ "learning_rate": 3.992880650144402e-05,
51
+ "loss": 3.5414,
52
+ "step": 3000
53
+ },
54
+ {
55
+ "epoch": 0.7052186177715092,
56
+ "grad_norm": 13.406410217285156,
57
+ "learning_rate": 3.8249714554368995e-05,
58
+ "loss": 3.467,
59
+ "step": 3500
60
+ },
61
+ {
62
+ "epoch": 0.8059641345960105,
63
+ "grad_norm": 17.33591079711914,
64
+ "learning_rate": 3.657062260729397e-05,
65
+ "loss": 3.3742,
66
+ "step": 4000
67
+ },
68
+ {
69
+ "epoch": 0.9067096514205117,
70
+ "grad_norm": 11.469970703125,
71
+ "learning_rate": 3.4891530660218955e-05,
72
+ "loss": 3.3254,
73
+ "step": 4500
74
+ },
75
+ {
76
+ "epoch": 1.0,
77
+ "eval_accuracy": 0.42128131925101936,
78
+ "eval_loss": 3.2208595275878906,
79
+ "eval_runtime": 16.4691,
80
+ "eval_samples_per_second": 133.037,
81
+ "eval_steps_per_second": 16.637,
82
+ "step": 4963
83
+ },
84
+ {
85
+ "epoch": 1.007455168245013,
86
+ "grad_norm": 10.576227188110352,
87
+ "learning_rate": 3.321243871314393e-05,
88
+ "loss": 3.2632,
89
+ "step": 5000
90
+ },
91
+ {
92
+ "epoch": 1.1082006850695145,
93
+ "grad_norm": 10.408350944519043,
94
+ "learning_rate": 3.153334676606891e-05,
95
+ "loss": 3.177,
96
+ "step": 5500
97
+ },
98
+ {
99
+ "epoch": 1.2089462018940158,
100
+ "grad_norm": 8.68959903717041,
101
+ "learning_rate": 2.9854254818993887e-05,
102
+ "loss": 3.1321,
103
+ "step": 6000
104
+ },
105
+ {
106
+ "epoch": 1.309691718718517,
107
+ "grad_norm": 10.844682693481445,
108
+ "learning_rate": 2.817516287191887e-05,
109
+ "loss": 3.0864,
110
+ "step": 6500
111
+ },
112
+ {
113
+ "epoch": 1.4104372355430184,
114
+ "grad_norm": 11.559004783630371,
115
+ "learning_rate": 2.6496070924843847e-05,
116
+ "loss": 3.0802,
117
+ "step": 7000
118
+ },
119
+ {
120
+ "epoch": 1.5111827523675196,
121
+ "grad_norm": 11.332049369812012,
122
+ "learning_rate": 2.4816978977768823e-05,
123
+ "loss": 3.021,
124
+ "step": 7500
125
+ },
126
+ {
127
+ "epoch": 1.611928269192021,
128
+ "grad_norm": 8.480116844177246,
129
+ "learning_rate": 2.3137887030693803e-05,
130
+ "loss": 3.0068,
131
+ "step": 8000
132
+ },
133
+ {
134
+ "epoch": 1.7126737860165222,
135
+ "grad_norm": 11.450610160827637,
136
+ "learning_rate": 2.145879508361878e-05,
137
+ "loss": 2.9865,
138
+ "step": 8500
139
+ },
140
+ {
141
+ "epoch": 1.8134193028410235,
142
+ "grad_norm": 7.126410007476807,
143
+ "learning_rate": 1.977970313654376e-05,
144
+ "loss": 2.946,
145
+ "step": 9000
146
+ },
147
+ {
148
+ "epoch": 1.9141648196655248,
149
+ "grad_norm": 9.249116897583008,
150
+ "learning_rate": 1.8100611189468735e-05,
151
+ "loss": 2.9236,
152
+ "step": 9500
153
+ },
154
+ {
155
+ "epoch": 2.0,
156
+ "eval_accuracy": 0.4685653073780457,
157
+ "eval_loss": 2.902540445327759,
158
+ "eval_runtime": 17.401,
159
+ "eval_samples_per_second": 125.912,
160
+ "eval_steps_per_second": 15.746,
161
+ "step": 9926
162
+ },
163
+ {
164
+ "epoch": 2.014910336490026,
165
+ "grad_norm": 7.414788246154785,
166
+ "learning_rate": 1.6421519242393715e-05,
167
+ "loss": 2.9077,
168
+ "step": 10000
169
+ },
170
+ {
171
+ "epoch": 2.1156558533145273,
172
+ "grad_norm": 10.728218078613281,
173
+ "learning_rate": 1.4742427295318693e-05,
174
+ "loss": 2.8403,
175
+ "step": 10500
176
+ },
177
+ {
178
+ "epoch": 2.216401370139029,
179
+ "grad_norm": 9.297683715820312,
180
+ "learning_rate": 1.3063335348243671e-05,
181
+ "loss": 2.8482,
182
+ "step": 11000
183
+ },
184
+ {
185
+ "epoch": 2.3171468869635303,
186
+ "grad_norm": 6.445867538452148,
187
+ "learning_rate": 1.1384243401168649e-05,
188
+ "loss": 2.8066,
189
+ "step": 11500
190
+ },
191
+ {
192
+ "epoch": 2.4178924037880316,
193
+ "grad_norm": 8.223403930664062,
194
+ "learning_rate": 9.705151454093627e-06,
195
+ "loss": 2.8088,
196
+ "step": 12000
197
+ },
198
+ {
199
+ "epoch": 2.518637920612533,
200
+ "grad_norm": 6.288245677947998,
201
+ "learning_rate": 8.026059507018605e-06,
202
+ "loss": 2.7869,
203
+ "step": 12500
204
+ },
205
+ {
206
+ "epoch": 2.619383437437034,
207
+ "grad_norm": 7.528985977172852,
208
+ "learning_rate": 6.346967559943583e-06,
209
+ "loss": 2.7774,
210
+ "step": 13000
211
+ },
212
+ {
213
+ "epoch": 2.7201289542615354,
214
+ "grad_norm": 8.420631408691406,
215
+ "learning_rate": 4.667875612868561e-06,
216
+ "loss": 2.7582,
217
+ "step": 13500
218
+ },
219
+ {
220
+ "epoch": 2.8208744710860367,
221
+ "grad_norm": 7.883354187011719,
222
+ "learning_rate": 2.988783665793539e-06,
223
+ "loss": 2.7613,
224
+ "step": 14000
225
+ },
226
+ {
227
+ "epoch": 2.921619987910538,
228
+ "grad_norm": 7.071943759918213,
229
+ "learning_rate": 1.309691718718517e-06,
230
+ "loss": 2.7526,
231
+ "step": 14500
232
+ },
233
+ {
234
+ "epoch": 3.0,
235
+ "eval_accuracy": 0.48612581550848066,
236
+ "eval_loss": 2.79274320602417,
237
+ "eval_runtime": 16.6412,
238
+ "eval_samples_per_second": 131.661,
239
+ "eval_steps_per_second": 16.465,
240
+ "step": 14889
241
+ },
242
+ {
243
+ "epoch": 3.0223655047350393,
244
+ "grad_norm": 7.703913688659668,
245
+ "learning_rate": 1.97783598629861e-05,
246
+ "loss": 2.7412,
247
+ "step": 15000
248
+ },
249
+ {
250
+ "epoch": 3.1231110215595406,
251
+ "grad_norm": 9.797307968139648,
252
+ "learning_rate": 1.8770904694741083e-05,
253
+ "loss": 2.7521,
254
+ "step": 15500
255
+ },
256
+ {
257
+ "epoch": 3.223856538384042,
258
+ "grad_norm": 11.222009658813477,
259
+ "learning_rate": 1.776344952649607e-05,
260
+ "loss": 2.7435,
261
+ "step": 16000
262
+ },
263
+ {
264
+ "epoch": 3.324602055208543,
265
+ "grad_norm": 9.812458038330078,
266
+ "learning_rate": 1.675599435825106e-05,
267
+ "loss": 2.7507,
268
+ "step": 16500
269
+ },
270
+ {
271
+ "epoch": 3.4253475720330444,
272
+ "grad_norm": 7.502840995788574,
273
+ "learning_rate": 1.5748539190006045e-05,
274
+ "loss": 2.7181,
275
+ "step": 17000
276
+ },
277
+ {
278
+ "epoch": 3.5260930888575457,
279
+ "grad_norm": 6.958070278167725,
280
+ "learning_rate": 1.4741084021761032e-05,
281
+ "loss": 2.7264,
282
+ "step": 17500
283
+ },
284
+ {
285
+ "epoch": 3.626838605682047,
286
+ "grad_norm": 7.174421787261963,
287
+ "learning_rate": 1.373362885351602e-05,
288
+ "loss": 2.7289,
289
+ "step": 18000
290
+ },
291
+ {
292
+ "epoch": 3.7275841225065482,
293
+ "grad_norm": 5.985280990600586,
294
+ "learning_rate": 1.2726173685271006e-05,
295
+ "loss": 2.6752,
296
+ "step": 18500
297
+ },
298
+ {
299
+ "epoch": 3.8283296393310495,
300
+ "grad_norm": 7.28540563583374,
301
+ "learning_rate": 1.1718718517025991e-05,
302
+ "loss": 2.6896,
303
+ "step": 19000
304
+ },
305
+ {
306
+ "epoch": 3.929075156155551,
307
+ "grad_norm": 9.331765174865723,
308
+ "learning_rate": 1.071126334878098e-05,
309
+ "loss": 2.683,
310
+ "step": 19500
311
+ },
312
+ {
313
+ "epoch": 4.0,
314
+ "eval_accuracy": 0.4999047467356238,
315
+ "eval_loss": 2.7130751609802246,
316
+ "eval_runtime": 16.5004,
317
+ "eval_samples_per_second": 132.785,
318
+ "eval_steps_per_second": 16.606,
319
+ "step": 19852
320
+ },
321
+ {
322
+ "epoch": 4.029820672980052,
323
+ "grad_norm": 8.58707046508789,
324
+ "learning_rate": 9.703808180535967e-06,
325
+ "loss": 2.6525,
326
+ "step": 20000
327
+ },
328
+ {
329
+ "epoch": 4.130566189804553,
330
+ "grad_norm": 7.965948581695557,
331
+ "learning_rate": 8.696353012290953e-06,
332
+ "loss": 2.6099,
333
+ "step": 20500
334
+ },
335
+ {
336
+ "epoch": 4.231311706629055,
337
+ "grad_norm": 5.951071739196777,
338
+ "learning_rate": 7.68889784404594e-06,
339
+ "loss": 2.633,
340
+ "step": 21000
341
+ },
342
+ {
343
+ "epoch": 4.332057223453556,
344
+ "grad_norm": 6.427402973175049,
345
+ "learning_rate": 6.681442675800926e-06,
346
+ "loss": 2.6239,
347
+ "step": 21500
348
+ },
349
+ {
350
+ "epoch": 4.432802740278058,
351
+ "grad_norm": 6.754240989685059,
352
+ "learning_rate": 5.673987507555914e-06,
353
+ "loss": 2.6033,
354
+ "step": 22000
355
+ },
356
+ {
357
+ "epoch": 4.533548257102559,
358
+ "grad_norm": 6.711804389953613,
359
+ "learning_rate": 4.666532339310901e-06,
360
+ "loss": 2.6101,
361
+ "step": 22500
362
+ },
363
+ {
364
+ "epoch": 4.634293773927061,
365
+ "grad_norm": 7.764159202575684,
366
+ "learning_rate": 3.6590771710658874e-06,
367
+ "loss": 2.6117,
368
+ "step": 23000
369
+ },
370
+ {
371
+ "epoch": 4.735039290751562,
372
+ "grad_norm": 8.959000587463379,
373
+ "learning_rate": 2.6516220028208746e-06,
374
+ "loss": 2.6101,
375
+ "step": 23500
376
+ },
377
+ {
378
+ "epoch": 4.835784807576063,
379
+ "grad_norm": 6.05490255355835,
380
+ "learning_rate": 1.6441668345758613e-06,
381
+ "loss": 2.5942,
382
+ "step": 24000
383
+ },
384
+ {
385
+ "epoch": 4.9365303244005645,
386
+ "grad_norm": 6.731525897979736,
387
+ "learning_rate": 6.367116663308483e-07,
388
+ "loss": 2.6099,
389
+ "step": 24500
390
+ },
391
+ {
392
+ "epoch": 5.0,
393
+ "eval_accuracy": 0.5068343659501033,
394
+ "eval_loss": 2.6661746501922607,
395
+ "eval_runtime": 16.4299,
396
+ "eval_samples_per_second": 133.355,
397
+ "eval_steps_per_second": 16.677,
398
+ "step": 24815
399
+ },
400
+ {
401
+ "epoch": 5.0,
402
+ "step": 24815,
403
+ "total_flos": 5.448549676351488e+16,
404
+ "train_loss": 1.0647943758421696,
405
+ "train_runtime": 2230.6964,
406
+ "train_samples_per_second": 88.995,
407
+ "train_steps_per_second": 11.124
408
+ }
409
+ ],
410
+ "logging_steps": 500,
411
+ "max_steps": 24815,
412
+ "num_input_tokens_seen": 0,
413
+ "num_train_epochs": 5,
414
+ "save_steps": 500,
415
+ "stateful_callbacks": {
416
+ "TrainerControl": {
417
+ "args": {
418
+ "should_epoch_stop": false,
419
+ "should_evaluate": false,
420
+ "should_log": false,
421
+ "should_save": true,
422
+ "should_training_stop": true
423
+ },
424
+ "attributes": {}
425
+ }
426
+ },
427
+ "total_flos": 5.448549676351488e+16,
428
+ "train_batch_size": 8,
429
+ "trial_name": null,
430
+ "trial_params": null
431
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:23c44b75cf4380a9e687930934c69ecf9c853747fb8d22f05f69f3e5b075d802
3
+ size 5304