van-ng commited on
Commit
4ad8267
1 Parent(s): d714d24

Upload 11 files

Browse files
config.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "EleutherAI/pythia-160m",
3
+ "architectures": [
4
+ "GPTNeoXForCausalLM"
5
+ ],
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": 768,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 3072,
15
+ "layer_norm_eps": 1e-05,
16
+ "max_position_embeddings": 2048,
17
+ "model_type": "gpt_neox",
18
+ "num_attention_heads": 12,
19
+ "num_hidden_layers": 12,
20
+ "rope_scaling": null,
21
+ "rotary_emb_base": 10000,
22
+ "rotary_pct": 0.25,
23
+ "tie_word_embeddings": false,
24
+ "torch_dtype": "float32",
25
+ "transformers_version": "4.32.1",
26
+ "use_cache": true,
27
+ "use_parallel_residual": true,
28
+ "vocab_size": 50304
29
+ }
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.32.1"
6
+ }
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e6e0c60fe14626eb3e99d0eb051e6a491b5497d1d20440aac27e387035665873
3
+ size 1582330
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f2b34b1ab72b491bd6e6ccfe42d5822a7746c46f2918c27bac057138aa494982
3
+ size 649346058
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:10d8b236bdc23f0833f773106a29bfe4c02f59cb3956e26802cc1c0318ef7145
3
+ size 14244
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7c85a5188461ba83a37731f34904a2d05a31f89c98bafd672f1c994cdc2a6c7a
3
+ size 1064
special_tokens_map.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "<|endoftext|>",
3
+ "eos_token": "<|endoftext|>",
4
+ "pad_token": "<|endoftext|>",
5
+ "unk_token": "<|endoftext|>"
6
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "bos_token": "<|endoftext|>",
4
+ "clean_up_tokenization_spaces": true,
5
+ "eos_token": "<|endoftext|>",
6
+ "model_max_length": 1000000000000000019884624838656,
7
+ "tokenizer_class": "GPTNeoXTokenizer",
8
+ "unk_token": "<|endoftext|>"
9
+ }
trainer_state.json ADDED
@@ -0,0 +1,199 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.27965888381004333,
3
+ "best_model_checkpoint": "pythia-XYZCompany-1000-steps\\checkpoint-450",
4
+ "epoch": 2.848101265822785,
5
+ "eval_steps": 50,
6
+ "global_step": 450,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.16,
13
+ "learning_rate": 9.75975975975976e-06,
14
+ "loss": 1.497,
15
+ "step": 25
16
+ },
17
+ {
18
+ "epoch": 0.32,
19
+ "learning_rate": 9.50950950950951e-06,
20
+ "loss": 0.3452,
21
+ "step": 50
22
+ },
23
+ {
24
+ "epoch": 0.32,
25
+ "eval_loss": 0.32888519763946533,
26
+ "eval_runtime": 6.4693,
27
+ "eval_samples_per_second": 21.641,
28
+ "eval_steps_per_second": 5.41,
29
+ "step": 50
30
+ },
31
+ {
32
+ "epoch": 0.47,
33
+ "learning_rate": 9.25925925925926e-06,
34
+ "loss": 0.318,
35
+ "step": 75
36
+ },
37
+ {
38
+ "epoch": 0.63,
39
+ "learning_rate": 9.00900900900901e-06,
40
+ "loss": 0.3598,
41
+ "step": 100
42
+ },
43
+ {
44
+ "epoch": 0.63,
45
+ "eval_loss": 0.3072541356086731,
46
+ "eval_runtime": 6.5211,
47
+ "eval_samples_per_second": 21.469,
48
+ "eval_steps_per_second": 5.367,
49
+ "step": 100
50
+ },
51
+ {
52
+ "epoch": 0.79,
53
+ "learning_rate": 8.75875875875876e-06,
54
+ "loss": 0.2963,
55
+ "step": 125
56
+ },
57
+ {
58
+ "epoch": 0.95,
59
+ "learning_rate": 8.50850850850851e-06,
60
+ "loss": 0.3045,
61
+ "step": 150
62
+ },
63
+ {
64
+ "epoch": 0.95,
65
+ "eval_loss": 0.2949899435043335,
66
+ "eval_runtime": 6.7431,
67
+ "eval_samples_per_second": 20.762,
68
+ "eval_steps_per_second": 5.19,
69
+ "step": 150
70
+ },
71
+ {
72
+ "epoch": 1.11,
73
+ "learning_rate": 8.258258258258259e-06,
74
+ "loss": 0.2857,
75
+ "step": 175
76
+ },
77
+ {
78
+ "epoch": 1.27,
79
+ "learning_rate": 8.00800800800801e-06,
80
+ "loss": 0.252,
81
+ "step": 200
82
+ },
83
+ {
84
+ "epoch": 1.27,
85
+ "eval_loss": 0.2895931303501129,
86
+ "eval_runtime": 6.5147,
87
+ "eval_samples_per_second": 21.49,
88
+ "eval_steps_per_second": 5.372,
89
+ "step": 200
90
+ },
91
+ {
92
+ "epoch": 1.42,
93
+ "learning_rate": 7.757757757757758e-06,
94
+ "loss": 0.2613,
95
+ "step": 225
96
+ },
97
+ {
98
+ "epoch": 1.58,
99
+ "learning_rate": 7.507507507507507e-06,
100
+ "loss": 0.2541,
101
+ "step": 250
102
+ },
103
+ {
104
+ "epoch": 1.58,
105
+ "eval_loss": 0.2829298973083496,
106
+ "eval_runtime": 6.7922,
107
+ "eval_samples_per_second": 20.612,
108
+ "eval_steps_per_second": 5.153,
109
+ "step": 250
110
+ },
111
+ {
112
+ "epoch": 1.74,
113
+ "learning_rate": 7.257257257257258e-06,
114
+ "loss": 0.2388,
115
+ "step": 275
116
+ },
117
+ {
118
+ "epoch": 1.9,
119
+ "learning_rate": 7.007007007007007e-06,
120
+ "loss": 0.2455,
121
+ "step": 300
122
+ },
123
+ {
124
+ "epoch": 1.9,
125
+ "eval_loss": 0.28180426359176636,
126
+ "eval_runtime": 6.7633,
127
+ "eval_samples_per_second": 20.7,
128
+ "eval_steps_per_second": 5.175,
129
+ "step": 300
130
+ },
131
+ {
132
+ "epoch": 2.06,
133
+ "learning_rate": 6.7567567567567575e-06,
134
+ "loss": 0.2336,
135
+ "step": 325
136
+ },
137
+ {
138
+ "epoch": 2.22,
139
+ "learning_rate": 6.506506506506507e-06,
140
+ "loss": 0.2067,
141
+ "step": 350
142
+ },
143
+ {
144
+ "epoch": 2.22,
145
+ "eval_loss": 0.28442302346229553,
146
+ "eval_runtime": 6.758,
147
+ "eval_samples_per_second": 20.716,
148
+ "eval_steps_per_second": 5.179,
149
+ "step": 350
150
+ },
151
+ {
152
+ "epoch": 2.37,
153
+ "learning_rate": 6.2562562562562565e-06,
154
+ "loss": 0.2049,
155
+ "step": 375
156
+ },
157
+ {
158
+ "epoch": 2.53,
159
+ "learning_rate": 6.006006006006007e-06,
160
+ "loss": 0.2169,
161
+ "step": 400
162
+ },
163
+ {
164
+ "epoch": 2.53,
165
+ "eval_loss": 0.28421658277511597,
166
+ "eval_runtime": 6.7687,
167
+ "eval_samples_per_second": 20.683,
168
+ "eval_steps_per_second": 5.171,
169
+ "step": 400
170
+ },
171
+ {
172
+ "epoch": 2.69,
173
+ "learning_rate": 5.755755755755756e-06,
174
+ "loss": 0.1923,
175
+ "step": 425
176
+ },
177
+ {
178
+ "epoch": 2.85,
179
+ "learning_rate": 5.505505505505506e-06,
180
+ "loss": 0.2112,
181
+ "step": 450
182
+ },
183
+ {
184
+ "epoch": 2.85,
185
+ "eval_loss": 0.27965888381004333,
186
+ "eval_runtime": 6.7745,
187
+ "eval_samples_per_second": 20.666,
188
+ "eval_steps_per_second": 5.166,
189
+ "step": 450
190
+ }
191
+ ],
192
+ "logging_steps": 25,
193
+ "max_steps": 1000,
194
+ "num_train_epochs": 7,
195
+ "save_steps": 50,
196
+ "total_flos": 1364866816278528.0,
197
+ "trial_name": null,
198
+ "trial_params": null
199
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eca7858e48b6a07b3ded2df389e4f1c434ee1d50ec0e9bfbf6eb82a942f5edc1
3
+ size 4472