File size: 6,280 Bytes
6586016
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
{
  "best_metric": 0.9650432050274941,
  "best_model_checkpoint": "swin-base-patch4-window7-224-in22k-finetuned-brain-tumor-final_08/checkpoint-154",
  "epoch": 9.87012987012987,
  "global_step": 190,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.52,
      "learning_rate": 2.6315789473684212e-05,
      "loss": 1.2882,
      "step": 10
    },
    {
      "epoch": 0.99,
      "eval_accuracy": 0.7961508248232522,
      "eval_f1_score": 0.7862764915452001,
      "eval_loss": 0.5468631982803345,
      "eval_precision": 0.807682539972508,
      "eval_runtime": 146.6896,
      "eval_samples_per_second": 17.356,
      "eval_steps_per_second": 0.177,
      "step": 19
    },
    {
      "epoch": 1.04,
      "learning_rate": 4.970760233918128e-05,
      "loss": 0.7346,
      "step": 20
    },
    {
      "epoch": 1.56,
      "learning_rate": 4.678362573099415e-05,
      "loss": 0.3491,
      "step": 30
    },
    {
      "epoch": 1.97,
      "eval_accuracy": 0.8860958366064414,
      "eval_f1_score": 0.8877831452046004,
      "eval_loss": 0.3029632568359375,
      "eval_precision": 0.8981099107198931,
      "eval_runtime": 142.8458,
      "eval_samples_per_second": 17.823,
      "eval_steps_per_second": 0.182,
      "step": 38
    },
    {
      "epoch": 2.08,
      "learning_rate": 4.3859649122807014e-05,
      "loss": 0.2344,
      "step": 40
    },
    {
      "epoch": 2.6,
      "learning_rate": 4.093567251461988e-05,
      "loss": 0.1791,
      "step": 50
    },
    {
      "epoch": 2.96,
      "eval_accuracy": 0.9210526315789473,
      "eval_f1_score": 0.9229181063056022,
      "eval_loss": 0.2076900452375412,
      "eval_precision": 0.9307481907300132,
      "eval_runtime": 146.5105,
      "eval_samples_per_second": 17.378,
      "eval_steps_per_second": 0.177,
      "step": 57
    },
    {
      "epoch": 3.12,
      "learning_rate": 3.8011695906432746e-05,
      "loss": 0.1365,
      "step": 60
    },
    {
      "epoch": 3.64,
      "learning_rate": 3.508771929824561e-05,
      "loss": 0.122,
      "step": 70
    },
    {
      "epoch": 4.0,
      "eval_accuracy": 0.9253731343283582,
      "eval_f1_score": 0.9271544463725769,
      "eval_loss": 0.20073173940181732,
      "eval_precision": 0.936852899850916,
      "eval_runtime": 143.3187,
      "eval_samples_per_second": 17.765,
      "eval_steps_per_second": 0.181,
      "step": 77
    },
    {
      "epoch": 4.16,
      "learning_rate": 3.216374269005848e-05,
      "loss": 0.084,
      "step": 80
    },
    {
      "epoch": 4.68,
      "learning_rate": 2.9239766081871346e-05,
      "loss": 0.0671,
      "step": 90
    },
    {
      "epoch": 4.99,
      "eval_accuracy": 0.9269442262372348,
      "eval_f1_score": 0.9294001146483667,
      "eval_loss": 0.20734713971614838,
      "eval_precision": 0.9400926555225891,
      "eval_runtime": 150.1774,
      "eval_samples_per_second": 16.953,
      "eval_steps_per_second": 0.173,
      "step": 96
    },
    {
      "epoch": 5.19,
      "learning_rate": 2.6315789473684212e-05,
      "loss": 0.0545,
      "step": 100
    },
    {
      "epoch": 5.71,
      "learning_rate": 2.3391812865497074e-05,
      "loss": 0.0474,
      "step": 110
    },
    {
      "epoch": 5.97,
      "eval_accuracy": 0.9481539670070699,
      "eval_f1_score": 0.9493645555588069,
      "eval_loss": 0.13840945065021515,
      "eval_precision": 0.9547416860173351,
      "eval_runtime": 148.6502,
      "eval_samples_per_second": 17.127,
      "eval_steps_per_second": 0.175,
      "step": 115
    },
    {
      "epoch": 6.23,
      "learning_rate": 2.046783625730994e-05,
      "loss": 0.0359,
      "step": 120
    },
    {
      "epoch": 6.75,
      "learning_rate": 1.7543859649122806e-05,
      "loss": 0.032,
      "step": 130
    },
    {
      "epoch": 6.96,
      "eval_accuracy": 0.9430479183032208,
      "eval_f1_score": 0.9447208627751791,
      "eval_loss": 0.1683335155248642,
      "eval_precision": 0.9510759669338126,
      "eval_runtime": 146.9976,
      "eval_samples_per_second": 17.32,
      "eval_steps_per_second": 0.177,
      "step": 134
    },
    {
      "epoch": 7.27,
      "learning_rate": 1.4619883040935673e-05,
      "loss": 0.0284,
      "step": 140
    },
    {
      "epoch": 7.79,
      "learning_rate": 1.1695906432748537e-05,
      "loss": 0.0225,
      "step": 150
    },
    {
      "epoch": 8.0,
      "eval_accuracy": 0.9650432050274941,
      "eval_f1_score": 0.9657333174830298,
      "eval_loss": 0.11006712168455124,
      "eval_precision": 0.9671471415955248,
      "eval_runtime": 144.8303,
      "eval_samples_per_second": 17.579,
      "eval_steps_per_second": 0.18,
      "step": 154
    },
    {
      "epoch": 8.31,
      "learning_rate": 8.771929824561403e-06,
      "loss": 0.0299,
      "step": 160
    },
    {
      "epoch": 8.83,
      "learning_rate": 5.8479532163742686e-06,
      "loss": 0.0193,
      "step": 170
    },
    {
      "epoch": 8.99,
      "eval_accuracy": 0.9532600157109191,
      "eval_f1_score": 0.9543637358051864,
      "eval_loss": 0.13721555471420288,
      "eval_precision": 0.9584728079885907,
      "eval_runtime": 148.8815,
      "eval_samples_per_second": 17.101,
      "eval_steps_per_second": 0.175,
      "step": 173
    },
    {
      "epoch": 9.35,
      "learning_rate": 2.9239766081871343e-06,
      "loss": 0.0175,
      "step": 180
    },
    {
      "epoch": 9.87,
      "learning_rate": 0.0,
      "loss": 0.0193,
      "step": 190
    },
    {
      "epoch": 9.87,
      "eval_accuracy": 0.9591516103692066,
      "eval_f1_score": 0.9600211574702097,
      "eval_loss": 0.12095439434051514,
      "eval_precision": 0.9627515459909033,
      "eval_runtime": 143.4592,
      "eval_samples_per_second": 17.747,
      "eval_steps_per_second": 0.181,
      "step": 190
    },
    {
      "epoch": 9.87,
      "step": 190,
      "total_flos": 5.909979961466413e+18,
      "train_loss": 0.1843041891330167,
      "train_runtime": 18944.3436,
      "train_samples_per_second": 4.031,
      "train_steps_per_second": 0.01
    }
  ],
  "max_steps": 190,
  "num_train_epochs": 10,
  "total_flos": 5.909979961466413e+18,
  "trial_name": null,
  "trial_params": null
}