chansung commited on
Commit
9f00398
1 Parent(s): f8a064f

Model save

Browse files
README.md ADDED
@@ -0,0 +1,78 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: gemma
3
+ library_name: peft
4
+ tags:
5
+ - trl
6
+ - sft
7
+ - generated_from_trainer
8
+ base_model: google/gemma-7b
9
+ datasets:
10
+ - generator
11
+ model-index:
12
+ - name: gemma7b-summarize-gpt4o-2k
13
+ results: []
14
+ ---
15
+
16
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
17
+ should probably proofread and complete it, then remove this comment. -->
18
+
19
+ # gemma7b-summarize-gpt4o-2k
20
+
21
+ This model is a fine-tuned version of [google/gemma-7b](https://huggingface.co/google/gemma-7b) on the generator dataset.
22
+ It achieves the following results on the evaluation set:
23
+ - Loss: 3.2983
24
+
25
+ ## Model description
26
+
27
+ More information needed
28
+
29
+ ## Intended uses & limitations
30
+
31
+ More information needed
32
+
33
+ ## Training and evaluation data
34
+
35
+ More information needed
36
+
37
+ ## Training procedure
38
+
39
+ ### Training hyperparameters
40
+
41
+ The following hyperparameters were used during training:
42
+ - learning_rate: 0.0002
43
+ - train_batch_size: 4
44
+ - eval_batch_size: 4
45
+ - seed: 42
46
+ - distributed_type: multi-GPU
47
+ - num_devices: 2
48
+ - gradient_accumulation_steps: 2
49
+ - total_train_batch_size: 16
50
+ - total_eval_batch_size: 8
51
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
52
+ - lr_scheduler_type: cosine
53
+ - lr_scheduler_warmup_ratio: 0.1
54
+ - num_epochs: 10
55
+
56
+ ### Training results
57
+
58
+ | Training Loss | Epoch | Step | Validation Loss |
59
+ |:-------------:|:-----:|:----:|:---------------:|
60
+ | 23.4809 | 1.0 | 13 | 8.2506 |
61
+ | 7.0034 | 2.0 | 26 | 3.5645 |
62
+ | 1.9278 | 3.0 | 39 | 2.7831 |
63
+ | 1.3284 | 4.0 | 52 | 2.7295 |
64
+ | 1.046 | 5.0 | 65 | 2.7647 |
65
+ | 0.8611 | 6.0 | 78 | 2.8629 |
66
+ | 0.7242 | 7.0 | 91 | 3.0405 |
67
+ | 0.6192 | 8.0 | 104 | 3.1549 |
68
+ | 0.5312 | 9.0 | 117 | 3.3041 |
69
+ | 0.5202 | 10.0 | 130 | 3.2983 |
70
+
71
+
72
+ ### Framework versions
73
+
74
+ - PEFT 0.10.0
75
+ - Transformers 4.40.0
76
+ - Pytorch 2.2.1+cu121
77
+ - Datasets 2.18.0
78
+ - Tokenizers 0.19.1
adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1645caf0248c521fa7b2eade77d65029d7571b0d0b7c7a6caac0c768bc42b880
3
  size 50056096
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d553ae6702fe97cb32e10a3ba799b7a9b102497c3e9a1c2320c4275af05e6c45
3
  size 50056096
all_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 10.0,
3
+ "total_flos": 9.937675599478784e+16,
4
+ "train_loss": 4.9734316147290745,
5
+ "train_runtime": 870.6795,
6
+ "train_samples": 1874,
7
+ "train_samples_per_second": 2.366,
8
+ "train_steps_per_second": 0.149
9
+ }
runs/Jun08_13-06-54_user-WS-C621E-SAGE-Series/events.out.tfevents.1717819654.user-WS-C621E-SAGE-Series.8289.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:73b02aab48f359567f87907eb197b7f5faa185df256365d86f6aecea2984c11a
3
- size 11637
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e7d7053a3eeba6f9384755ab456b899b26bce72966f228d4e12bd97ad43b24d7
3
+ size 14040
train_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 10.0,
3
+ "total_flos": 9.937675599478784e+16,
4
+ "train_loss": 4.9734316147290745,
5
+ "train_runtime": 870.6795,
6
+ "train_samples": 1874,
7
+ "train_samples_per_second": 2.366,
8
+ "train_steps_per_second": 0.149
9
+ }
trainer_state.json ADDED
@@ -0,0 +1,299 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 10.0,
5
+ "eval_steps": 500,
6
+ "global_step": 130,
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.07692307692307693,
13
+ "grad_norm": 480.0,
14
+ "learning_rate": 1.5384615384615387e-05,
15
+ "loss": 49.667,
16
+ "step": 1
17
+ },
18
+ {
19
+ "epoch": 0.38461538461538464,
20
+ "grad_norm": 260.0,
21
+ "learning_rate": 7.692307692307693e-05,
22
+ "loss": 40.4711,
23
+ "step": 5
24
+ },
25
+ {
26
+ "epoch": 0.7692307692307693,
27
+ "grad_norm": 26.75,
28
+ "learning_rate": 0.00015384615384615385,
29
+ "loss": 23.4809,
30
+ "step": 10
31
+ },
32
+ {
33
+ "epoch": 1.0,
34
+ "eval_loss": 8.250589370727539,
35
+ "eval_runtime": 2.0471,
36
+ "eval_samples_per_second": 4.885,
37
+ "eval_steps_per_second": 0.977,
38
+ "step": 13
39
+ },
40
+ {
41
+ "epoch": 1.1538461538461537,
42
+ "grad_norm": 15.25,
43
+ "learning_rate": 0.00019985583705641418,
44
+ "loss": 19.4813,
45
+ "step": 15
46
+ },
47
+ {
48
+ "epoch": 1.5384615384615383,
49
+ "grad_norm": 34.75,
50
+ "learning_rate": 0.00019823877374156647,
51
+ "loss": 16.6271,
52
+ "step": 20
53
+ },
54
+ {
55
+ "epoch": 1.9230769230769231,
56
+ "grad_norm": 21.875,
57
+ "learning_rate": 0.00019485364419471454,
58
+ "loss": 7.0034,
59
+ "step": 25
60
+ },
61
+ {
62
+ "epoch": 2.0,
63
+ "eval_loss": 3.5644798278808594,
64
+ "eval_runtime": 2.0643,
65
+ "eval_samples_per_second": 4.844,
66
+ "eval_steps_per_second": 0.969,
67
+ "step": 26
68
+ },
69
+ {
70
+ "epoch": 2.3076923076923075,
71
+ "grad_norm": 5.5625,
72
+ "learning_rate": 0.0001897613727639014,
73
+ "loss": 2.3031,
74
+ "step": 30
75
+ },
76
+ {
77
+ "epoch": 2.6923076923076925,
78
+ "grad_norm": 3.09375,
79
+ "learning_rate": 0.00018305360832480117,
80
+ "loss": 1.9278,
81
+ "step": 35
82
+ },
83
+ {
84
+ "epoch": 3.0,
85
+ "eval_loss": 2.7830893993377686,
86
+ "eval_runtime": 2.0487,
87
+ "eval_samples_per_second": 4.881,
88
+ "eval_steps_per_second": 0.976,
89
+ "step": 39
90
+ },
91
+ {
92
+ "epoch": 3.076923076923077,
93
+ "grad_norm": 1.8671875,
94
+ "learning_rate": 0.00017485107481711012,
95
+ "loss": 1.6366,
96
+ "step": 40
97
+ },
98
+ {
99
+ "epoch": 3.4615384615384617,
100
+ "grad_norm": 5.84375,
101
+ "learning_rate": 0.0001653013984983585,
102
+ "loss": 1.4332,
103
+ "step": 45
104
+ },
105
+ {
106
+ "epoch": 3.8461538461538463,
107
+ "grad_norm": 1.5703125,
108
+ "learning_rate": 0.00015457645101945046,
109
+ "loss": 1.3284,
110
+ "step": 50
111
+ },
112
+ {
113
+ "epoch": 4.0,
114
+ "eval_loss": 2.7295384407043457,
115
+ "eval_runtime": 2.0468,
116
+ "eval_samples_per_second": 4.886,
117
+ "eval_steps_per_second": 0.977,
118
+ "step": 52
119
+ },
120
+ {
121
+ "epoch": 4.230769230769231,
122
+ "grad_norm": 2.0625,
123
+ "learning_rate": 0.00014286925614030542,
124
+ "loss": 1.1714,
125
+ "step": 55
126
+ },
127
+ {
128
+ "epoch": 4.615384615384615,
129
+ "grad_norm": 1.7578125,
130
+ "learning_rate": 0.0001303905157574247,
131
+ "loss": 1.0989,
132
+ "step": 60
133
+ },
134
+ {
135
+ "epoch": 5.0,
136
+ "grad_norm": 2.921875,
137
+ "learning_rate": 0.00011736481776669306,
138
+ "loss": 1.046,
139
+ "step": 65
140
+ },
141
+ {
142
+ "epoch": 5.0,
143
+ "eval_loss": 2.7646915912628174,
144
+ "eval_runtime": 2.047,
145
+ "eval_samples_per_second": 4.885,
146
+ "eval_steps_per_second": 0.977,
147
+ "step": 65
148
+ },
149
+ {
150
+ "epoch": 5.384615384615385,
151
+ "grad_norm": 2.65625,
152
+ "learning_rate": 0.00010402659401094152,
153
+ "loss": 0.9141,
154
+ "step": 70
155
+ },
156
+ {
157
+ "epoch": 5.769230769230769,
158
+ "grad_norm": 1.7734375,
159
+ "learning_rate": 9.061590105968208e-05,
160
+ "loss": 0.8611,
161
+ "step": 75
162
+ },
163
+ {
164
+ "epoch": 6.0,
165
+ "eval_loss": 2.862928867340088,
166
+ "eval_runtime": 2.0464,
167
+ "eval_samples_per_second": 4.887,
168
+ "eval_steps_per_second": 0.977,
169
+ "step": 78
170
+ },
171
+ {
172
+ "epoch": 6.153846153846154,
173
+ "grad_norm": 11.625,
174
+ "learning_rate": 7.73740997570278e-05,
175
+ "loss": 0.7981,
176
+ "step": 80
177
+ },
178
+ {
179
+ "epoch": 6.538461538461538,
180
+ "grad_norm": 2.09375,
181
+ "learning_rate": 6.453951129574644e-05,
182
+ "loss": 0.7228,
183
+ "step": 85
184
+ },
185
+ {
186
+ "epoch": 6.923076923076923,
187
+ "grad_norm": 2.078125,
188
+ "learning_rate": 5.234312799786921e-05,
189
+ "loss": 0.7242,
190
+ "step": 90
191
+ },
192
+ {
193
+ "epoch": 7.0,
194
+ "eval_loss": 3.040492296218872,
195
+ "eval_runtime": 2.0484,
196
+ "eval_samples_per_second": 4.882,
197
+ "eval_steps_per_second": 0.976,
198
+ "step": 91
199
+ },
200
+ {
201
+ "epoch": 7.3076923076923075,
202
+ "grad_norm": 1.421875,
203
+ "learning_rate": 4.100445599768774e-05,
204
+ "loss": 0.6348,
205
+ "step": 95
206
+ },
207
+ {
208
+ "epoch": 7.6923076923076925,
209
+ "grad_norm": 2.25,
210
+ "learning_rate": 3.072756464904006e-05,
211
+ "loss": 0.6192,
212
+ "step": 100
213
+ },
214
+ {
215
+ "epoch": 8.0,
216
+ "eval_loss": 3.1549479961395264,
217
+ "eval_runtime": 2.0433,
218
+ "eval_samples_per_second": 4.894,
219
+ "eval_steps_per_second": 0.979,
220
+ "step": 104
221
+ },
222
+ {
223
+ "epoch": 8.076923076923077,
224
+ "grad_norm": 1.3359375,
225
+ "learning_rate": 2.1697413758237784e-05,
226
+ "loss": 0.5742,
227
+ "step": 105
228
+ },
229
+ {
230
+ "epoch": 8.461538461538462,
231
+ "grad_norm": 1.5703125,
232
+ "learning_rate": 1.4076524743778319e-05,
233
+ "loss": 0.5514,
234
+ "step": 110
235
+ },
236
+ {
237
+ "epoch": 8.846153846153847,
238
+ "grad_norm": 1.40625,
239
+ "learning_rate": 8.002055634117578e-06,
240
+ "loss": 0.5312,
241
+ "step": 115
242
+ },
243
+ {
244
+ "epoch": 9.0,
245
+ "eval_loss": 3.3040664196014404,
246
+ "eval_runtime": 2.0464,
247
+ "eval_samples_per_second": 4.887,
248
+ "eval_steps_per_second": 0.977,
249
+ "step": 117
250
+ },
251
+ {
252
+ "epoch": 9.23076923076923,
253
+ "grad_norm": 1.265625,
254
+ "learning_rate": 3.5833325466437694e-06,
255
+ "loss": 0.5132,
256
+ "step": 120
257
+ },
258
+ {
259
+ "epoch": 9.615384615384615,
260
+ "grad_norm": 1.3984375,
261
+ "learning_rate": 8.998820754091531e-07,
262
+ "loss": 0.4966,
263
+ "step": 125
264
+ },
265
+ {
266
+ "epoch": 10.0,
267
+ "grad_norm": 1.2109375,
268
+ "learning_rate": 0.0,
269
+ "loss": 0.5202,
270
+ "step": 130
271
+ },
272
+ {
273
+ "epoch": 10.0,
274
+ "eval_loss": 3.298276901245117,
275
+ "eval_runtime": 2.0482,
276
+ "eval_samples_per_second": 4.882,
277
+ "eval_steps_per_second": 0.976,
278
+ "step": 130
279
+ },
280
+ {
281
+ "epoch": 10.0,
282
+ "step": 130,
283
+ "total_flos": 9.937675599478784e+16,
284
+ "train_loss": 4.9734316147290745,
285
+ "train_runtime": 870.6795,
286
+ "train_samples_per_second": 2.366,
287
+ "train_steps_per_second": 0.149
288
+ }
289
+ ],
290
+ "logging_steps": 5,
291
+ "max_steps": 130,
292
+ "num_input_tokens_seen": 0,
293
+ "num_train_epochs": 10,
294
+ "save_steps": 100,
295
+ "total_flos": 9.937675599478784e+16,
296
+ "train_batch_size": 4,
297
+ "trial_name": null,
298
+ "trial_params": null
299
+ }