AlekseyKorshuk commited on
Commit
0063a18
1 Parent(s): ec3024d

huggingartists

Browse files
README.md ADDED
@@ -0,0 +1,97 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ datasets:
4
+ - huggingartists/oxxxymiron
5
+ tags:
6
+ - huggingartists
7
+ - lyrics
8
+ - lm-head
9
+ - causal-lm
10
+ widget:
11
+ - text: "I am"
12
+ ---
13
+
14
+ <div class="inline-flex flex-col" style="line-height: 1.5;">
15
+ <div class="flex">
16
+ <div
17
+ style="display:DISPLAY_1; margin-left: auto; margin-right: auto; width: 92px; height:92px; border-radius: 50%; background-size: cover; background-image: url(&#39;https://images.genius.com/e525200b65911e414a9d38c348eb1c6b.667x667x1.jpg&#39;)">
18
+ </div>
19
+ </div>
20
+ <div style="text-align: center; margin-top: 3px; font-size: 16px; font-weight: 800">🤖 HuggingArtists Model 🤖</div>
21
+ <div style="text-align: center; font-size: 16px; font-weight: 800">Oxxxymiron</div>
22
+ <a href="https://genius.com/artists/oxxxymiron">
23
+ <div style="text-align: center; font-size: 14px;">@oxxxymiron</div>
24
+ </a>
25
+ </div>
26
+
27
+ I was made with [huggingartists](https://github.com/AlekseyKorshuk/huggingartists).
28
+
29
+ Create your own bot based on your favorite artist with [the demo](https://colab.research.google.com/github/AlekseyKorshuk/huggingartists/blob/master/huggingartists-demo.ipynb)!
30
+
31
+ ## How does it work?
32
+
33
+ To understand how the model was developed, check the [W&B report](https://wandb.ai/huggingartists/huggingartists/reportlist).
34
+
35
+ ## Training data
36
+
37
+ The model was trained on lyrics from Oxxxymiron.
38
+
39
+ Dataset is available [here](https://huggingface.co/datasets/huggingartists/oxxxymiron).
40
+ And can be used with:
41
+
42
+ ```python
43
+ from datasets import load_dataset
44
+
45
+ dataset = load_dataset("huggingartists/oxxxymiron")
46
+ ```
47
+
48
+ [Explore the data](https://wandb.ai/huggingartists/huggingartists/runs/1hspe90t/artifacts), which is tracked with [W&B artifacts](https://docs.wandb.com/artifacts) at every step of the pipeline.
49
+
50
+ ## Training procedure
51
+
52
+ The model is based on a pre-trained [GPT-2](https://huggingface.co/gpt2) which is fine-tuned on Oxxxymiron's lyrics.
53
+
54
+ Hyperparameters and metrics are recorded in the [W&B training run](https://wandb.ai/huggingartists/huggingartists/runs/128ekp1a) for full transparency and reproducibility.
55
+
56
+ At the end of training, [the final model](https://wandb.ai/huggingartists/huggingartists/runs/128ekp1a/artifacts) is logged and versioned.
57
+
58
+ ## How to use
59
+
60
+ You can use this model directly with a pipeline for text generation:
61
+
62
+ ```python
63
+ from transformers import pipeline
64
+ generator = pipeline('text-generation',
65
+ model='huggingartists/oxxxymiron')
66
+ generator("I am", num_return_sequences=5)
67
+ ```
68
+
69
+ Or with Transformers library:
70
+
71
+ ```python
72
+ from transformers import AutoTokenizer, AutoModelWithLMHead
73
+
74
+ tokenizer = AutoTokenizer.from_pretrained("huggingartists/oxxxymiron")
75
+
76
+ model = AutoModelWithLMHead.from_pretrained("huggingartists/oxxxymiron")
77
+ ```
78
+
79
+ ## Limitations and bias
80
+
81
+ The model suffers from [the same limitations and bias as GPT-2](https://huggingface.co/gpt2#limitations-and-bias).
82
+
83
+ In addition, the data present in the user's tweets further affects the text generated by the model.
84
+
85
+ ## About
86
+
87
+ *Built by Aleksey Korshuk*
88
+
89
+ [![Follow](https://img.shields.io/github/followers/AlekseyKorshuk?style=social)](https://github.com/AlekseyKorshuk)
90
+
91
+ [![Follow](https://img.shields.io/twitter/follow/alekseykorshuk?style=social)](https://twitter.com/intent/follow?screen_name=alekseykorshuk)
92
+
93
+ [![Follow](https://img.shields.io/badge/dynamic/json?color=blue&label=Telegram%20Channel&query=%24.result&url=https%3A%2F%2Fapi.telegram.org%2Fbot1929545866%3AAAFGhV-KKnegEcLiyYJxsc4zV6C-bdPEBtQ%2FgetChatMemberCount%3Fchat_id%3D-1001253621662&style=social&logo=telegram)](https://t.me/joinchat/_CQ04KjcJ-4yZTky)
94
+
95
+ For more details, visit the project repository.
96
+
97
+ [![GitHub stars](https://img.shields.io/github/stars/AlekseyKorshuk/huggingartists?style=social)](https://github.com/AlekseyKorshuk/huggingartists)
config.json ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "gpt2",
3
+ "activation_function": "gelu_new",
4
+ "architectures": [
5
+ "GPT2LMHeadModel"
6
+ ],
7
+ "attn_pdrop": 0.1,
8
+ "bos_token_id": 50256,
9
+ "embd_pdrop": 0.1,
10
+ "eos_token_id": 50256,
11
+ "gradient_checkpointing": false,
12
+ "initializer_range": 0.02,
13
+ "layer_norm_epsilon": 1e-05,
14
+ "model_type": "gpt2",
15
+ "n_ctx": 1024,
16
+ "n_embd": 768,
17
+ "n_head": 12,
18
+ "n_inner": null,
19
+ "n_layer": 12,
20
+ "n_positions": 1024,
21
+ "resid_pdrop": 0.1,
22
+ "scale_attn_weights": true,
23
+ "summary_activation": null,
24
+ "summary_first_dropout": 0.1,
25
+ "summary_proj_to_labels": true,
26
+ "summary_type": "cls_index",
27
+ "summary_use_proj": true,
28
+ "task_specific_params": {
29
+ "text-generation": {
30
+ "do_sample": true,
31
+ "max_length": 200,
32
+ "min_length": 100,
33
+ "temperature": 1.0,
34
+ "top_p": 0.95
35
+ }
36
+ },
37
+ "torch_dtype": "float32",
38
+ "transformers_version": "4.9.2",
39
+ "use_cache": true,
40
+ "vocab_size": 50257
41
+ }
flax_model.msgpack ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ac4cf6a4bf61083f345dafb51f48b425222e4be63498c60332f05a6f2e0e22b
3
+ size 497764120
merges.txt ADDED
The diff for this file is too large to render. See raw diff
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2e93f0da009d8b801c5faae8d9843a872ebfab4e3039cb0bffa0fa0875fad771
3
+ size 995604017
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0b291f749d6354df814b9ba0c5b259b0e94ea9e3d97b132031e674298a1ad7e1
3
+ size 510403817
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6d74db31a0d8982efe9dc40546a3693dc5af9aacf52ce2f4a7551ad7e3e1e232
3
+ size 14503
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:24e3a97a9e192b59907cc3ef0039e0bc2d4534437003eccea3487bcaea3049db
3
+ size 623
special_tokens_map.json ADDED
@@ -0,0 +1 @@
 
1
+ {"bos_token": "<|endoftext|>", "eos_token": "<|endoftext|>", "unk_token": "<|endoftext|>"}
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
tokenizer_config.json ADDED
@@ -0,0 +1 @@
 
1
+ {"unk_token": "<|endoftext|>", "bos_token": "<|endoftext|>", "eos_token": "<|endoftext|>", "add_prefix_space": false, "model_max_length": 1024, "special_tokens_map_file": null, "name_or_path": "gpt2", "tokenizer_class": "GPT2Tokenizer"}
trainer_state.json ADDED
@@ -0,0 +1,946 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.0,
5
+ "global_step": 777,
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.02,
12
+ "learning_rate": 0.00013707387498762246,
13
+ "loss": 2.6224,
14
+ "step": 5
15
+ },
16
+ {
17
+ "epoch": 0.04,
18
+ "learning_rate": 0.00013669596372654658,
19
+ "loss": 2.5403,
20
+ "step": 10
21
+ },
22
+ {
23
+ "epoch": 0.06,
24
+ "learning_rate": 0.00013606765583958522,
25
+ "loss": 2.4692,
26
+ "step": 15
27
+ },
28
+ {
29
+ "epoch": 0.08,
30
+ "learning_rate": 0.00013519126168650572,
31
+ "loss": 2.3553,
32
+ "step": 20
33
+ },
34
+ {
35
+ "epoch": 0.1,
36
+ "learning_rate": 0.00013407000386857356,
37
+ "loss": 2.2478,
38
+ "step": 25
39
+ },
40
+ {
41
+ "epoch": 0.12,
42
+ "learning_rate": 0.0001327080053786798,
43
+ "loss": 2.4781,
44
+ "step": 30
45
+ },
46
+ {
47
+ "epoch": 0.14,
48
+ "learning_rate": 0.0001311102744406261,
49
+ "loss": 2.2512,
50
+ "step": 35
51
+ },
52
+ {
53
+ "epoch": 0.15,
54
+ "learning_rate": 0.00012928268609331444,
55
+ "loss": 2.2522,
56
+ "step": 40
57
+ },
58
+ {
59
+ "epoch": 0.17,
60
+ "learning_rate": 0.00012723196058755901,
61
+ "loss": 2.4658,
62
+ "step": 45
63
+ },
64
+ {
65
+ "epoch": 0.19,
66
+ "learning_rate": 0.00012496563867495734,
67
+ "loss": 2.2714,
68
+ "step": 50
69
+ },
70
+ {
71
+ "epoch": 0.21,
72
+ "learning_rate": 0.0001224920538796866,
73
+ "loss": 2.2569,
74
+ "step": 55
75
+ },
76
+ {
77
+ "epoch": 0.23,
78
+ "learning_rate": 0.00011982030185518478,
79
+ "loss": 2.2897,
80
+ "step": 60
81
+ },
82
+ {
83
+ "epoch": 0.25,
84
+ "learning_rate": 0.00011696020693839531,
85
+ "loss": 2.1927,
86
+ "step": 65
87
+ },
88
+ {
89
+ "epoch": 0.27,
90
+ "learning_rate": 0.00011392228602455959,
91
+ "loss": 2.1867,
92
+ "step": 70
93
+ },
94
+ {
95
+ "epoch": 0.29,
96
+ "learning_rate": 0.00011071770989539364,
97
+ "loss": 2.1953,
98
+ "step": 75
99
+ },
100
+ {
101
+ "epoch": 0.31,
102
+ "learning_rate": 0.00010735826214284955,
103
+ "loss": 2.2485,
104
+ "step": 80
105
+ },
106
+ {
107
+ "epoch": 0.33,
108
+ "learning_rate": 0.00010385629583950372,
109
+ "loss": 2.1775,
110
+ "step": 85
111
+ },
112
+ {
113
+ "epoch": 0.35,
114
+ "learning_rate": 0.00010022468811490008,
115
+ "loss": 2.2653,
116
+ "step": 90
117
+ },
118
+ {
119
+ "epoch": 0.37,
120
+ "learning_rate": 9.647679280487579e-05,
121
+ "loss": 2.2253,
122
+ "step": 95
123
+ },
124
+ {
125
+ "epoch": 0.39,
126
+ "learning_rate": 9.262639134798382e-05,
127
+ "loss": 2.2196,
128
+ "step": 100
129
+ },
130
+ {
131
+ "epoch": 0.41,
132
+ "learning_rate": 8.868764210957135e-05,
133
+ "loss": 2.1454,
134
+ "step": 105
135
+ },
136
+ {
137
+ "epoch": 0.42,
138
+ "learning_rate": 8.467502831985555e-05,
139
+ "loss": 2.1683,
140
+ "step": 110
141
+ },
142
+ {
143
+ "epoch": 0.44,
144
+ "learning_rate": 8.060330481743385e-05,
145
+ "loss": 2.1135,
146
+ "step": 115
147
+ },
148
+ {
149
+ "epoch": 0.46,
150
+ "learning_rate": 7.648744379405968e-05,
151
+ "loss": 2.2169,
152
+ "step": 120
153
+ },
154
+ {
155
+ "epoch": 0.48,
156
+ "learning_rate": 7.234257974018543e-05,
157
+ "loss": 2.0504,
158
+ "step": 125
159
+ },
160
+ {
161
+ "epoch": 0.5,
162
+ "learning_rate": 6.818395379371482e-05,
163
+ "loss": 2.1956,
164
+ "step": 130
165
+ },
166
+ {
167
+ "epoch": 0.52,
168
+ "learning_rate": 6.40268576966004e-05,
169
+ "loss": 2.1556,
170
+ "step": 135
171
+ },
172
+ {
173
+ "epoch": 0.54,
174
+ "learning_rate": 5.9886577565364115e-05,
175
+ "loss": 2.1258,
176
+ "step": 140
177
+ },
178
+ {
179
+ "epoch": 0.56,
180
+ "learning_rate": 5.577833768230335e-05,
181
+ "loss": 2.0906,
182
+ "step": 145
183
+ },
184
+ {
185
+ "epoch": 0.58,
186
+ "learning_rate": 5.171724451406823e-05,
187
+ "loss": 2.0648,
188
+ "step": 150
189
+ },
190
+ {
191
+ "epoch": 0.6,
192
+ "learning_rate": 4.7718231163460816e-05,
193
+ "loss": 2.1031,
194
+ "step": 155
195
+ },
196
+ {
197
+ "epoch": 0.62,
198
+ "learning_rate": 4.379600245871268e-05,
199
+ "loss": 2.456,
200
+ "step": 160
201
+ },
202
+ {
203
+ "epoch": 0.64,
204
+ "learning_rate": 3.99649808821543e-05,
205
+ "loss": 2.1486,
206
+ "step": 165
207
+ },
208
+ {
209
+ "epoch": 0.66,
210
+ "learning_rate": 3.6239253537102286e-05,
211
+ "loss": 2.1327,
212
+ "step": 170
213
+ },
214
+ {
215
+ "epoch": 0.68,
216
+ "learning_rate": 3.263252034797391e-05,
217
+ "loss": 2.1147,
218
+ "step": 175
219
+ },
220
+ {
221
+ "epoch": 0.69,
222
+ "learning_rate": 2.9158043684102103e-05,
223
+ "loss": 2.0989,
224
+ "step": 180
225
+ },
226
+ {
227
+ "epoch": 0.71,
228
+ "learning_rate": 2.5828599592490882e-05,
229
+ "loss": 2.082,
230
+ "step": 185
231
+ },
232
+ {
233
+ "epoch": 0.73,
234
+ "learning_rate": 2.265643081883308e-05,
235
+ "loss": 2.0394,
236
+ "step": 190
237
+ },
238
+ {
239
+ "epoch": 0.75,
240
+ "learning_rate": 1.9653201789538038e-05,
241
+ "loss": 2.0346,
242
+ "step": 195
243
+ },
244
+ {
245
+ "epoch": 0.77,
246
+ "learning_rate": 1.682995572030529e-05,
247
+ "loss": 2.1082,
248
+ "step": 200
249
+ },
250
+ {
251
+ "epoch": 0.79,
252
+ "learning_rate": 1.419707400896067e-05,
253
+ "loss": 2.1204,
254
+ "step": 205
255
+ },
256
+ {
257
+ "epoch": 0.81,
258
+ "learning_rate": 1.1764238061872434e-05,
259
+ "loss": 1.976,
260
+ "step": 210
261
+ },
262
+ {
263
+ "epoch": 0.83,
264
+ "learning_rate": 9.540393694315812e-06,
265
+ "loss": 1.9811,
266
+ "step": 215
267
+ },
268
+ {
269
+ "epoch": 0.85,
270
+ "learning_rate": 7.533718235689727e-06,
271
+ "loss": 2.0259,
272
+ "step": 220
273
+ },
274
+ {
275
+ "epoch": 0.87,
276
+ "learning_rate": 5.751590460543944e-06,
277
+ "loss": 2.0717,
278
+ "step": 225
279
+ },
280
+ {
281
+ "epoch": 0.89,
282
+ "learning_rate": 4.2005634559834046e-06,
283
+ "loss": 2.0042,
284
+ "step": 230
285
+ },
286
+ {
287
+ "epoch": 0.91,
288
+ "learning_rate": 2.8863405252193965e-06,
289
+ "loss": 2.0301,
290
+ "step": 235
291
+ },
292
+ {
293
+ "epoch": 0.93,
294
+ "learning_rate": 1.8137542158731914e-06,
295
+ "loss": 2.0554,
296
+ "step": 240
297
+ },
298
+ {
299
+ "epoch": 0.95,
300
+ "learning_rate": 9.867485501471999e-07,
301
+ "loss": 2.0167,
302
+ "step": 245
303
+ },
304
+ {
305
+ "epoch": 0.97,
306
+ "learning_rate": 4.083645222054329e-07,
307
+ "loss": 2.0252,
308
+ "step": 250
309
+ },
310
+ {
311
+ "epoch": 0.98,
312
+ "learning_rate": 8.072891609113784e-08,
313
+ "loss": 2.0128,
314
+ "step": 255
315
+ },
316
+ {
317
+ "epoch": 1.0,
318
+ "learning_rate": 5.046485299251069e-09,
319
+ "loss": 2.1545,
320
+ "step": 260
321
+ },
322
+ {
323
+ "epoch": 1.02,
324
+ "learning_rate": 1.815955227603999e-07,
325
+ "loss": 2.1436,
326
+ "step": 265
327
+ },
328
+ {
329
+ "epoch": 1.04,
330
+ "learning_rate": 6.097268375260298e-07,
331
+ "loss": 2.055,
332
+ "step": 270
333
+ },
334
+ {
335
+ "epoch": 1.06,
336
+ "learning_rate": 1.2878661419176275e-06,
337
+ "loss": 2.0446,
338
+ "step": 275
339
+ },
340
+ {
341
+ "epoch": 1.08,
342
+ "learning_rate": 2.2135198403619775e-06,
343
+ "loss": 2.0962,
344
+ "step": 280
345
+ },
346
+ {
347
+ "epoch": 1.1,
348
+ "learning_rate": 3.3832841986266328e-06,
349
+ "loss": 1.9858,
350
+ "step": 285
351
+ },
352
+ {
353
+ "epoch": 1.12,
354
+ "learning_rate": 4.7928578597388414e-06,
355
+ "loss": 2.0331,
356
+ "step": 290
357
+ },
358
+ {
359
+ "epoch": 1.14,
360
+ "learning_rate": 6.437057660565834e-06,
361
+ "loss": 2.0861,
362
+ "step": 295
363
+ },
364
+ {
365
+ "epoch": 1.16,
366
+ "learning_rate": 8.309837690896773e-06,
367
+ "loss": 2.0373,
368
+ "step": 300
369
+ },
370
+ {
371
+ "epoch": 1.18,
372
+ "learning_rate": 1.0404311524944352e-05,
373
+ "loss": 2.0409,
374
+ "step": 305
375
+ },
376
+ {
377
+ "epoch": 1.2,
378
+ "learning_rate": 1.271277754351776e-05,
379
+ "loss": 2.1014,
380
+ "step": 310
381
+ },
382
+ {
383
+ "epoch": 1.22,
384
+ "learning_rate": 1.5226747253755048e-05,
385
+ "loss": 2.0414,
386
+ "step": 315
387
+ },
388
+ {
389
+ "epoch": 1.24,
390
+ "learning_rate": 1.79369765022795e-05,
391
+ "loss": 2.1258,
392
+ "step": 320
393
+ },
394
+ {
395
+ "epoch": 1.25,
396
+ "learning_rate": 2.083349946700612e-05,
397
+ "loss": 2.0382,
398
+ "step": 325
399
+ },
400
+ {
401
+ "epoch": 1.27,
402
+ "learning_rate": 2.3905665302606086e-05,
403
+ "loss": 1.9499,
404
+ "step": 330
405
+ },
406
+ {
407
+ "epoch": 1.29,
408
+ "learning_rate": 2.7142177304879985e-05,
409
+ "loss": 2.1053,
410
+ "step": 335
411
+ },
412
+ {
413
+ "epoch": 1.31,
414
+ "learning_rate": 3.05311344500276e-05,
415
+ "loss": 2.0373,
416
+ "step": 340
417
+ },
418
+ {
419
+ "epoch": 1.33,
420
+ "learning_rate": 3.4060075156069894e-05,
421
+ "loss": 1.9227,
422
+ "step": 345
423
+ },
424
+ {
425
+ "epoch": 1.35,
426
+ "learning_rate": 3.771602310550759e-05,
427
+ "loss": 2.0504,
428
+ "step": 350
429
+ },
430
+ {
431
+ "epoch": 1.37,
432
+ "learning_rate": 4.148553496072039e-05,
433
+ "loss": 2.0583,
434
+ "step": 355
435
+ },
436
+ {
437
+ "epoch": 1.39,
438
+ "learning_rate": 4.5354749796652995e-05,
439
+ "loss": 2.0166,
440
+ "step": 360
441
+ },
442
+ {
443
+ "epoch": 1.41,
444
+ "learning_rate": 4.930944006901758e-05,
445
+ "loss": 2.0524,
446
+ "step": 365
447
+ },
448
+ {
449
+ "epoch": 1.43,
450
+ "learning_rate": 5.333506393059682e-05,
451
+ "loss": 2.0846,
452
+ "step": 370
453
+ },
454
+ {
455
+ "epoch": 1.45,
456
+ "learning_rate": 5.741681870327513e-05,
457
+ "loss": 2.0408,
458
+ "step": 375
459
+ },
460
+ {
461
+ "epoch": 1.47,
462
+ "learning_rate": 6.153969530917408e-05,
463
+ "loss": 1.9538,
464
+ "step": 380
465
+ },
466
+ {
467
+ "epoch": 1.49,
468
+ "learning_rate": 6.56885334607442e-05,
469
+ "loss": 2.1089,
470
+ "step": 385
471
+ },
472
+ {
473
+ "epoch": 1.51,
474
+ "learning_rate": 6.984807740687121e-05,
475
+ "loss": 2.0182,
476
+ "step": 390
477
+ },
478
+ {
479
+ "epoch": 1.53,
480
+ "learning_rate": 7.400303203001308e-05,
481
+ "loss": 1.9797,
482
+ "step": 395
483
+ },
484
+ {
485
+ "epoch": 1.54,
486
+ "learning_rate": 7.813811908809188e-05,
487
+ "loss": 2.0459,
488
+ "step": 400
489
+ },
490
+ {
491
+ "epoch": 1.56,
492
+ "learning_rate": 8.22381333943327e-05,
493
+ "loss": 2.0528,
494
+ "step": 405
495
+ },
496
+ {
497
+ "epoch": 1.58,
498
+ "learning_rate": 8.628799872846947e-05,
499
+ "loss": 2.0164,
500
+ "step": 410
501
+ },
502
+ {
503
+ "epoch": 1.6,
504
+ "learning_rate": 9.027282327372695e-05,
505
+ "loss": 1.973,
506
+ "step": 415
507
+ },
508
+ {
509
+ "epoch": 1.62,
510
+ "learning_rate": 9.417795437572915e-05,
511
+ "loss": 2.1012,
512
+ "step": 420
513
+ },
514
+ {
515
+ "epoch": 1.64,
516
+ "learning_rate": 9.798903242198118e-05,
517
+ "loss": 1.9342,
518
+ "step": 425
519
+ },
520
+ {
521
+ "epoch": 1.66,
522
+ "learning_rate": 0.00010169204364380207,
523
+ "loss": 1.9812,
524
+ "step": 430
525
+ },
526
+ {
527
+ "epoch": 1.68,
528
+ "learning_rate": 0.0001052733716465509,
529
+ "loss": 2.1037,
530
+ "step": 435
531
+ },
532
+ {
533
+ "epoch": 1.7,
534
+ "learning_rate": 0.0001087198474786628,
535
+ "loss": 2.0833,
536
+ "step": 440
537
+ },
538
+ {
539
+ "epoch": 1.72,
540
+ "learning_rate": 0.00011201879805538593,
541
+ "loss": 2.088,
542
+ "step": 445
543
+ },
544
+ {
545
+ "epoch": 1.74,
546
+ "learning_rate": 0.00011515809275916009,
547
+ "loss": 2.0211,
548
+ "step": 450
549
+ },
550
+ {
551
+ "epoch": 1.76,
552
+ "learning_rate": 0.00011812618804528036,
553
+ "loss": 1.9749,
554
+ "step": 455
555
+ },
556
+ {
557
+ "epoch": 1.78,
558
+ "learning_rate": 0.00012091216988882859,
559
+ "loss": 2.1269,
560
+ "step": 460
561
+ },
562
+ {
563
+ "epoch": 1.8,
564
+ "learning_rate": 0.0001235057939167871,
565
+ "loss": 2.1407,
566
+ "step": 465
567
+ },
568
+ {
569
+ "epoch": 1.81,
570
+ "learning_rate": 0.0001258975230777674,
571
+ "loss": 1.9911,
572
+ "step": 470
573
+ },
574
+ {
575
+ "epoch": 1.83,
576
+ "learning_rate": 0.00012807856271083559,
577
+ "loss": 1.8852,
578
+ "step": 475
579
+ },
580
+ {
581
+ "epoch": 1.85,
582
+ "learning_rate": 0.00013004089288448385,
583
+ "loss": 2.084,
584
+ "step": 480
585
+ },
586
+ {
587
+ "epoch": 1.87,
588
+ "learning_rate": 0.00013177729788683344,
589
+ "loss": 2.0223,
590
+ "step": 485
591
+ },
592
+ {
593
+ "epoch": 1.89,
594
+ "learning_rate": 0.00013328139275863037,
595
+ "loss": 1.8615,
596
+ "step": 490
597
+ },
598
+ {
599
+ "epoch": 1.91,
600
+ "learning_rate": 0.00013454764677146868,
601
+ "loss": 1.9669,
602
+ "step": 495
603
+ },
604
+ {
605
+ "epoch": 1.93,
606
+ "learning_rate": 0.00013557140376490993,
607
+ "loss": 1.997,
608
+ "step": 500
609
+ },
610
+ {
611
+ "epoch": 1.95,
612
+ "learning_rate": 0.00013634889926771582,
613
+ "loss": 2.0132,
614
+ "step": 505
615
+ },
616
+ {
617
+ "epoch": 1.97,
618
+ "learning_rate": 0.00013687727434023874,
619
+ "loss": 2.1001,
620
+ "step": 510
621
+ },
622
+ {
623
+ "epoch": 1.99,
624
+ "learning_rate": 0.00013715458608706872,
625
+ "loss": 1.9605,
626
+ "step": 515
627
+ },
628
+ {
629
+ "epoch": 2.01,
630
+ "learning_rate": 0.00013717981480128154,
631
+ "loss": 1.9686,
632
+ "step": 520
633
+ },
634
+ {
635
+ "epoch": 2.03,
636
+ "learning_rate": 0.0001369528677140173,
637
+ "loss": 1.9054,
638
+ "step": 525
639
+ },
640
+ {
641
+ "epoch": 2.05,
642
+ "learning_rate": 0.00013647457933560234,
643
+ "loss": 1.9621,
644
+ "step": 530
645
+ },
646
+ {
647
+ "epoch": 2.07,
648
+ "learning_rate": 0.00013574670838695924,
649
+ "loss": 1.9001,
650
+ "step": 535
651
+ },
652
+ {
653
+ "epoch": 2.08,
654
+ "learning_rate": 0.00013477193133258972,
655
+ "loss": 1.9553,
656
+ "step": 540
657
+ },
658
+ {
659
+ "epoch": 2.1,
660
+ "learning_rate": 0.0001335538325389091,
661
+ "loss": 1.9963,
662
+ "step": 545
663
+ },
664
+ {
665
+ "epoch": 2.12,
666
+ "learning_rate": 0.00013209689109412243,
667
+ "loss": 1.9554,
668
+ "step": 550
669
+ },
670
+ {
671
+ "epoch": 2.14,
672
+ "learning_rate": 0.00013040646433810595,
673
+ "loss": 2.0022,
674
+ "step": 555
675
+ },
676
+ {
677
+ "epoch": 2.16,
678
+ "learning_rate": 0.00012848876816285752,
679
+ "loss": 1.8702,
680
+ "step": 560
681
+ },
682
+ {
683
+ "epoch": 2.18,
684
+ "learning_rate": 0.00012635085415595263,
685
+ "loss": 1.9121,
686
+ "step": 565
687
+ },
688
+ {
689
+ "epoch": 2.2,
690
+ "learning_rate": 0.00012400058367105258,
691
+ "loss": 2.0007,
692
+ "step": 570
693
+ },
694
+ {
695
+ "epoch": 2.22,
696
+ "learning_rate": 0.0001214465989208104,
697
+ "loss": 2.0254,
698
+ "step": 575
699
+ },
700
+ {
701
+ "epoch": 2.24,
702
+ "learning_rate": 0.00011869829119846914,
703
+ "loss": 1.9571,
704
+ "step": 580
705
+ },
706
+ {
707
+ "epoch": 2.26,
708
+ "learning_rate": 0.00011576576634500534,
709
+ "loss": 1.9448,
710
+ "step": 585
711
+ },
712
+ {
713
+ "epoch": 2.28,
714
+ "learning_rate": 0.00011265980758879933,
715
+ "loss": 2.0852,
716
+ "step": 590
717
+ },
718
+ {
719
+ "epoch": 2.3,
720
+ "learning_rate": 0.00010939183589447406,
721
+ "loss": 1.9757,
722
+ "step": 595
723
+ },
724
+ {
725
+ "epoch": 2.32,
726
+ "learning_rate": 0.00010597386796670587,
727
+ "loss": 2.0051,
728
+ "step": 600
729
+ },
730
+ {
731
+ "epoch": 2.34,
732
+ "learning_rate": 0.0001024184720634304,
733
+ "loss": 2.015,
734
+ "step": 605
735
+ },
736
+ {
737
+ "epoch": 2.36,
738
+ "learning_rate": 9.873872178092492e-05,
739
+ "loss": 1.9478,
740
+ "step": 610
741
+ },
742
+ {
743
+ "epoch": 2.37,
744
+ "learning_rate": 9.494814798070336e-05,
745
+ "loss": 1.8666,
746
+ "step": 615
747
+ },
748
+ {
749
+ "epoch": 2.39,
750
+ "learning_rate": 9.106068903499513e-05,
751
+ "loss": 1.9268,
752
+ "step": 620
753
+ },
754
+ {
755
+ "epoch": 2.41,
756
+ "learning_rate": 8.709063957376078e-05,
757
+ "loss": 1.8729,
758
+ "step": 625
759
+ },
760
+ {
761
+ "epoch": 2.43,
762
+ "learning_rate": 8.305259792170682e-05,
763
+ "loss": 1.9046,
764
+ "step": 630
765
+ },
766
+ {
767
+ "epoch": 2.45,
768
+ "learning_rate": 7.89614124185811e-05,
769
+ "loss": 1.8255,
770
+ "step": 635
771
+ },
772
+ {
773
+ "epoch": 2.47,
774
+ "learning_rate": 7.48321268201335e-05,
775
+ "loss": 1.9063,
776
+ "step": 640
777
+ },
778
+ {
779
+ "epoch": 2.49,
780
+ "learning_rate": 7.06799249805101e-05,
781
+ "loss": 1.9487,
782
+ "step": 645
783
+ },
784
+ {
785
+ "epoch": 2.51,
786
+ "learning_rate": 6.652007501948994e-05,
787
+ "loss": 1.9612,
788
+ "step": 650
789
+ },
790
+ {
791
+ "epoch": 2.53,
792
+ "learning_rate": 6.236787317986654e-05,
793
+ "loss": 1.829,
794
+ "step": 655
795
+ },
796
+ {
797
+ "epoch": 2.55,
798
+ "learning_rate": 5.8238587581418946e-05,
799
+ "loss": 1.8833,
800
+ "step": 660
801
+ },
802
+ {
803
+ "epoch": 2.57,
804
+ "learning_rate": 5.414740207829316e-05,
805
+ "loss": 1.9743,
806
+ "step": 665
807
+ },
808
+ {
809
+ "epoch": 2.59,
810
+ "learning_rate": 5.010936042623931e-05,
811
+ "loss": 1.8592,
812
+ "step": 670
813
+ },
814
+ {
815
+ "epoch": 2.61,
816
+ "learning_rate": 4.6139310965004967e-05,
817
+ "loss": 1.8143,
818
+ "step": 675
819
+ },
820
+ {
821
+ "epoch": 2.63,
822
+ "learning_rate": 4.225185201929667e-05,
823
+ "loss": 1.9597,
824
+ "step": 680
825
+ },
826
+ {
827
+ "epoch": 2.64,
828
+ "learning_rate": 3.8461278219075114e-05,
829
+ "loss": 1.9673,
830
+ "step": 685
831
+ },
832
+ {
833
+ "epoch": 2.66,
834
+ "learning_rate": 3.478152793656968e-05,
835
+ "loss": 1.9703,
836
+ "step": 690
837
+ },
838
+ {
839
+ "epoch": 2.68,
840
+ "learning_rate": 3.1226132033294165e-05,
841
+ "loss": 1.8479,
842
+ "step": 695
843
+ },
844
+ {
845
+ "epoch": 2.7,
846
+ "learning_rate": 2.7808164105525978e-05,
847
+ "loss": 1.8916,
848
+ "step": 700
849
+ },
850
+ {
851
+ "epoch": 2.72,
852
+ "learning_rate": 2.454019241120065e-05,
853
+ "loss": 1.9233,
854
+ "step": 705
855
+ },
856
+ {
857
+ "epoch": 2.74,
858
+ "learning_rate": 2.1434233654994646e-05,
859
+ "loss": 1.8525,
860
+ "step": 710
861
+ },
862
+ {
863
+ "epoch": 2.76,
864
+ "learning_rate": 1.850170880153093e-05,
865
+ "loss": 1.9783,
866
+ "step": 715
867
+ },
868
+ {
869
+ "epoch": 2.78,
870
+ "learning_rate": 1.5753401079189615e-05,
871
+ "loss": 1.8898,
872
+ "step": 720
873
+ },
874
+ {
875
+ "epoch": 2.8,
876
+ "learning_rate": 1.3199416328947464e-05,
877
+ "loss": 1.9018,
878
+ "step": 725
879
+ },
880
+ {
881
+ "epoch": 2.82,
882
+ "learning_rate": 1.0849145844047363e-05,
883
+ "loss": 1.9754,
884
+ "step": 730
885
+ },
886
+ {
887
+ "epoch": 2.84,
888
+ "learning_rate": 8.711231837142545e-06,
889
+ "loss": 1.9263,
890
+ "step": 735
891
+ },
892
+ {
893
+ "epoch": 2.86,
894
+ "learning_rate": 6.793535661894062e-06,
895
+ "loss": 1.836,
896
+ "step": 740
897
+ },
898
+ {
899
+ "epoch": 2.88,
900
+ "learning_rate": 5.103108905877591e-06,
901
+ "loss": 1.9056,
902
+ "step": 745
903
+ },
904
+ {
905
+ "epoch": 2.9,
906
+ "learning_rate": 3.6461674610908713e-06,
907
+ "loss": 1.9471,
908
+ "step": 750
909
+ },
910
+ {
911
+ "epoch": 2.92,
912
+ "learning_rate": 2.4280686674102973e-06,
913
+ "loss": 1.7961,
914
+ "step": 755
915
+ },
916
+ {
917
+ "epoch": 2.93,
918
+ "learning_rate": 1.453291613040777e-06,
919
+ "loss": 1.9143,
920
+ "step": 760
921
+ },
922
+ {
923
+ "epoch": 2.95,
924
+ "learning_rate": 7.254206643976737e-07,
925
+ "loss": 1.9633,
926
+ "step": 765
927
+ },
928
+ {
929
+ "epoch": 2.97,
930
+ "learning_rate": 2.4713228598268823e-07,
931
+ "loss": 1.9096,
932
+ "step": 770
933
+ },
934
+ {
935
+ "epoch": 2.99,
936
+ "learning_rate": 2.018519871846962e-08,
937
+ "loss": 1.7972,
938
+ "step": 775
939
+ }
940
+ ],
941
+ "max_steps": 777,
942
+ "num_train_epochs": 3,
943
+ "total_flos": 811703697408000.0,
944
+ "trial_name": null,
945
+ "trial_params": null
946
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:100c85c50ebb4281d2e3e83d1d7e4182a972554b3437d8264b44adb0e8333b79
3
+ size 2671
vocab.json ADDED
The diff for this file is too large to render. See raw diff