File size: 5,545 Bytes
bff892a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "best_metric": 0.9094533029612756,
  "best_model_checkpoint": "swin-base-patch4-window7-224-in22k-finetuned-brain-tumor-final_07/checkpoint-130",
  "epoch": 9.81132075471698,
  "global_step": 130,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.75,
      "learning_rate": 3.846153846153846e-05,
      "loss": 1.3335,
      "step": 10
    },
    {
      "epoch": 0.98,
      "eval_accuracy": 0.628132118451025,
      "eval_f1_score": 0.6110605089750341,
      "eval_loss": 0.9195472002029419,
      "eval_precision": 0.7245146412168497,
      "eval_runtime": 97.9643,
      "eval_samples_per_second": 17.925,
      "eval_steps_per_second": 0.184,
      "step": 13
    },
    {
      "epoch": 1.51,
      "learning_rate": 4.700854700854701e-05,
      "loss": 0.6062,
      "step": 20
    },
    {
      "epoch": 1.96,
      "eval_accuracy": 0.7625284738041003,
      "eval_f1_score": 0.7672952986163042,
      "eval_loss": 0.6113870739936829,
      "eval_precision": 0.8385249066441203,
      "eval_runtime": 97.5664,
      "eval_samples_per_second": 17.998,
      "eval_steps_per_second": 0.184,
      "step": 26
    },
    {
      "epoch": 2.26,
      "learning_rate": 4.2735042735042735e-05,
      "loss": 0.274,
      "step": 30
    },
    {
      "epoch": 2.94,
      "eval_accuracy": 0.7801822323462415,
      "eval_f1_score": 0.7771794448506547,
      "eval_loss": 0.5468003153800964,
      "eval_precision": 0.853286516043446,
      "eval_runtime": 97.7463,
      "eval_samples_per_second": 17.965,
      "eval_steps_per_second": 0.184,
      "step": 39
    },
    {
      "epoch": 3.02,
      "learning_rate": 3.846153846153846e-05,
      "loss": 0.1784,
      "step": 40
    },
    {
      "epoch": 3.77,
      "learning_rate": 3.418803418803419e-05,
      "loss": 0.1211,
      "step": 50
    },
    {
      "epoch": 4.0,
      "eval_accuracy": 0.8416856492027335,
      "eval_f1_score": 0.8416977239442979,
      "eval_loss": 0.3922261893749237,
      "eval_precision": 0.874900462813103,
      "eval_runtime": 98.1852,
      "eval_samples_per_second": 17.885,
      "eval_steps_per_second": 0.183,
      "step": 53
    },
    {
      "epoch": 4.53,
      "learning_rate": 2.9914529914529915e-05,
      "loss": 0.0991,
      "step": 60
    },
    {
      "epoch": 4.98,
      "eval_accuracy": 0.8171981776765376,
      "eval_f1_score": 0.8208785805809438,
      "eval_loss": 0.47336336970329285,
      "eval_precision": 0.8802151863176946,
      "eval_runtime": 98.7771,
      "eval_samples_per_second": 17.777,
      "eval_steps_per_second": 0.182,
      "step": 66
    },
    {
      "epoch": 5.28,
      "learning_rate": 2.564102564102564e-05,
      "loss": 0.0682,
      "step": 70
    },
    {
      "epoch": 5.96,
      "eval_accuracy": 0.8599088838268792,
      "eval_f1_score": 0.8599673085486621,
      "eval_loss": 0.3750819265842438,
      "eval_precision": 0.8882136644762273,
      "eval_runtime": 97.3464,
      "eval_samples_per_second": 18.039,
      "eval_steps_per_second": 0.185,
      "step": 79
    },
    {
      "epoch": 6.04,
      "learning_rate": 2.1367521367521368e-05,
      "loss": 0.0491,
      "step": 80
    },
    {
      "epoch": 6.79,
      "learning_rate": 1.7094017094017095e-05,
      "loss": 0.0414,
      "step": 90
    },
    {
      "epoch": 6.94,
      "eval_accuracy": 0.8986332574031891,
      "eval_f1_score": 0.8994870623108869,
      "eval_loss": 0.29505661129951477,
      "eval_precision": 0.9099680823778629,
      "eval_runtime": 97.1448,
      "eval_samples_per_second": 18.076,
      "eval_steps_per_second": 0.185,
      "step": 92
    },
    {
      "epoch": 7.55,
      "learning_rate": 1.282051282051282e-05,
      "loss": 0.0264,
      "step": 100
    },
    {
      "epoch": 8.0,
      "eval_accuracy": 0.8843963553530751,
      "eval_f1_score": 0.8854880922887878,
      "eval_loss": 0.34849926829338074,
      "eval_precision": 0.9069081629635263,
      "eval_runtime": 98.0596,
      "eval_samples_per_second": 17.907,
      "eval_steps_per_second": 0.184,
      "step": 106
    },
    {
      "epoch": 8.3,
      "learning_rate": 8.547008547008548e-06,
      "loss": 0.021,
      "step": 110
    },
    {
      "epoch": 8.98,
      "eval_accuracy": 0.8764236902050114,
      "eval_f1_score": 0.8782390518192311,
      "eval_loss": 0.38029375672340393,
      "eval_precision": 0.9030614046715562,
      "eval_runtime": 97.9149,
      "eval_samples_per_second": 17.934,
      "eval_steps_per_second": 0.184,
      "step": 119
    },
    {
      "epoch": 9.06,
      "learning_rate": 4.273504273504274e-06,
      "loss": 0.0195,
      "step": 120
    },
    {
      "epoch": 9.81,
      "learning_rate": 0.0,
      "loss": 0.0151,
      "step": 130
    },
    {
      "epoch": 9.81,
      "eval_accuracy": 0.9094533029612756,
      "eval_f1_score": 0.9094938197442792,
      "eval_loss": 0.29040291905403137,
      "eval_precision": 0.9188664294996836,
      "eval_runtime": 97.8314,
      "eval_samples_per_second": 17.949,
      "eval_steps_per_second": 0.184,
      "step": 130
    },
    {
      "epoch": 9.81,
      "step": 130,
      "total_flos": 4.0507959904511017e+18,
      "train_loss": 0.21944753378629683,
      "train_runtime": 12678.0007,
      "train_samples_per_second": 4.154,
      "train_steps_per_second": 0.01
    }
  ],
  "max_steps": 130,
  "num_train_epochs": 10,
  "total_flos": 4.0507959904511017e+18,
  "trial_name": null,
  "trial_params": null
}