penfever commited on
Commit
897a9f8
·
verified ·
1 Parent(s): f9c8442

End of training

Browse files
Files changed (5) hide show
  1. README.md +2 -1
  2. all_results.json +16 -0
  3. train_results.json +16 -0
  4. trainer_state.json +1774 -0
  5. training_loss.png +0 -0
README.md CHANGED
@@ -4,6 +4,7 @@ license: apache-2.0
4
  base_model: Qwen/Qwen3-8B
5
  tags:
6
  - llama-factory
 
7
  - generated_from_trainer
8
  model-index:
9
  - name: nemo-prism-math-sandboxes-traces-terminus-2
@@ -15,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  # nemo-prism-math-sandboxes-traces-terminus-2
17
 
18
- This model is a fine-tuned version of [Qwen/Qwen3-8B](https://huggingface.co/Qwen/Qwen3-8B) on an unknown dataset.
19
 
20
  ## Model description
21
 
 
4
  base_model: Qwen/Qwen3-8B
5
  tags:
6
  - llama-factory
7
+ - full
8
  - generated_from_trainer
9
  model-index:
10
  - name: nemo-prism-math-sandboxes-traces-terminus-2
 
16
 
17
  # nemo-prism-math-sandboxes-traces-terminus-2
18
 
19
+ This model is a fine-tuned version of [Qwen/Qwen3-8B](https://huggingface.co/Qwen/Qwen3-8B) on the mlfoundations-dev/nemo-prism-math-sandboxes-traces-terminus-2 dataset.
20
 
21
  ## Model description
22
 
all_results.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "achieved_tflops_per_gpu": 2.3865554350500977,
3
+ "achieved_tflops_per_gpu_theoretical": 338.7904562823168,
4
+ "epoch": 5.0,
5
+ "loss_nan_ranks": 0,
6
+ "loss_rank_avg": 0.391443133354187,
7
+ "mfu_percent": 0.1686611614876394,
8
+ "mfu_percent_theoretical": 23.94278842984571,
9
+ "total_flos": 2.9988290575938355e+17,
10
+ "train_loss": 0.4420624729934012,
11
+ "train_runtime": 7853.4449,
12
+ "train_samples_per_second": 6.363,
13
+ "train_steps_per_second": 0.1,
14
+ "valid_targets_mean": 719.2,
15
+ "valid_targets_min": 319
16
+ }
train_results.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "achieved_tflops_per_gpu": 2.3865554350500977,
3
+ "achieved_tflops_per_gpu_theoretical": 338.7904562823168,
4
+ "epoch": 5.0,
5
+ "loss_nan_ranks": 0,
6
+ "loss_rank_avg": 0.391443133354187,
7
+ "mfu_percent": 0.1686611614876394,
8
+ "mfu_percent_theoretical": 23.94278842984571,
9
+ "total_flos": 2.9988290575938355e+17,
10
+ "train_loss": 0.4420624729934012,
11
+ "train_runtime": 7853.4449,
12
+ "train_samples_per_second": 6.363,
13
+ "train_steps_per_second": 0.1,
14
+ "valid_targets_mean": 719.2,
15
+ "valid_targets_min": 319
16
+ }
trainer_state.json ADDED
@@ -0,0 +1,1774 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 5.0,
6
+ "eval_steps": 500,
7
+ "global_step": 785,
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.032,
14
+ "grad_norm": 13.266562800371773,
15
+ "learning_rate": 2.0253164556962026e-06,
16
+ "loss": 0.9687,
17
+ "loss_nan_ranks": 0,
18
+ "loss_rank_avg": 0.29333972930908203,
19
+ "step": 5,
20
+ "valid_targets_mean": 997.2,
21
+ "valid_targets_min": 343
22
+ },
23
+ {
24
+ "epoch": 0.064,
25
+ "grad_norm": 8.489508619950364,
26
+ "learning_rate": 4.556962025316456e-06,
27
+ "loss": 0.9291,
28
+ "loss_nan_ranks": 0,
29
+ "loss_rank_avg": 0.19505365192890167,
30
+ "step": 10,
31
+ "valid_targets_mean": 701.6,
32
+ "valid_targets_min": 428
33
+ },
34
+ {
35
+ "epoch": 0.096,
36
+ "grad_norm": 3.7845870692684294,
37
+ "learning_rate": 7.08860759493671e-06,
38
+ "loss": 0.8215,
39
+ "loss_nan_ranks": 0,
40
+ "loss_rank_avg": 0.2177591770887375,
41
+ "step": 15,
42
+ "valid_targets_mean": 873.1,
43
+ "valid_targets_min": 416
44
+ },
45
+ {
46
+ "epoch": 0.128,
47
+ "grad_norm": 1.9414984171383554,
48
+ "learning_rate": 9.620253164556963e-06,
49
+ "loss": 0.7485,
50
+ "loss_nan_ranks": 0,
51
+ "loss_rank_avg": 0.15389665961265564,
52
+ "step": 20,
53
+ "valid_targets_mean": 859.7,
54
+ "valid_targets_min": 408
55
+ },
56
+ {
57
+ "epoch": 0.16,
58
+ "grad_norm": 1.4709191649573787,
59
+ "learning_rate": 1.2151898734177216e-05,
60
+ "loss": 0.7104,
61
+ "loss_nan_ranks": 0,
62
+ "loss_rank_avg": 0.17423586547374725,
63
+ "step": 25,
64
+ "valid_targets_mean": 877.4,
65
+ "valid_targets_min": 370
66
+ },
67
+ {
68
+ "epoch": 0.192,
69
+ "grad_norm": 1.1650092738181974,
70
+ "learning_rate": 1.468354430379747e-05,
71
+ "loss": 0.6583,
72
+ "loss_nan_ranks": 0,
73
+ "loss_rank_avg": 0.13358977437019348,
74
+ "step": 30,
75
+ "valid_targets_mean": 705.3,
76
+ "valid_targets_min": 362
77
+ },
78
+ {
79
+ "epoch": 0.224,
80
+ "grad_norm": 0.9252713386370359,
81
+ "learning_rate": 1.7215189873417723e-05,
82
+ "loss": 0.5997,
83
+ "loss_nan_ranks": 0,
84
+ "loss_rank_avg": 0.12491467595100403,
85
+ "step": 35,
86
+ "valid_targets_mean": 662.6,
87
+ "valid_targets_min": 366
88
+ },
89
+ {
90
+ "epoch": 0.256,
91
+ "grad_norm": 0.7978370195902237,
92
+ "learning_rate": 1.974683544303798e-05,
93
+ "loss": 0.5912,
94
+ "loss_nan_ranks": 0,
95
+ "loss_rank_avg": 0.12133462727069855,
96
+ "step": 40,
97
+ "valid_targets_mean": 708.8,
98
+ "valid_targets_min": 304
99
+ },
100
+ {
101
+ "epoch": 0.288,
102
+ "grad_norm": 0.6443855909829759,
103
+ "learning_rate": 2.2278481012658228e-05,
104
+ "loss": 0.5653,
105
+ "loss_nan_ranks": 0,
106
+ "loss_rank_avg": 0.1608869433403015,
107
+ "step": 45,
108
+ "valid_targets_mean": 1044.8,
109
+ "valid_targets_min": 363
110
+ },
111
+ {
112
+ "epoch": 0.32,
113
+ "grad_norm": 0.6275594218254698,
114
+ "learning_rate": 2.481012658227848e-05,
115
+ "loss": 0.5612,
116
+ "loss_nan_ranks": 0,
117
+ "loss_rank_avg": 0.12222383916378021,
118
+ "step": 50,
119
+ "valid_targets_mean": 880.1,
120
+ "valid_targets_min": 394
121
+ },
122
+ {
123
+ "epoch": 0.352,
124
+ "grad_norm": 0.5428436498161326,
125
+ "learning_rate": 2.7341772151898737e-05,
126
+ "loss": 0.5246,
127
+ "loss_nan_ranks": 0,
128
+ "loss_rank_avg": 0.12391236424446106,
129
+ "step": 55,
130
+ "valid_targets_mean": 865.2,
131
+ "valid_targets_min": 407
132
+ },
133
+ {
134
+ "epoch": 0.384,
135
+ "grad_norm": 0.6335065649840791,
136
+ "learning_rate": 2.987341772151899e-05,
137
+ "loss": 0.5621,
138
+ "loss_nan_ranks": 0,
139
+ "loss_rank_avg": 0.13935324549674988,
140
+ "step": 60,
141
+ "valid_targets_mean": 877.2,
142
+ "valid_targets_min": 331
143
+ },
144
+ {
145
+ "epoch": 0.416,
146
+ "grad_norm": 0.7837301485260351,
147
+ "learning_rate": 3.240506329113924e-05,
148
+ "loss": 0.5275,
149
+ "loss_nan_ranks": 0,
150
+ "loss_rank_avg": 0.12021635472774506,
151
+ "step": 65,
152
+ "valid_targets_mean": 592.8,
153
+ "valid_targets_min": 373
154
+ },
155
+ {
156
+ "epoch": 0.448,
157
+ "grad_norm": 0.5784603563747543,
158
+ "learning_rate": 3.49367088607595e-05,
159
+ "loss": 0.5115,
160
+ "loss_nan_ranks": 0,
161
+ "loss_rank_avg": 0.09628315269947052,
162
+ "step": 70,
163
+ "valid_targets_mean": 794.9,
164
+ "valid_targets_min": 348
165
+ },
166
+ {
167
+ "epoch": 0.48,
168
+ "grad_norm": 0.5661489350396759,
169
+ "learning_rate": 3.746835443037975e-05,
170
+ "loss": 0.5105,
171
+ "loss_nan_ranks": 0,
172
+ "loss_rank_avg": 0.15161114931106567,
173
+ "step": 75,
174
+ "valid_targets_mean": 1041.4,
175
+ "valid_targets_min": 396
176
+ },
177
+ {
178
+ "epoch": 0.512,
179
+ "grad_norm": 0.6020958811116237,
180
+ "learning_rate": 4e-05,
181
+ "loss": 0.5101,
182
+ "loss_nan_ranks": 0,
183
+ "loss_rank_avg": 0.1513483226299286,
184
+ "step": 80,
185
+ "valid_targets_mean": 1158.1,
186
+ "valid_targets_min": 419
187
+ },
188
+ {
189
+ "epoch": 0.544,
190
+ "grad_norm": 0.7041879443596871,
191
+ "learning_rate": 3.999504991751045e-05,
192
+ "loss": 0.4906,
193
+ "loss_nan_ranks": 0,
194
+ "loss_rank_avg": 0.11582835763692856,
195
+ "step": 85,
196
+ "valid_targets_mean": 546.4,
197
+ "valid_targets_min": 348
198
+ },
199
+ {
200
+ "epoch": 0.576,
201
+ "grad_norm": 0.6350096833420561,
202
+ "learning_rate": 3.9980202120373464e-05,
203
+ "loss": 0.4942,
204
+ "loss_nan_ranks": 0,
205
+ "loss_rank_avg": 0.11067851632833481,
206
+ "step": 90,
207
+ "valid_targets_mean": 770.5,
208
+ "valid_targets_min": 365
209
+ },
210
+ {
211
+ "epoch": 0.608,
212
+ "grad_norm": 0.5823516591675933,
213
+ "learning_rate": 3.995546395837111e-05,
214
+ "loss": 0.4833,
215
+ "loss_nan_ranks": 0,
216
+ "loss_rank_avg": 0.11685052514076233,
217
+ "step": 95,
218
+ "valid_targets_mean": 844.8,
219
+ "valid_targets_min": 358
220
+ },
221
+ {
222
+ "epoch": 0.64,
223
+ "grad_norm": 0.6959370180161534,
224
+ "learning_rate": 3.992084767709763e-05,
225
+ "loss": 0.492,
226
+ "loss_nan_ranks": 0,
227
+ "loss_rank_avg": 0.11015971750020981,
228
+ "step": 100,
229
+ "valid_targets_mean": 649.9,
230
+ "valid_targets_min": 346
231
+ },
232
+ {
233
+ "epoch": 0.672,
234
+ "grad_norm": 0.7368399273439582,
235
+ "learning_rate": 3.987637041189781e-05,
236
+ "loss": 0.4814,
237
+ "loss_nan_ranks": 0,
238
+ "loss_rank_avg": 0.12616831064224243,
239
+ "step": 105,
240
+ "valid_targets_mean": 813.2,
241
+ "valid_targets_min": 369
242
+ },
243
+ {
244
+ "epoch": 0.704,
245
+ "grad_norm": 0.6272126363527385,
246
+ "learning_rate": 3.982205417938482e-05,
247
+ "loss": 0.4864,
248
+ "loss_nan_ranks": 0,
249
+ "loss_rank_avg": 0.09629425406455994,
250
+ "step": 110,
251
+ "valid_targets_mean": 738.4,
252
+ "valid_targets_min": 346
253
+ },
254
+ {
255
+ "epoch": 0.736,
256
+ "grad_norm": 0.6408625248348573,
257
+ "learning_rate": 3.975792586654179e-05,
258
+ "loss": 0.487,
259
+ "loss_nan_ranks": 0,
260
+ "loss_rank_avg": 0.10672964155673981,
261
+ "step": 115,
262
+ "valid_targets_mean": 582.9,
263
+ "valid_targets_min": 394
264
+ },
265
+ {
266
+ "epoch": 0.768,
267
+ "grad_norm": 0.6804203420373807,
268
+ "learning_rate": 3.968401721741259e-05,
269
+ "loss": 0.487,
270
+ "loss_nan_ranks": 0,
271
+ "loss_rank_avg": 0.11347218602895737,
272
+ "step": 120,
273
+ "valid_targets_mean": 730.2,
274
+ "valid_targets_min": 451
275
+ },
276
+ {
277
+ "epoch": 0.8,
278
+ "grad_norm": 0.5723844040912542,
279
+ "learning_rate": 3.960036481738819e-05,
280
+ "loss": 0.4942,
281
+ "loss_nan_ranks": 0,
282
+ "loss_rank_avg": 0.15321877598762512,
283
+ "step": 125,
284
+ "valid_targets_mean": 1487.5,
285
+ "valid_targets_min": 316
286
+ },
287
+ {
288
+ "epoch": 0.832,
289
+ "grad_norm": 0.6956004153841853,
290
+ "learning_rate": 3.950701007509667e-05,
291
+ "loss": 0.479,
292
+ "loss_nan_ranks": 0,
293
+ "loss_rank_avg": 0.11619652807712555,
294
+ "step": 130,
295
+ "valid_targets_mean": 745.4,
296
+ "valid_targets_min": 386
297
+ },
298
+ {
299
+ "epoch": 0.864,
300
+ "grad_norm": 0.6296253097471561,
301
+ "learning_rate": 3.940399920190552e-05,
302
+ "loss": 0.4654,
303
+ "loss_nan_ranks": 0,
304
+ "loss_rank_avg": 0.09865935146808624,
305
+ "step": 135,
306
+ "valid_targets_mean": 767.4,
307
+ "valid_targets_min": 397
308
+ },
309
+ {
310
+ "epoch": 0.896,
311
+ "grad_norm": 0.6474475974143318,
312
+ "learning_rate": 3.92913831890467e-05,
313
+ "loss": 0.494,
314
+ "loss_nan_ranks": 0,
315
+ "loss_rank_avg": 0.1201113611459732,
316
+ "step": 140,
317
+ "valid_targets_mean": 823.1,
318
+ "valid_targets_min": 334
319
+ },
320
+ {
321
+ "epoch": 0.928,
322
+ "grad_norm": 0.5614028338623152,
323
+ "learning_rate": 3.916921778237556e-05,
324
+ "loss": 0.4736,
325
+ "loss_nan_ranks": 0,
326
+ "loss_rank_avg": 0.11008624732494354,
327
+ "step": 145,
328
+ "valid_targets_mean": 724.4,
329
+ "valid_targets_min": 310
330
+ },
331
+ {
332
+ "epoch": 0.96,
333
+ "grad_norm": 0.6417707676376314,
334
+ "learning_rate": 3.903756345477612e-05,
335
+ "loss": 0.4894,
336
+ "loss_nan_ranks": 0,
337
+ "loss_rank_avg": 0.13074582815170288,
338
+ "step": 150,
339
+ "valid_targets_mean": 799.7,
340
+ "valid_targets_min": 366
341
+ },
342
+ {
343
+ "epoch": 0.992,
344
+ "grad_norm": 0.5556077488737904,
345
+ "learning_rate": 3.889648537622657e-05,
346
+ "loss": 0.4647,
347
+ "loss_nan_ranks": 0,
348
+ "loss_rank_avg": 0.09520229697227478,
349
+ "step": 155,
350
+ "valid_targets_mean": 759.1,
351
+ "valid_targets_min": 379
352
+ },
353
+ {
354
+ "epoch": 1.0192,
355
+ "grad_norm": 0.518917132469663,
356
+ "learning_rate": 3.874605338153952e-05,
357
+ "loss": 0.4877,
358
+ "loss_nan_ranks": 0,
359
+ "loss_rank_avg": 0.1457328051328659,
360
+ "step": 160,
361
+ "valid_targets_mean": 1207.9,
362
+ "valid_targets_min": 401
363
+ },
364
+ {
365
+ "epoch": 1.0512,
366
+ "grad_norm": 0.6519204578424899,
367
+ "learning_rate": 3.8586341935793265e-05,
368
+ "loss": 0.4549,
369
+ "loss_nan_ranks": 0,
370
+ "loss_rank_avg": 0.09436912834644318,
371
+ "step": 165,
372
+ "valid_targets_mean": 591.2,
373
+ "valid_targets_min": 358
374
+ },
375
+ {
376
+ "epoch": 1.0832,
377
+ "grad_norm": 0.7191821046729394,
378
+ "learning_rate": 3.841743009747089e-05,
379
+ "loss": 0.4724,
380
+ "loss_nan_ranks": 0,
381
+ "loss_rank_avg": 0.09062425047159195,
382
+ "step": 170,
383
+ "valid_targets_mean": 553.2,
384
+ "valid_targets_min": 413
385
+ },
386
+ {
387
+ "epoch": 1.1152,
388
+ "grad_norm": 0.5941223668586252,
389
+ "learning_rate": 3.8239401479325714e-05,
390
+ "loss": 0.4416,
391
+ "loss_nan_ranks": 0,
392
+ "loss_rank_avg": 0.11501722037792206,
393
+ "step": 175,
394
+ "valid_targets_mean": 767.8,
395
+ "valid_targets_min": 385
396
+ },
397
+ {
398
+ "epoch": 1.1472,
399
+ "grad_norm": 0.5716063684906239,
400
+ "learning_rate": 3.8052344206992276e-05,
401
+ "loss": 0.4658,
402
+ "loss_nan_ranks": 0,
403
+ "loss_rank_avg": 0.15316523611545563,
404
+ "step": 180,
405
+ "valid_targets_mean": 1364.5,
406
+ "valid_targets_min": 520
407
+ },
408
+ {
409
+ "epoch": 1.1792,
410
+ "grad_norm": 0.5527348861402952,
411
+ "learning_rate": 3.7856350875363396e-05,
412
+ "loss": 0.4557,
413
+ "loss_nan_ranks": 0,
414
+ "loss_rank_avg": 0.1328214555978775,
415
+ "step": 185,
416
+ "valid_targets_mean": 979.0,
417
+ "valid_targets_min": 416
418
+ },
419
+ {
420
+ "epoch": 1.2112,
421
+ "grad_norm": 0.5274840156315275,
422
+ "learning_rate": 3.765151850275497e-05,
423
+ "loss": 0.4457,
424
+ "loss_nan_ranks": 0,
425
+ "loss_rank_avg": 0.09974588453769684,
426
+ "step": 190,
427
+ "valid_targets_mean": 1234.8,
428
+ "valid_targets_min": 294
429
+ },
430
+ {
431
+ "epoch": 1.2432,
432
+ "grad_norm": 0.5987542988521145,
433
+ "learning_rate": 3.7437948482881104e-05,
434
+ "loss": 0.4495,
435
+ "loss_nan_ranks": 0,
436
+ "loss_rank_avg": 0.1128249317407608,
437
+ "step": 195,
438
+ "valid_targets_mean": 824.9,
439
+ "valid_targets_min": 397
440
+ },
441
+ {
442
+ "epoch": 1.2752,
443
+ "grad_norm": 0.5732559336354344,
444
+ "learning_rate": 3.721574653466336e-05,
445
+ "loss": 0.4424,
446
+ "loss_nan_ranks": 0,
447
+ "loss_rank_avg": 0.13284024596214294,
448
+ "step": 200,
449
+ "valid_targets_mean": 1010.7,
450
+ "valid_targets_min": 367
451
+ },
452
+ {
453
+ "epoch": 1.3072,
454
+ "grad_norm": 0.6697760935036775,
455
+ "learning_rate": 3.698502264989903e-05,
456
+ "loss": 0.4334,
457
+ "loss_nan_ranks": 0,
458
+ "loss_rank_avg": 0.13111239671707153,
459
+ "step": 205,
460
+ "valid_targets_mean": 944.9,
461
+ "valid_targets_min": 423
462
+ },
463
+ {
464
+ "epoch": 1.3392,
465
+ "grad_norm": 0.6167253454576184,
466
+ "learning_rate": 3.674589103881432e-05,
467
+ "loss": 0.4611,
468
+ "loss_nan_ranks": 0,
469
+ "loss_rank_avg": 0.13232994079589844,
470
+ "step": 210,
471
+ "valid_targets_mean": 811.1,
472
+ "valid_targets_min": 372
473
+ },
474
+ {
475
+ "epoch": 1.3712,
476
+ "grad_norm": 0.5210643419125344,
477
+ "learning_rate": 3.64984700735293e-05,
478
+ "loss": 0.4459,
479
+ "loss_nan_ranks": 0,
480
+ "loss_rank_avg": 0.06730905920267105,
481
+ "step": 215,
482
+ "valid_targets_mean": 581.2,
483
+ "valid_targets_min": 303
484
+ },
485
+ {
486
+ "epoch": 1.4032,
487
+ "grad_norm": 0.6506348468397748,
488
+ "learning_rate": 3.624288222946273e-05,
489
+ "loss": 0.4643,
490
+ "loss_nan_ranks": 0,
491
+ "loss_rank_avg": 0.11295449733734131,
492
+ "step": 220,
493
+ "valid_targets_mean": 779.0,
494
+ "valid_targets_min": 291
495
+ },
496
+ {
497
+ "epoch": 1.4352,
498
+ "grad_norm": 0.6198424313674729,
499
+ "learning_rate": 3.597925402470578e-05,
500
+ "loss": 0.4502,
501
+ "loss_nan_ranks": 0,
502
+ "loss_rank_avg": 0.09262027591466904,
503
+ "step": 225,
504
+ "valid_targets_mean": 662.2,
505
+ "valid_targets_min": 356
506
+ },
507
+ {
508
+ "epoch": 1.4672,
509
+ "grad_norm": 0.5636413590661971,
510
+ "learning_rate": 3.570771595739445e-05,
511
+ "loss": 0.4613,
512
+ "loss_nan_ranks": 0,
513
+ "loss_rank_avg": 0.09992943704128265,
514
+ "step": 230,
515
+ "valid_targets_mean": 901.2,
516
+ "valid_targets_min": 363
517
+ },
518
+ {
519
+ "epoch": 1.4992,
520
+ "grad_norm": 0.5770044697343024,
521
+ "learning_rate": 3.5428402441111964e-05,
522
+ "loss": 0.4609,
523
+ "loss_nan_ranks": 0,
524
+ "loss_rank_avg": 0.09955944120883942,
525
+ "step": 235,
526
+ "valid_targets_mean": 701.8,
527
+ "valid_targets_min": 471
528
+ },
529
+ {
530
+ "epoch": 1.5312000000000001,
531
+ "grad_norm": 0.5816016693103543,
532
+ "learning_rate": 3.5141451738352936e-05,
533
+ "loss": 0.4354,
534
+ "loss_nan_ranks": 0,
535
+ "loss_rank_avg": 0.09309546649456024,
536
+ "step": 240,
537
+ "valid_targets_mean": 610.4,
538
+ "valid_targets_min": 358
539
+ },
540
+ {
541
+ "epoch": 1.5632000000000001,
542
+ "grad_norm": 0.5432246001353459,
543
+ "learning_rate": 3.4847005892082266e-05,
544
+ "loss": 0.4421,
545
+ "loss_nan_ranks": 0,
546
+ "loss_rank_avg": 0.08772096782922745,
547
+ "step": 245,
548
+ "valid_targets_mean": 664.1,
549
+ "valid_targets_min": 359
550
+ },
551
+ {
552
+ "epoch": 1.5952,
553
+ "grad_norm": 0.5245659004342419,
554
+ "learning_rate": 3.454521065542273e-05,
555
+ "loss": 0.432,
556
+ "loss_nan_ranks": 0,
557
+ "loss_rank_avg": 0.09933168441057205,
558
+ "step": 250,
559
+ "valid_targets_mean": 855.1,
560
+ "valid_targets_min": 383
561
+ },
562
+ {
563
+ "epoch": 1.6272,
564
+ "grad_norm": 0.5071814981485833,
565
+ "learning_rate": 3.423621541950597e-05,
566
+ "loss": 0.4395,
567
+ "loss_nan_ranks": 0,
568
+ "loss_rank_avg": 0.08219840377569199,
569
+ "step": 255,
570
+ "valid_targets_mean": 697.7,
571
+ "valid_targets_min": 378
572
+ },
573
+ {
574
+ "epoch": 1.6592,
575
+ "grad_norm": 0.6421027216281103,
576
+ "learning_rate": 3.3920173139522664e-05,
577
+ "loss": 0.4522,
578
+ "loss_nan_ranks": 0,
579
+ "loss_rank_avg": 0.09957920759916306,
580
+ "step": 260,
581
+ "valid_targets_mean": 635.2,
582
+ "valid_targets_min": 329
583
+ },
584
+ {
585
+ "epoch": 1.6912,
586
+ "grad_norm": 0.8918683611862951,
587
+ "learning_rate": 3.35972402590084e-05,
588
+ "loss": 0.4452,
589
+ "loss_nan_ranks": 0,
590
+ "loss_rank_avg": 0.09719720482826233,
591
+ "step": 265,
592
+ "valid_targets_mean": 654.4,
593
+ "valid_targets_min": 368
594
+ },
595
+ {
596
+ "epoch": 1.7231999999999998,
597
+ "grad_norm": 0.5937318332037429,
598
+ "learning_rate": 3.326757663240291e-05,
599
+ "loss": 0.4355,
600
+ "loss_nan_ranks": 0,
601
+ "loss_rank_avg": 0.129220113158226,
602
+ "step": 270,
603
+ "valid_targets_mean": 767.7,
604
+ "valid_targets_min": 334
605
+ },
606
+ {
607
+ "epoch": 1.7551999999999999,
608
+ "grad_norm": 0.615788760605738,
609
+ "learning_rate": 3.293134544592073e-05,
610
+ "loss": 0.4556,
611
+ "loss_nan_ranks": 0,
612
+ "loss_rank_avg": 0.17965131998062134,
613
+ "step": 275,
614
+ "valid_targets_mean": 1080.9,
615
+ "valid_targets_min": 451
616
+ },
617
+ {
618
+ "epoch": 1.7872,
619
+ "grad_norm": 0.575288638742944,
620
+ "learning_rate": 3.258871313677274e-05,
621
+ "loss": 0.4428,
622
+ "loss_nan_ranks": 0,
623
+ "loss_rank_avg": 0.09257206320762634,
624
+ "step": 280,
625
+ "valid_targets_mean": 597.2,
626
+ "valid_targets_min": 320
627
+ },
628
+ {
629
+ "epoch": 1.8192,
630
+ "grad_norm": 0.6226156825477193,
631
+ "learning_rate": 3.2239849310778316e-05,
632
+ "loss": 0.4417,
633
+ "loss_nan_ranks": 0,
634
+ "loss_rank_avg": 0.11135537922382355,
635
+ "step": 285,
636
+ "valid_targets_mean": 747.9,
637
+ "valid_targets_min": 376
638
+ },
639
+ {
640
+ "epoch": 1.8512,
641
+ "grad_norm": 0.5474766803313764,
642
+ "learning_rate": 3.188492665840909e-05,
643
+ "loss": 0.4316,
644
+ "loss_nan_ranks": 0,
645
+ "loss_rank_avg": 0.1179531142115593,
646
+ "step": 290,
647
+ "valid_targets_mean": 1048.9,
648
+ "valid_targets_min": 321
649
+ },
650
+ {
651
+ "epoch": 1.8832,
652
+ "grad_norm": 0.555344017183991,
653
+ "learning_rate": 3.1524120869305726e-05,
654
+ "loss": 0.448,
655
+ "loss_nan_ranks": 0,
656
+ "loss_rank_avg": 0.09857015311717987,
657
+ "step": 295,
658
+ "valid_targets_mean": 785.6,
659
+ "valid_targets_min": 302
660
+ },
661
+ {
662
+ "epoch": 1.9152,
663
+ "grad_norm": 0.5088681384764417,
664
+ "learning_rate": 3.11576105453101e-05,
665
+ "loss": 0.4296,
666
+ "loss_nan_ranks": 0,
667
+ "loss_rank_avg": 0.09216314554214478,
668
+ "step": 300,
669
+ "valid_targets_mean": 784.4,
670
+ "valid_targets_min": 380
671
+ },
672
+ {
673
+ "epoch": 1.9472,
674
+ "grad_norm": 0.6660264590729095,
675
+ "learning_rate": 3.0785577112055916e-05,
676
+ "loss": 0.4423,
677
+ "loss_nan_ranks": 0,
678
+ "loss_rank_avg": 0.07816915959119797,
679
+ "step": 305,
680
+ "valid_targets_mean": 565.1,
681
+ "valid_targets_min": 320
682
+ },
683
+ {
684
+ "epoch": 1.9792,
685
+ "grad_norm": 0.5429833393761442,
686
+ "learning_rate": 3.040820472916153e-05,
687
+ "loss": 0.4335,
688
+ "loss_nan_ranks": 0,
689
+ "loss_rank_avg": 0.11158039420843124,
690
+ "step": 310,
691
+ "valid_targets_mean": 977.5,
692
+ "valid_targets_min": 344
693
+ },
694
+ {
695
+ "epoch": 2.0064,
696
+ "grad_norm": 0.5775581645620036,
697
+ "learning_rate": 3.002568019906939e-05,
698
+ "loss": 0.4487,
699
+ "loss_nan_ranks": 0,
700
+ "loss_rank_avg": 0.07440705597400665,
701
+ "step": 315,
702
+ "valid_targets_mean": 643.6,
703
+ "valid_targets_min": 404
704
+ },
705
+ {
706
+ "epoch": 2.0384,
707
+ "grad_norm": 0.5129583670120458,
708
+ "learning_rate": 2.963819287457733e-05,
709
+ "loss": 0.4121,
710
+ "loss_nan_ranks": 0,
711
+ "loss_rank_avg": 0.07239961624145508,
712
+ "step": 320,
713
+ "valid_targets_mean": 864.1,
714
+ "valid_targets_min": 403
715
+ },
716
+ {
717
+ "epoch": 2.0704,
718
+ "grad_norm": 0.6077115293347921,
719
+ "learning_rate": 2.924593456510733e-05,
720
+ "loss": 0.4031,
721
+ "loss_nan_ranks": 0,
722
+ "loss_rank_avg": 0.12308503687381744,
723
+ "step": 325,
724
+ "valid_targets_mean": 956.9,
725
+ "valid_targets_min": 391
726
+ },
727
+ {
728
+ "epoch": 2.1024,
729
+ "grad_norm": 0.6031214474986006,
730
+ "learning_rate": 2.8849099441758306e-05,
731
+ "loss": 0.4127,
732
+ "loss_nan_ranks": 0,
733
+ "loss_rank_avg": 0.0778675377368927,
734
+ "step": 330,
735
+ "valid_targets_mean": 551.2,
736
+ "valid_targets_min": 404
737
+ },
738
+ {
739
+ "epoch": 2.1344,
740
+ "grad_norm": 0.49034633652229753,
741
+ "learning_rate": 2.844788394118979e-05,
742
+ "loss": 0.4212,
743
+ "loss_nan_ranks": 0,
744
+ "loss_rank_avg": 0.1424838900566101,
745
+ "step": 335,
746
+ "valid_targets_mean": 1388.6,
747
+ "valid_targets_min": 412
748
+ },
749
+ {
750
+ "epoch": 2.1664,
751
+ "grad_norm": 0.6002144395513832,
752
+ "learning_rate": 2.8042486668384164e-05,
753
+ "loss": 0.4195,
754
+ "loss_nan_ranks": 0,
755
+ "loss_rank_avg": 0.09713360667228699,
756
+ "step": 340,
757
+ "valid_targets_mean": 841.2,
758
+ "valid_targets_min": 338
759
+ },
760
+ {
761
+ "epoch": 2.1984,
762
+ "grad_norm": 0.5953750346604982,
763
+ "learning_rate": 2.7633108298335582e-05,
764
+ "loss": 0.4139,
765
+ "loss_nan_ranks": 0,
766
+ "loss_rank_avg": 0.12253813445568085,
767
+ "step": 345,
768
+ "valid_targets_mean": 1029.0,
769
+ "valid_targets_min": 398
770
+ },
771
+ {
772
+ "epoch": 2.2304,
773
+ "grad_norm": 0.5664035810551488,
774
+ "learning_rate": 2.721995147671416e-05,
775
+ "loss": 0.4173,
776
+ "loss_nan_ranks": 0,
777
+ "loss_rank_avg": 0.14460813999176025,
778
+ "step": 350,
779
+ "valid_targets_mean": 1296.0,
780
+ "valid_targets_min": 433
781
+ },
782
+ {
783
+ "epoch": 2.2624,
784
+ "grad_norm": 0.5575494465624244,
785
+ "learning_rate": 2.68032207195547e-05,
786
+ "loss": 0.4159,
787
+ "loss_nan_ranks": 0,
788
+ "loss_rank_avg": 0.11319831013679504,
789
+ "step": 355,
790
+ "valid_targets_mean": 998.4,
791
+ "valid_targets_min": 405
792
+ },
793
+ {
794
+ "epoch": 2.2944,
795
+ "grad_norm": 0.5722654942074888,
796
+ "learning_rate": 2.6383122312019604e-05,
797
+ "loss": 0.4136,
798
+ "loss_nan_ranks": 0,
799
+ "loss_rank_avg": 0.11292511224746704,
800
+ "step": 360,
801
+ "valid_targets_mean": 791.4,
802
+ "valid_targets_min": 357
803
+ },
804
+ {
805
+ "epoch": 2.3264,
806
+ "grad_norm": 0.6156699181787635,
807
+ "learning_rate": 2.595986420628597e-05,
808
+ "loss": 0.4323,
809
+ "loss_nan_ranks": 0,
810
+ "loss_rank_avg": 0.0936831682920456,
811
+ "step": 365,
812
+ "valid_targets_mean": 645.4,
813
+ "valid_targets_min": 413
814
+ },
815
+ {
816
+ "epoch": 2.3584,
817
+ "grad_norm": 0.5799332606546033,
818
+ "learning_rate": 2.5533655918607573e-05,
819
+ "loss": 0.42,
820
+ "loss_nan_ranks": 0,
821
+ "loss_rank_avg": 0.0842270478606224,
822
+ "step": 370,
823
+ "valid_targets_mean": 631.9,
824
+ "valid_targets_min": 319
825
+ },
826
+ {
827
+ "epoch": 2.3904,
828
+ "grad_norm": 0.622585761390544,
829
+ "learning_rate": 2.510470842560259e-05,
830
+ "loss": 0.4189,
831
+ "loss_nan_ranks": 0,
832
+ "loss_rank_avg": 0.12394921481609344,
833
+ "step": 375,
834
+ "valid_targets_mean": 862.4,
835
+ "valid_targets_min": 464
836
+ },
837
+ {
838
+ "epoch": 2.4224,
839
+ "grad_norm": 0.5722309355051565,
840
+ "learning_rate": 2.467323405981841e-05,
841
+ "loss": 0.4051,
842
+ "loss_nan_ranks": 0,
843
+ "loss_rank_avg": 0.0961218997836113,
844
+ "step": 380,
845
+ "valid_targets_mean": 768.4,
846
+ "valid_targets_min": 348
847
+ },
848
+ {
849
+ "epoch": 2.4544,
850
+ "grad_norm": 0.5321841680118645,
851
+ "learning_rate": 2.423944640462533e-05,
852
+ "loss": 0.4146,
853
+ "loss_nan_ranks": 0,
854
+ "loss_rank_avg": 0.0827847570180893,
855
+ "step": 385,
856
+ "valid_targets_mean": 868.4,
857
+ "valid_targets_min": 390
858
+ },
859
+ {
860
+ "epoch": 2.4864,
861
+ "grad_norm": 0.6700888637238239,
862
+ "learning_rate": 2.3803560188490968e-05,
863
+ "loss": 0.431,
864
+ "loss_nan_ranks": 0,
865
+ "loss_rank_avg": 0.09786651283502579,
866
+ "step": 390,
867
+ "valid_targets_mean": 746.6,
868
+ "valid_targets_min": 315
869
+ },
870
+ {
871
+ "epoch": 2.5183999999999997,
872
+ "grad_norm": 0.6323804353095867,
873
+ "learning_rate": 2.336579117868789e-05,
874
+ "loss": 0.4229,
875
+ "loss_nan_ranks": 0,
876
+ "loss_rank_avg": 0.10056309401988983,
877
+ "step": 395,
878
+ "valid_targets_mean": 806.9,
879
+ "valid_targets_min": 394
880
+ },
881
+ {
882
+ "epoch": 2.5504,
883
+ "grad_norm": 0.6550381034455577,
884
+ "learning_rate": 2.292635607448711e-05,
885
+ "loss": 0.4229,
886
+ "loss_nan_ranks": 0,
887
+ "loss_rank_avg": 0.13149245083332062,
888
+ "step": 400,
889
+ "valid_targets_mean": 998.6,
890
+ "valid_targets_min": 403
891
+ },
892
+ {
893
+ "epoch": 2.5824,
894
+ "grad_norm": 0.6645834209618975,
895
+ "learning_rate": 2.248547239989008e-05,
896
+ "loss": 0.4154,
897
+ "loss_nan_ranks": 0,
898
+ "loss_rank_avg": 0.10545995831489563,
899
+ "step": 405,
900
+ "valid_targets_mean": 689.8,
901
+ "valid_targets_min": 440
902
+ },
903
+ {
904
+ "epoch": 2.6144,
905
+ "grad_norm": 0.5941577749283523,
906
+ "learning_rate": 2.204335839595255e-05,
907
+ "loss": 0.4216,
908
+ "loss_nan_ranks": 0,
909
+ "loss_rank_avg": 0.1064852923154831,
910
+ "step": 410,
911
+ "valid_targets_mean": 724.1,
912
+ "valid_targets_min": 376
913
+ },
914
+ {
915
+ "epoch": 2.6464,
916
+ "grad_norm": 0.5056780394022312,
917
+ "learning_rate": 2.1600232912753452e-05,
918
+ "loss": 0.4019,
919
+ "loss_nan_ranks": 0,
920
+ "loss_rank_avg": 0.11357980221509933,
921
+ "step": 415,
922
+ "valid_targets_mean": 1238.4,
923
+ "valid_targets_min": 323
924
+ },
925
+ {
926
+ "epoch": 2.6784,
927
+ "grad_norm": 0.610168786701435,
928
+ "learning_rate": 2.1156315301062293e-05,
929
+ "loss": 0.4064,
930
+ "loss_nan_ranks": 0,
931
+ "loss_rank_avg": 0.09324711561203003,
932
+ "step": 420,
933
+ "valid_targets_mean": 613.2,
934
+ "valid_targets_min": 359
935
+ },
936
+ {
937
+ "epoch": 2.7104,
938
+ "grad_norm": 0.5504726610083012,
939
+ "learning_rate": 2.0711825303758712e-05,
940
+ "loss": 0.4232,
941
+ "loss_nan_ranks": 0,
942
+ "loss_rank_avg": 0.1077696830034256,
943
+ "step": 425,
944
+ "valid_targets_mean": 818.8,
945
+ "valid_targets_min": 390
946
+ },
947
+ {
948
+ "epoch": 2.7424,
949
+ "grad_norm": 0.6641939339447608,
950
+ "learning_rate": 2.0266982947057962e-05,
951
+ "loss": 0.422,
952
+ "loss_nan_ranks": 0,
953
+ "loss_rank_avg": 0.0970541387796402,
954
+ "step": 430,
955
+ "valid_targets_mean": 685.2,
956
+ "valid_targets_min": 321
957
+ },
958
+ {
959
+ "epoch": 2.7744,
960
+ "grad_norm": 0.653531813006077,
961
+ "learning_rate": 1.9822008431596083e-05,
962
+ "loss": 0.4177,
963
+ "loss_nan_ranks": 0,
964
+ "loss_rank_avg": 0.10846106708049774,
965
+ "step": 435,
966
+ "valid_targets_mean": 723.3,
967
+ "valid_targets_min": 404
968
+ },
969
+ {
970
+ "epoch": 2.8064,
971
+ "grad_norm": 0.6681671556144005,
972
+ "learning_rate": 1.937712202342881e-05,
973
+ "loss": 0.4196,
974
+ "loss_nan_ranks": 0,
975
+ "loss_rank_avg": 0.09692089259624481,
976
+ "step": 440,
977
+ "valid_targets_mean": 633.7,
978
+ "valid_targets_min": 427
979
+ },
980
+ {
981
+ "epoch": 2.8384,
982
+ "grad_norm": 0.6380439037814938,
983
+ "learning_rate": 1.8932543944998037e-05,
984
+ "loss": 0.4183,
985
+ "loss_nan_ranks": 0,
986
+ "loss_rank_avg": 0.11875765770673752,
987
+ "step": 445,
988
+ "valid_targets_mean": 746.5,
989
+ "valid_targets_min": 423
990
+ },
991
+ {
992
+ "epoch": 2.8704,
993
+ "grad_norm": 0.5219284155044907,
994
+ "learning_rate": 1.8488494266119877e-05,
995
+ "loss": 0.4058,
996
+ "loss_nan_ranks": 0,
997
+ "loss_rank_avg": 0.09141748398542404,
998
+ "step": 450,
999
+ "valid_targets_mean": 716.4,
1000
+ "valid_targets_min": 366
1001
+ },
1002
+ {
1003
+ "epoch": 2.9024,
1004
+ "grad_norm": 0.5566392528555462,
1005
+ "learning_rate": 1.804519279504834e-05,
1006
+ "loss": 0.4113,
1007
+ "loss_nan_ranks": 0,
1008
+ "loss_rank_avg": 0.09680511057376862,
1009
+ "step": 455,
1010
+ "valid_targets_mean": 934.8,
1011
+ "valid_targets_min": 432
1012
+ },
1013
+ {
1014
+ "epoch": 2.9344,
1015
+ "grad_norm": 0.6626446716495282,
1016
+ "learning_rate": 1.7602858969668365e-05,
1017
+ "loss": 0.409,
1018
+ "loss_nan_ranks": 0,
1019
+ "loss_rank_avg": 0.08213680982589722,
1020
+ "step": 460,
1021
+ "valid_targets_mean": 650.0,
1022
+ "valid_targets_min": 334
1023
+ },
1024
+ {
1025
+ "epoch": 2.9664,
1026
+ "grad_norm": 0.593266768884052,
1027
+ "learning_rate": 1.716171174887231e-05,
1028
+ "loss": 0.4113,
1029
+ "loss_nan_ranks": 0,
1030
+ "loss_rank_avg": 0.10700133442878723,
1031
+ "step": 465,
1032
+ "valid_targets_mean": 696.6,
1033
+ "valid_targets_min": 320
1034
+ },
1035
+ {
1036
+ "epoch": 2.9984,
1037
+ "grad_norm": 0.5752624662339391,
1038
+ "learning_rate": 1.6721969504173484e-05,
1039
+ "loss": 0.42,
1040
+ "loss_nan_ranks": 0,
1041
+ "loss_rank_avg": 0.08804523199796677,
1042
+ "step": 470,
1043
+ "valid_targets_mean": 729.1,
1044
+ "valid_targets_min": 418
1045
+ },
1046
+ {
1047
+ "epoch": 3.0256,
1048
+ "grad_norm": 0.5610382190573227,
1049
+ "learning_rate": 1.628384991161041e-05,
1050
+ "loss": 0.4063,
1051
+ "loss_nan_ranks": 0,
1052
+ "loss_rank_avg": 0.08425762504339218,
1053
+ "step": 475,
1054
+ "valid_targets_mean": 1020.8,
1055
+ "valid_targets_min": 399
1056
+ },
1057
+ {
1058
+ "epoch": 3.0576,
1059
+ "grad_norm": 0.6370374136408906,
1060
+ "learning_rate": 1.5847569843995452e-05,
1061
+ "loss": 0.4027,
1062
+ "loss_nan_ranks": 0,
1063
+ "loss_rank_avg": 0.10353687405586243,
1064
+ "step": 480,
1065
+ "valid_targets_mean": 837.4,
1066
+ "valid_targets_min": 361
1067
+ },
1068
+ {
1069
+ "epoch": 3.0896,
1070
+ "grad_norm": 0.5873572112472938,
1071
+ "learning_rate": 1.5413345263560922e-05,
1072
+ "loss": 0.4117,
1073
+ "loss_nan_ranks": 0,
1074
+ "loss_rank_avg": 0.09108766168355942,
1075
+ "step": 485,
1076
+ "valid_targets_mean": 887.4,
1077
+ "valid_targets_min": 416
1078
+ },
1079
+ {
1080
+ "epoch": 3.1216,
1081
+ "grad_norm": 0.59032555685135,
1082
+ "learning_rate": 1.4981391115056032e-05,
1083
+ "loss": 0.3966,
1084
+ "loss_nan_ranks": 0,
1085
+ "loss_rank_avg": 0.12735995650291443,
1086
+ "step": 490,
1087
+ "valid_targets_mean": 928.2,
1088
+ "valid_targets_min": 397
1089
+ },
1090
+ {
1091
+ "epoch": 3.1536,
1092
+ "grad_norm": 0.5517995378600631,
1093
+ "learning_rate": 1.455192121934748e-05,
1094
+ "loss": 0.3876,
1095
+ "loss_nan_ranks": 0,
1096
+ "loss_rank_avg": 0.09549345076084137,
1097
+ "step": 495,
1098
+ "valid_targets_mean": 842.0,
1099
+ "valid_targets_min": 326
1100
+ },
1101
+ {
1102
+ "epoch": 3.1856,
1103
+ "grad_norm": 0.5767640192840071,
1104
+ "learning_rate": 1.4125148167576303e-05,
1105
+ "loss": 0.3875,
1106
+ "loss_nan_ranks": 0,
1107
+ "loss_rank_avg": 0.09618216753005981,
1108
+ "step": 500,
1109
+ "valid_targets_mean": 1051.3,
1110
+ "valid_targets_min": 357
1111
+ },
1112
+ {
1113
+ "epoch": 3.2176,
1114
+ "grad_norm": 0.614531690871058,
1115
+ "learning_rate": 1.3701283215923563e-05,
1116
+ "loss": 0.3918,
1117
+ "loss_nan_ranks": 0,
1118
+ "loss_rank_avg": 0.11241084337234497,
1119
+ "step": 505,
1120
+ "valid_targets_mean": 977.4,
1121
+ "valid_targets_min": 430
1122
+ },
1123
+ {
1124
+ "epoch": 3.2496,
1125
+ "grad_norm": 0.585115408442605,
1126
+ "learning_rate": 1.328053618103677e-05,
1127
+ "loss": 0.389,
1128
+ "loss_nan_ranks": 0,
1129
+ "loss_rank_avg": 0.07872745394706726,
1130
+ "step": 510,
1131
+ "valid_targets_mean": 600.2,
1132
+ "valid_targets_min": 401
1133
+ },
1134
+ {
1135
+ "epoch": 3.2816,
1136
+ "grad_norm": 0.5155757805806573,
1137
+ "learning_rate": 1.2863115336168916e-05,
1138
+ "loss": 0.3959,
1139
+ "loss_nan_ranks": 0,
1140
+ "loss_rank_avg": 0.07585766911506653,
1141
+ "step": 515,
1142
+ "valid_targets_mean": 734.4,
1143
+ "valid_targets_min": 416
1144
+ },
1145
+ {
1146
+ "epoch": 3.3136,
1147
+ "grad_norm": 0.5335000876085558,
1148
+ "learning_rate": 1.2449227308081509e-05,
1149
+ "loss": 0.3871,
1150
+ "loss_nan_ranks": 0,
1151
+ "loss_rank_avg": 0.0835937038064003,
1152
+ "step": 520,
1153
+ "valid_targets_mean": 825.9,
1154
+ "valid_targets_min": 368
1155
+ },
1156
+ {
1157
+ "epoch": 3.3456,
1158
+ "grad_norm": 0.5841210865260744,
1159
+ "learning_rate": 1.2039076974762587e-05,
1160
+ "loss": 0.3844,
1161
+ "loss_nan_ranks": 0,
1162
+ "loss_rank_avg": 0.07614777237176895,
1163
+ "step": 525,
1164
+ "valid_targets_mean": 763.9,
1165
+ "valid_targets_min": 373
1166
+ },
1167
+ {
1168
+ "epoch": 3.3776,
1169
+ "grad_norm": 0.6483216131482832,
1170
+ "learning_rate": 1.163286736401044e-05,
1171
+ "loss": 0.4018,
1172
+ "loss_nan_ranks": 0,
1173
+ "loss_rank_avg": 0.10778698325157166,
1174
+ "step": 530,
1175
+ "valid_targets_mean": 766.4,
1176
+ "valid_targets_min": 309
1177
+ },
1178
+ {
1179
+ "epoch": 3.4096,
1180
+ "grad_norm": 0.5326152721218573,
1181
+ "learning_rate": 1.123079955293322e-05,
1182
+ "loss": 0.3901,
1183
+ "loss_nan_ranks": 0,
1184
+ "loss_rank_avg": 0.06915357708930969,
1185
+ "step": 535,
1186
+ "valid_targets_mean": 703.1,
1187
+ "valid_targets_min": 329
1188
+ },
1189
+ {
1190
+ "epoch": 3.4416,
1191
+ "grad_norm": 0.600429095738304,
1192
+ "learning_rate": 1.0833072568414037e-05,
1193
+ "loss": 0.4061,
1194
+ "loss_nan_ranks": 0,
1195
+ "loss_rank_avg": 0.07404685020446777,
1196
+ "step": 540,
1197
+ "valid_targets_mean": 642.3,
1198
+ "valid_targets_min": 342
1199
+ },
1200
+ {
1201
+ "epoch": 3.4736000000000002,
1202
+ "grad_norm": 0.6338259376396709,
1203
+ "learning_rate": 1.0439883288591057e-05,
1204
+ "loss": 0.4,
1205
+ "loss_nan_ranks": 0,
1206
+ "loss_rank_avg": 0.1003207415342331,
1207
+ "step": 545,
1208
+ "valid_targets_mean": 815.3,
1209
+ "valid_targets_min": 408
1210
+ },
1211
+ {
1212
+ "epoch": 3.5056000000000003,
1213
+ "grad_norm": 0.5714090152959421,
1214
+ "learning_rate": 1.0051426345401202e-05,
1215
+ "loss": 0.3834,
1216
+ "loss_nan_ranks": 0,
1217
+ "loss_rank_avg": 0.10465012490749359,
1218
+ "step": 550,
1219
+ "valid_targets_mean": 1010.7,
1220
+ "valid_targets_min": 389
1221
+ },
1222
+ {
1223
+ "epoch": 3.5376,
1224
+ "grad_norm": 0.5888843808215544,
1225
+ "learning_rate": 9.667894028235704e-06,
1226
+ "loss": 0.4093,
1227
+ "loss_nan_ranks": 0,
1228
+ "loss_rank_avg": 0.13038086891174316,
1229
+ "step": 555,
1230
+ "valid_targets_mean": 1027.1,
1231
+ "valid_targets_min": 358
1232
+ },
1233
+ {
1234
+ "epoch": 3.5696,
1235
+ "grad_norm": 0.6041611052917156,
1236
+ "learning_rate": 9.289476188755315e-06,
1237
+ "loss": 0.4042,
1238
+ "loss_nan_ranks": 0,
1239
+ "loss_rank_avg": 0.08387233316898346,
1240
+ "step": 560,
1241
+ "valid_targets_mean": 657.9,
1242
+ "valid_targets_min": 401
1243
+ },
1244
+ {
1245
+ "epoch": 3.6016,
1246
+ "grad_norm": 0.6424612269453751,
1247
+ "learning_rate": 8.916360146912122e-06,
1248
+ "loss": 0.3946,
1249
+ "loss_nan_ranks": 0,
1250
+ "loss_rank_avg": 0.07798133790493011,
1251
+ "step": 565,
1252
+ "valid_targets_mean": 693.3,
1253
+ "valid_targets_min": 364
1254
+ },
1255
+ {
1256
+ "epoch": 3.6336,
1257
+ "grad_norm": 0.6656167537622161,
1258
+ "learning_rate": 8.548730598224646e-06,
1259
+ "loss": 0.3991,
1260
+ "loss_nan_ranks": 0,
1261
+ "loss_rank_avg": 0.07147403806447983,
1262
+ "step": 570,
1263
+ "valid_targets_mean": 540.2,
1264
+ "valid_targets_min": 334
1265
+ },
1266
+ {
1267
+ "epoch": 3.6656,
1268
+ "grad_norm": 0.6024773862090805,
1269
+ "learning_rate": 8.186769522352053e-06,
1270
+ "loss": 0.4047,
1271
+ "loss_nan_ranks": 0,
1272
+ "loss_rank_avg": 0.08182119578123093,
1273
+ "step": 575,
1274
+ "valid_targets_mean": 724.2,
1275
+ "valid_targets_min": 268
1276
+ },
1277
+ {
1278
+ "epoch": 3.6976,
1279
+ "grad_norm": 0.5099213228173635,
1280
+ "learning_rate": 7.830656093012714e-06,
1281
+ "loss": 0.3909,
1282
+ "loss_nan_ranks": 0,
1283
+ "loss_rank_avg": 0.06938575953245163,
1284
+ "step": 580,
1285
+ "valid_targets_mean": 856.3,
1286
+ "valid_targets_min": 385
1287
+ },
1288
+ {
1289
+ "epoch": 3.7296,
1290
+ "grad_norm": 0.5878150081667711,
1291
+ "learning_rate": 7.480566589291696e-06,
1292
+ "loss": 0.3963,
1293
+ "loss_nan_ranks": 0,
1294
+ "loss_rank_avg": 0.09519918262958527,
1295
+ "step": 585,
1296
+ "valid_targets_mean": 777.5,
1297
+ "valid_targets_min": 378
1298
+ },
1299
+ {
1300
+ "epoch": 3.7616,
1301
+ "grad_norm": 0.542873280912327,
1302
+ "learning_rate": 7.1366743083812285e-06,
1303
+ "loss": 0.39,
1304
+ "loss_nan_ranks": 0,
1305
+ "loss_rank_avg": 0.06709694862365723,
1306
+ "step": 590,
1307
+ "valid_targets_mean": 632.2,
1308
+ "valid_targets_min": 385
1309
+ },
1310
+ {
1311
+ "epoch": 3.7936,
1312
+ "grad_norm": 2.31349167232845,
1313
+ "learning_rate": 6.799149479797101e-06,
1314
+ "loss": 0.3844,
1315
+ "loss_nan_ranks": 0,
1316
+ "loss_rank_avg": 0.0990547239780426,
1317
+ "step": 595,
1318
+ "valid_targets_mean": 654.9,
1319
+ "valid_targets_min": 302
1320
+ },
1321
+ {
1322
+ "epoch": 3.8256,
1323
+ "grad_norm": 0.590653745798261,
1324
+ "learning_rate": 6.4681591811137e-06,
1325
+ "loss": 0.3759,
1326
+ "loss_nan_ranks": 0,
1327
+ "loss_rank_avg": 0.07994721084833145,
1328
+ "step": 600,
1329
+ "valid_targets_mean": 684.1,
1330
+ "valid_targets_min": 435
1331
+ },
1332
+ {
1333
+ "epoch": 3.8576,
1334
+ "grad_norm": 0.6127118274714611,
1335
+ "learning_rate": 6.143867255259197e-06,
1336
+ "loss": 0.4117,
1337
+ "loss_nan_ranks": 0,
1338
+ "loss_rank_avg": 0.09342561662197113,
1339
+ "step": 605,
1340
+ "valid_targets_mean": 816.8,
1341
+ "valid_targets_min": 284
1342
+ },
1343
+ {
1344
+ "epoch": 3.8895999999999997,
1345
+ "grad_norm": 0.6391548156109519,
1346
+ "learning_rate": 5.8264342294119504e-06,
1347
+ "loss": 0.3895,
1348
+ "loss_nan_ranks": 0,
1349
+ "loss_rank_avg": 0.08812019228935242,
1350
+ "step": 610,
1351
+ "valid_targets_mean": 632.6,
1352
+ "valid_targets_min": 393
1353
+ },
1354
+ {
1355
+ "epoch": 3.9215999999999998,
1356
+ "grad_norm": 0.6060676769815595,
1357
+ "learning_rate": 5.516017235538258e-06,
1358
+ "loss": 0.3969,
1359
+ "loss_nan_ranks": 0,
1360
+ "loss_rank_avg": 0.09914878010749817,
1361
+ "step": 615,
1362
+ "valid_targets_mean": 743.7,
1363
+ "valid_targets_min": 441
1364
+ },
1365
+ {
1366
+ "epoch": 3.9536,
1367
+ "grad_norm": 0.5982168918074733,
1368
+ "learning_rate": 5.212769932610695e-06,
1369
+ "loss": 0.3819,
1370
+ "loss_nan_ranks": 0,
1371
+ "loss_rank_avg": 0.058783818036317825,
1372
+ "step": 620,
1373
+ "valid_targets_mean": 507.6,
1374
+ "valid_targets_min": 400
1375
+ },
1376
+ {
1377
+ "epoch": 3.9856,
1378
+ "grad_norm": 0.5972327172351439,
1379
+ "learning_rate": 4.916842430545681e-06,
1380
+ "loss": 0.4038,
1381
+ "loss_nan_ranks": 0,
1382
+ "loss_rank_avg": 0.08714640885591507,
1383
+ "step": 625,
1384
+ "valid_targets_mean": 827.5,
1385
+ "valid_targets_min": 397
1386
+ },
1387
+ {
1388
+ "epoch": 4.0128,
1389
+ "grad_norm": 0.4847513858068835,
1390
+ "learning_rate": 4.628381215897837e-06,
1391
+ "loss": 0.3936,
1392
+ "loss_nan_ranks": 0,
1393
+ "loss_rank_avg": 0.0801130086183548,
1394
+ "step": 630,
1395
+ "valid_targets_mean": 1127.3,
1396
+ "valid_targets_min": 368
1397
+ },
1398
+ {
1399
+ "epoch": 4.0448,
1400
+ "grad_norm": 0.5611576007645827,
1401
+ "learning_rate": 4.347529079347914e-06,
1402
+ "loss": 0.3824,
1403
+ "loss_nan_ranks": 0,
1404
+ "loss_rank_avg": 0.08794663846492767,
1405
+ "step": 635,
1406
+ "valid_targets_mean": 680.2,
1407
+ "valid_targets_min": 379
1408
+ },
1409
+ {
1410
+ "epoch": 4.0768,
1411
+ "grad_norm": 0.5528385055241838,
1412
+ "learning_rate": 4.074425045020247e-06,
1413
+ "loss": 0.3745,
1414
+ "loss_nan_ranks": 0,
1415
+ "loss_rank_avg": 0.08833988755941391,
1416
+ "step": 640,
1417
+ "valid_targets_mean": 1072.6,
1418
+ "valid_targets_min": 362
1419
+ },
1420
+ {
1421
+ "epoch": 4.1088,
1422
+ "grad_norm": 0.6360019127829928,
1423
+ "learning_rate": 3.8092043016646487e-06,
1424
+ "loss": 0.3931,
1425
+ "loss_nan_ranks": 0,
1426
+ "loss_rank_avg": 0.07920222729444504,
1427
+ "step": 645,
1428
+ "valid_targets_mean": 576.0,
1429
+ "valid_targets_min": 346
1430
+ },
1431
+ {
1432
+ "epoch": 4.1408,
1433
+ "grad_norm": 0.5695054148744619,
1434
+ "learning_rate": 3.551998135736867e-06,
1435
+ "loss": 0.3741,
1436
+ "loss_nan_ranks": 0,
1437
+ "loss_rank_avg": 0.10163244605064392,
1438
+ "step": 650,
1439
+ "valid_targets_mean": 795.4,
1440
+ "valid_targets_min": 389
1441
+ },
1442
+ {
1443
+ "epoch": 4.1728,
1444
+ "grad_norm": 0.5823717131910814,
1445
+ "learning_rate": 3.3029338664107267e-06,
1446
+ "loss": 0.3852,
1447
+ "loss_nan_ranks": 0,
1448
+ "loss_rank_avg": 0.08717401325702667,
1449
+ "step": 655,
1450
+ "valid_targets_mean": 757.6,
1451
+ "valid_targets_min": 450
1452
+ },
1453
+ {
1454
+ "epoch": 4.2048,
1455
+ "grad_norm": 0.5431835475455258,
1456
+ "learning_rate": 3.0621347825540625e-06,
1457
+ "loss": 0.3931,
1458
+ "loss_nan_ranks": 0,
1459
+ "loss_rank_avg": 0.11405445635318756,
1460
+ "step": 660,
1461
+ "valid_targets_mean": 1233.8,
1462
+ "valid_targets_min": 477
1463
+ },
1464
+ {
1465
+ "epoch": 4.2368,
1466
+ "grad_norm": 0.6284908129512062,
1467
+ "learning_rate": 2.8297200816997183e-06,
1468
+ "loss": 0.3871,
1469
+ "loss_nan_ranks": 0,
1470
+ "loss_rank_avg": 0.08217090368270874,
1471
+ "step": 665,
1472
+ "valid_targets_mean": 670.6,
1473
+ "valid_targets_min": 381
1474
+ },
1475
+ {
1476
+ "epoch": 4.2688,
1477
+ "grad_norm": 0.4950334898816112,
1478
+ "learning_rate": 2.605804811041803e-06,
1479
+ "loss": 0.3754,
1480
+ "loss_nan_ranks": 0,
1481
+ "loss_rank_avg": 0.06383413821458817,
1482
+ "step": 670,
1483
+ "valid_targets_mean": 776.5,
1484
+ "valid_targets_min": 410
1485
+ },
1486
+ {
1487
+ "epoch": 4.3008,
1488
+ "grad_norm": 0.591098970902121,
1489
+ "learning_rate": 2.390499810486351e-06,
1490
+ "loss": 0.3935,
1491
+ "loss_nan_ranks": 0,
1492
+ "loss_rank_avg": 0.10034292936325073,
1493
+ "step": 675,
1494
+ "valid_targets_mean": 826.2,
1495
+ "valid_targets_min": 383
1496
+ },
1497
+ {
1498
+ "epoch": 4.3328,
1499
+ "grad_norm": 0.5651496158158206,
1500
+ "learning_rate": 2.183911657784685e-06,
1501
+ "loss": 0.3813,
1502
+ "loss_nan_ranks": 0,
1503
+ "loss_rank_avg": 0.07695989310741425,
1504
+ "step": 680,
1505
+ "valid_targets_mean": 672.1,
1506
+ "valid_targets_min": 385
1507
+ },
1508
+ {
1509
+ "epoch": 4.3648,
1510
+ "grad_norm": 0.5917327908907087,
1511
+ "learning_rate": 1.986142615776532e-06,
1512
+ "loss": 0.38,
1513
+ "loss_nan_ranks": 0,
1514
+ "loss_rank_avg": 0.07897144556045532,
1515
+ "step": 685,
1516
+ "valid_targets_mean": 684.1,
1517
+ "valid_targets_min": 387
1518
+ },
1519
+ {
1520
+ "epoch": 4.3968,
1521
+ "grad_norm": 0.5323303388546669,
1522
+ "learning_rate": 1.7972905817690644e-06,
1523
+ "loss": 0.3905,
1524
+ "loss_nan_ranks": 0,
1525
+ "loss_rank_avg": 0.092294842004776,
1526
+ "step": 690,
1527
+ "valid_targets_mean": 973.1,
1528
+ "valid_targets_min": 401
1529
+ },
1530
+ {
1531
+ "epoch": 4.4288,
1532
+ "grad_norm": 0.5460713542113188,
1533
+ "learning_rate": 1.617449039076955e-06,
1534
+ "loss": 0.3892,
1535
+ "loss_nan_ranks": 0,
1536
+ "loss_rank_avg": 0.09456771612167358,
1537
+ "step": 695,
1538
+ "valid_targets_mean": 962.5,
1539
+ "valid_targets_min": 438
1540
+ },
1541
+ {
1542
+ "epoch": 4.4608,
1543
+ "grad_norm": 0.6082925587977943,
1544
+ "learning_rate": 1.4467070107473413e-06,
1545
+ "loss": 0.3853,
1546
+ "loss_nan_ranks": 0,
1547
+ "loss_rank_avg": 0.1136481761932373,
1548
+ "step": 700,
1549
+ "valid_targets_mean": 849.4,
1550
+ "valid_targets_min": 412
1551
+ },
1552
+ {
1553
+ "epoch": 4.4928,
1554
+ "grad_norm": 0.6200633164112519,
1555
+ "learning_rate": 1.2851490154926816e-06,
1556
+ "loss": 0.3893,
1557
+ "loss_nan_ranks": 0,
1558
+ "loss_rank_avg": 0.10344002395868301,
1559
+ "step": 705,
1560
+ "valid_targets_mean": 861.0,
1561
+ "valid_targets_min": 459
1562
+ },
1563
+ {
1564
+ "epoch": 4.5248,
1565
+ "grad_norm": 0.5980979220481121,
1566
+ "learning_rate": 1.1328550258533211e-06,
1567
+ "loss": 0.387,
1568
+ "loss_nan_ranks": 0,
1569
+ "loss_rank_avg": 0.12262655794620514,
1570
+ "step": 710,
1571
+ "valid_targets_mean": 1103.6,
1572
+ "valid_targets_min": 302
1573
+ },
1574
+ {
1575
+ "epoch": 4.5568,
1576
+ "grad_norm": 1.0985283290768537,
1577
+ "learning_rate": 9.899004286103953e-07,
1578
+ "loss": 0.3909,
1579
+ "loss_nan_ranks": 0,
1580
+ "loss_rank_avg": 0.10326635837554932,
1581
+ "step": 715,
1582
+ "valid_targets_mean": 771.8,
1583
+ "valid_targets_min": 303
1584
+ },
1585
+ {
1586
+ "epoch": 4.5888,
1587
+ "grad_norm": 0.5349623349471714,
1588
+ "learning_rate": 8.5635598746876e-07,
1589
+ "loss": 0.3827,
1590
+ "loss_nan_ranks": 0,
1591
+ "loss_rank_avg": 0.10169753432273865,
1592
+ "step": 720,
1593
+ "valid_targets_mean": 993.0,
1594
+ "valid_targets_min": 347
1595
+ },
1596
+ {
1597
+ "epoch": 4.6208,
1598
+ "grad_norm": 0.5976359862832646,
1599
+ "learning_rate": 7.32287808028389e-07,
1600
+ "loss": 0.3848,
1601
+ "loss_nan_ranks": 0,
1602
+ "loss_rank_avg": 0.10116741061210632,
1603
+ "step": 725,
1604
+ "valid_targets_mean": 912.4,
1605
+ "valid_targets_min": 383
1606
+ },
1607
+ {
1608
+ "epoch": 4.6528,
1609
+ "grad_norm": 0.6339445206464914,
1610
+ "learning_rate": 6.177573050615327e-07,
1611
+ "loss": 0.3946,
1612
+ "loss_nan_ranks": 0,
1613
+ "loss_rank_avg": 0.08657615631818771,
1614
+ "step": 730,
1615
+ "valid_targets_mean": 686.4,
1616
+ "valid_targets_min": 384
1617
+ },
1618
+ {
1619
+ "epoch": 4.6848,
1620
+ "grad_norm": 0.5154419341958988,
1621
+ "learning_rate": 5.128211721119213e-07,
1622
+ "loss": 0.3741,
1623
+ "loss_nan_ranks": 0,
1624
+ "loss_rank_avg": 0.08375303447246552,
1625
+ "step": 735,
1626
+ "valid_targets_mean": 967.8,
1627
+ "valid_targets_min": 383
1628
+ },
1629
+ {
1630
+ "epoch": 4.7168,
1631
+ "grad_norm": 0.6321852313540388,
1632
+ "learning_rate": 4.175313534309755e-07,
1633
+ "loss": 0.3956,
1634
+ "loss_nan_ranks": 0,
1635
+ "loss_rank_avg": 0.13004527986049652,
1636
+ "step": 740,
1637
+ "valid_targets_mean": 931.1,
1638
+ "valid_targets_min": 399
1639
+ },
1640
+ {
1641
+ "epoch": 4.7488,
1642
+ "grad_norm": 0.6067056281796016,
1643
+ "learning_rate": 3.319350182649861e-07,
1644
+ "loss": 0.3782,
1645
+ "loss_nan_ranks": 0,
1646
+ "loss_rank_avg": 0.1023097038269043,
1647
+ "step": 745,
1648
+ "valid_targets_mean": 758.2,
1649
+ "valid_targets_min": 303
1650
+ },
1651
+ {
1652
+ "epoch": 4.7808,
1653
+ "grad_norm": 0.6919237537475112,
1654
+ "learning_rate": 2.560745375059392e-07,
1655
+ "loss": 0.3991,
1656
+ "loss_nan_ranks": 0,
1657
+ "loss_rank_avg": 0.08634231984615326,
1658
+ "step": 750,
1659
+ "valid_targets_mean": 746.2,
1660
+ "valid_targets_min": 390
1661
+ },
1662
+ {
1663
+ "epoch": 4.8128,
1664
+ "grad_norm": 0.5637550282681978,
1665
+ "learning_rate": 1.8998746271758016e-07,
1666
+ "loss": 0.383,
1667
+ "loss_nan_ranks": 0,
1668
+ "loss_rank_avg": 0.10297013819217682,
1669
+ "step": 755,
1670
+ "valid_targets_mean": 907.4,
1671
+ "valid_targets_min": 393
1672
+ },
1673
+ {
1674
+ "epoch": 4.8448,
1675
+ "grad_norm": 0.5702544895441023,
1676
+ "learning_rate": 1.337065075470778e-07,
1677
+ "loss": 0.3915,
1678
+ "loss_nan_ranks": 0,
1679
+ "loss_rank_avg": 0.15079092979431152,
1680
+ "step": 760,
1681
+ "valid_targets_mean": 1253.9,
1682
+ "valid_targets_min": 356
1683
+ },
1684
+ {
1685
+ "epoch": 4.8768,
1686
+ "grad_norm": 0.5882537057091088,
1687
+ "learning_rate": 8.725953153150279e-08,
1688
+ "loss": 0.3964,
1689
+ "loss_nan_ranks": 0,
1690
+ "loss_rank_avg": 0.07823218405246735,
1691
+ "step": 765,
1692
+ "valid_targets_mean": 649.6,
1693
+ "valid_targets_min": 448
1694
+ },
1695
+ {
1696
+ "epoch": 4.9088,
1697
+ "grad_norm": 0.6073688670028073,
1698
+ "learning_rate": 5.066952630711886e-08,
1699
+ "loss": 0.3864,
1700
+ "loss_nan_ranks": 0,
1701
+ "loss_rank_avg": 0.07631102204322815,
1702
+ "step": 770,
1703
+ "valid_targets_mean": 616.8,
1704
+ "valid_targets_min": 379
1705
+ },
1706
+ {
1707
+ "epoch": 4.9408,
1708
+ "grad_norm": 0.5206072014992669,
1709
+ "learning_rate": 2.3954604228342283e-08,
1710
+ "loss": 0.3705,
1711
+ "loss_nan_ranks": 0,
1712
+ "loss_rank_avg": 0.09787902235984802,
1713
+ "step": 775,
1714
+ "valid_targets_mean": 1027.2,
1715
+ "valid_targets_min": 365
1716
+ },
1717
+ {
1718
+ "epoch": 4.9728,
1719
+ "grad_norm": 0.5742498548264887,
1720
+ "learning_rate": 7.12798940197601e-09,
1721
+ "loss": 0.3679,
1722
+ "loss_nan_ranks": 0,
1723
+ "loss_rank_avg": 0.10835174471139908,
1724
+ "step": 780,
1725
+ "valid_targets_mean": 1055.2,
1726
+ "valid_targets_min": 405
1727
+ },
1728
+ {
1729
+ "epoch": 5.0,
1730
+ "grad_norm": 1.1933161078427301,
1731
+ "learning_rate": 1.9801114115480802e-10,
1732
+ "loss": 0.3872,
1733
+ "loss_nan_ranks": 0,
1734
+ "loss_rank_avg": 0.391443133354187,
1735
+ "step": 785,
1736
+ "valid_targets_mean": 719.2,
1737
+ "valid_targets_min": 319
1738
+ },
1739
+ {
1740
+ "epoch": 5.0,
1741
+ "loss_nan_ranks": 0,
1742
+ "loss_rank_avg": 0.391443133354187,
1743
+ "step": 785,
1744
+ "total_flos": 2.9988290575938355e+17,
1745
+ "train_loss": 0.4420624729934012,
1746
+ "train_runtime": 7853.4449,
1747
+ "train_samples_per_second": 6.363,
1748
+ "train_steps_per_second": 0.1,
1749
+ "valid_targets_mean": 719.2,
1750
+ "valid_targets_min": 319
1751
+ }
1752
+ ],
1753
+ "logging_steps": 5,
1754
+ "max_steps": 785,
1755
+ "num_input_tokens_seen": 0,
1756
+ "num_train_epochs": 5,
1757
+ "save_steps": 500,
1758
+ "stateful_callbacks": {
1759
+ "TrainerControl": {
1760
+ "args": {
1761
+ "should_epoch_stop": false,
1762
+ "should_evaluate": false,
1763
+ "should_log": false,
1764
+ "should_save": false,
1765
+ "should_training_stop": false
1766
+ },
1767
+ "attributes": {}
1768
+ }
1769
+ },
1770
+ "total_flos": 2.9988290575938355e+17,
1771
+ "train_batch_size": 1,
1772
+ "trial_name": null,
1773
+ "trial_params": null
1774
+ }
training_loss.png ADDED