DHEIVER commited on
Commit
dc2b267
·
1 Parent(s): 30276aa

Upload 11 files

Browse files
.gitattributes CHANGED
@@ -2,7 +2,6 @@
2
  *.arrow filter=lfs diff=lfs merge=lfs -text
3
  *.bin filter=lfs diff=lfs merge=lfs -text
4
  *.bz2 filter=lfs diff=lfs merge=lfs -text
5
- *.ckpt filter=lfs diff=lfs merge=lfs -text
6
  *.ftz filter=lfs diff=lfs merge=lfs -text
7
  *.gz filter=lfs diff=lfs merge=lfs -text
8
  *.h5 filter=lfs diff=lfs merge=lfs -text
@@ -25,7 +24,6 @@
25
  *.safetensors filter=lfs diff=lfs merge=lfs -text
26
  saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
  *.tar.* filter=lfs diff=lfs merge=lfs -text
28
- *.tar filter=lfs diff=lfs merge=lfs -text
29
  *.tflite filter=lfs diff=lfs merge=lfs -text
30
  *.tgz filter=lfs diff=lfs merge=lfs -text
31
  *.wasm filter=lfs diff=lfs merge=lfs -text
 
2
  *.arrow filter=lfs diff=lfs merge=lfs -text
3
  *.bin filter=lfs diff=lfs merge=lfs -text
4
  *.bz2 filter=lfs diff=lfs merge=lfs -text
 
5
  *.ftz filter=lfs diff=lfs merge=lfs -text
6
  *.gz filter=lfs diff=lfs merge=lfs -text
7
  *.h5 filter=lfs diff=lfs merge=lfs -text
 
24
  *.safetensors filter=lfs diff=lfs merge=lfs -text
25
  saved_model/**/* filter=lfs diff=lfs merge=lfs -text
26
  *.tar.* filter=lfs diff=lfs merge=lfs -text
 
27
  *.tflite filter=lfs diff=lfs merge=lfs -text
28
  *.tgz filter=lfs diff=lfs merge=lfs -text
29
  *.wasm filter=lfs diff=lfs merge=lfs -text
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ checkpoint-*/
README.md ADDED
@@ -0,0 +1,96 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - generated_from_trainer
5
+ datasets:
6
+ - imagefolder
7
+ metrics:
8
+ - accuracy
9
+ - f1
10
+ - recall
11
+ - precision
12
+ model-index:
13
+ - name: Brain_Tumor_Classification
14
+ results:
15
+ - task:
16
+ name: Image Classification
17
+ type: image-classification
18
+ dataset:
19
+ name: imagefolder
20
+ type: imagefolder
21
+ config: default
22
+ split: train
23
+ args: default
24
+ metrics:
25
+ - name: Accuracy
26
+ type: accuracy
27
+ value: 0.9646761984861227
28
+ - name: F1
29
+ type: f1
30
+ value: 0.9646761984861227
31
+ - name: Recall
32
+ type: recall
33
+ value: 0.9646761984861227
34
+ - name: Precision
35
+ type: precision
36
+ value: 0.9646761984861227
37
+ ---
38
+
39
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
40
+ should probably proofread and complete it, then remove this comment. -->
41
+
42
+ # Brain_Tumor_Classification
43
+
44
+ This model is a fine-tuned version of [microsoft/swin-tiny-patch4-window7-224](https://huggingface.co/microsoft/swin-tiny-patch4-window7-224) on the imagefolder dataset.
45
+ It achieves the following results on the evaluation set:
46
+ - Loss: 0.1012
47
+ - Accuracy: 0.9647
48
+ - F1: 0.9647
49
+ - Recall: 0.9647
50
+ - Precision: 0.9647
51
+
52
+ ## Model description
53
+
54
+ More information needed
55
+
56
+ ## Intended uses & limitations
57
+
58
+ More information needed
59
+
60
+ ## Training and evaluation data
61
+
62
+ More information needed
63
+
64
+ ## Training procedure
65
+
66
+ ### Training hyperparameters
67
+
68
+ The following hyperparameters were used during training:
69
+ - learning_rate: 5e-05
70
+ - train_batch_size: 32
71
+ - eval_batch_size: 32
72
+ - seed: 42
73
+ - gradient_accumulation_steps: 4
74
+ - total_train_batch_size: 128
75
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
76
+ - lr_scheduler_type: linear
77
+ - lr_scheduler_warmup_ratio: 0.1
78
+ - num_epochs: 5
79
+
80
+ ### Training results
81
+
82
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1 | Recall | Precision |
83
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:------:|:---------:|
84
+ | 0.4856 | 0.99 | 83 | 0.3771 | 0.8444 | 0.8444 | 0.8444 | 0.8444 |
85
+ | 0.3495 | 1.99 | 166 | 0.2608 | 0.8949 | 0.8949 | 0.8949 | 0.8949 |
86
+ | 0.252 | 2.99 | 249 | 0.1445 | 0.9487 | 0.9487 | 0.9487 | 0.9487 |
87
+ | 0.2364 | 3.99 | 332 | 0.1029 | 0.9588 | 0.9588 | 0.9588 | 0.9588 |
88
+ | 0.2178 | 4.99 | 415 | 0.1012 | 0.9647 | 0.9647 | 0.9647 | 0.9647 |
89
+
90
+
91
+ ### Framework versions
92
+
93
+ - Transformers 4.23.1
94
+ - Pytorch 1.12.1
95
+ - Datasets 2.6.1
96
+ - Tokenizers 0.13.1
all_results.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 4.99,
3
+ "eval_accuracy": 0.9646761984861227,
4
+ "eval_f1": 0.9646761984861227,
5
+ "eval_loss": 0.10117336362600327,
6
+ "eval_precision": 0.9646761984861227,
7
+ "eval_recall": 0.9646761984861227,
8
+ "eval_runtime": 60.8804,
9
+ "eval_samples_per_second": 19.53,
10
+ "eval_steps_per_second": 0.624,
11
+ "total_flos": 1.3278811358711808e+18,
12
+ "train_loss": 0.383831198531461,
13
+ "train_runtime": 7560.7227,
14
+ "train_samples_per_second": 7.075,
15
+ "train_steps_per_second": 0.055
16
+ }
config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "microsoft/swin-tiny-patch4-window7-224",
3
+ "architectures": [
4
+ "SwinForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "depths": [
8
+ 2,
9
+ 2,
10
+ 6,
11
+ 2
12
+ ],
13
+ "drop_path_rate": 0.1,
14
+ "embed_dim": 96,
15
+ "encoder_stride": 32,
16
+ "hidden_act": "gelu",
17
+ "hidden_dropout_prob": 0.0,
18
+ "hidden_size": 768,
19
+ "id2label": {
20
+ "0": "glioma_tumor",
21
+ "1": "meningioma_tumor",
22
+ "2": "no_tumor",
23
+ "3": "pituitary_tumor"
24
+ },
25
+ "image_size": 224,
26
+ "initializer_range": 0.02,
27
+ "label2id": {
28
+ "glioma_tumor": 0,
29
+ "meningioma_tumor": 1,
30
+ "no_tumor": 2,
31
+ "pituitary_tumor": 3
32
+ },
33
+ "layer_norm_eps": 1e-05,
34
+ "mlp_ratio": 4.0,
35
+ "model_type": "swin",
36
+ "num_channels": 3,
37
+ "num_heads": [
38
+ 3,
39
+ 6,
40
+ 12,
41
+ 24
42
+ ],
43
+ "num_layers": 4,
44
+ "patch_size": 4,
45
+ "path_norm": true,
46
+ "problem_type": "single_label_classification",
47
+ "qkv_bias": true,
48
+ "torch_dtype": "float32",
49
+ "transformers_version": "4.23.1",
50
+ "use_absolute_embeddings": false,
51
+ "window_size": 7
52
+ }
eval_results.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 4.99,
3
+ "eval_accuracy": 0.9646761984861227,
4
+ "eval_f1": 0.9646761984861227,
5
+ "eval_loss": 0.10117336362600327,
6
+ "eval_precision": 0.9646761984861227,
7
+ "eval_recall": 0.9646761984861227,
8
+ "eval_runtime": 60.8804,
9
+ "eval_samples_per_second": 19.53,
10
+ "eval_steps_per_second": 0.624
11
+ }
preprocessor_config.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_resize": true,
4
+ "feature_extractor_type": "ViTFeatureExtractor",
5
+ "image_mean": [
6
+ 0.485,
7
+ 0.456,
8
+ 0.406
9
+ ],
10
+ "image_std": [
11
+ 0.229,
12
+ 0.224,
13
+ 0.225
14
+ ],
15
+ "resample": 3,
16
+ "size": 224
17
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aa057f5f4594c8baa83623ad82e194d4139d97e45f0254a73cca2fdfc1b05218
3
+ size 110395759
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 4.99,
3
+ "total_flos": 1.3278811358711808e+18,
4
+ "train_loss": 0.383831198531461,
5
+ "train_runtime": 7560.7227,
6
+ "train_samples_per_second": 7.075,
7
+ "train_steps_per_second": 0.055
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,331 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.9646761984861227,
3
+ "best_model_checkpoint": "Brain_Tumor_Classification/checkpoint-415",
4
+ "epoch": 4.991044776119403,
5
+ "global_step": 415,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.12,
12
+ "learning_rate": 1.1904761904761905e-05,
13
+ "loss": 1.3844,
14
+ "step": 10
15
+ },
16
+ {
17
+ "epoch": 0.24,
18
+ "learning_rate": 2.380952380952381e-05,
19
+ "loss": 1.2375,
20
+ "step": 20
21
+ },
22
+ {
23
+ "epoch": 0.36,
24
+ "learning_rate": 3.571428571428572e-05,
25
+ "loss": 0.9742,
26
+ "step": 30
27
+ },
28
+ {
29
+ "epoch": 0.48,
30
+ "learning_rate": 4.761904761904762e-05,
31
+ "loss": 0.7601,
32
+ "step": 40
33
+ },
34
+ {
35
+ "epoch": 0.6,
36
+ "learning_rate": 4.8927613941018765e-05,
37
+ "loss": 0.7047,
38
+ "step": 50
39
+ },
40
+ {
41
+ "epoch": 0.72,
42
+ "learning_rate": 4.7587131367292224e-05,
43
+ "loss": 0.5255,
44
+ "step": 60
45
+ },
46
+ {
47
+ "epoch": 0.84,
48
+ "learning_rate": 4.624664879356568e-05,
49
+ "loss": 0.5059,
50
+ "step": 70
51
+ },
52
+ {
53
+ "epoch": 0.96,
54
+ "learning_rate": 4.490616621983915e-05,
55
+ "loss": 0.4856,
56
+ "step": 80
57
+ },
58
+ {
59
+ "epoch": 0.99,
60
+ "eval_accuracy": 0.8444070647603028,
61
+ "eval_f1": 0.8444070647603028,
62
+ "eval_loss": 0.3771306276321411,
63
+ "eval_precision": 0.8444070647603028,
64
+ "eval_recall": 0.8444070647603028,
65
+ "eval_runtime": 65.7487,
66
+ "eval_samples_per_second": 18.084,
67
+ "eval_steps_per_second": 0.578,
68
+ "step": 83
69
+ },
70
+ {
71
+ "epoch": 1.08,
72
+ "learning_rate": 4.35656836461126e-05,
73
+ "loss": 0.4626,
74
+ "step": 90
75
+ },
76
+ {
77
+ "epoch": 1.2,
78
+ "learning_rate": 4.222520107238606e-05,
79
+ "loss": 0.4249,
80
+ "step": 100
81
+ },
82
+ {
83
+ "epoch": 1.32,
84
+ "learning_rate": 4.088471849865952e-05,
85
+ "loss": 0.3855,
86
+ "step": 110
87
+ },
88
+ {
89
+ "epoch": 1.44,
90
+ "learning_rate": 3.954423592493298e-05,
91
+ "loss": 0.385,
92
+ "step": 120
93
+ },
94
+ {
95
+ "epoch": 1.56,
96
+ "learning_rate": 3.8203753351206436e-05,
97
+ "loss": 0.3522,
98
+ "step": 130
99
+ },
100
+ {
101
+ "epoch": 1.68,
102
+ "learning_rate": 3.6863270777479895e-05,
103
+ "loss": 0.367,
104
+ "step": 140
105
+ },
106
+ {
107
+ "epoch": 1.8,
108
+ "learning_rate": 3.5522788203753354e-05,
109
+ "loss": 0.3172,
110
+ "step": 150
111
+ },
112
+ {
113
+ "epoch": 1.92,
114
+ "learning_rate": 3.418230563002681e-05,
115
+ "loss": 0.3495,
116
+ "step": 160
117
+ },
118
+ {
119
+ "epoch": 1.99,
120
+ "eval_accuracy": 0.894869638351556,
121
+ "eval_f1": 0.894869638351556,
122
+ "eval_loss": 0.2608145773410797,
123
+ "eval_precision": 0.894869638351556,
124
+ "eval_recall": 0.894869638351556,
125
+ "eval_runtime": 66.7411,
126
+ "eval_samples_per_second": 17.815,
127
+ "eval_steps_per_second": 0.569,
128
+ "step": 166
129
+ },
130
+ {
131
+ "epoch": 2.05,
132
+ "learning_rate": 3.284182305630027e-05,
133
+ "loss": 0.3677,
134
+ "step": 170
135
+ },
136
+ {
137
+ "epoch": 2.17,
138
+ "learning_rate": 3.1501340482573724e-05,
139
+ "loss": 0.2809,
140
+ "step": 180
141
+ },
142
+ {
143
+ "epoch": 2.29,
144
+ "learning_rate": 3.0160857908847186e-05,
145
+ "loss": 0.297,
146
+ "step": 190
147
+ },
148
+ {
149
+ "epoch": 2.41,
150
+ "learning_rate": 2.8820375335120648e-05,
151
+ "loss": 0.3134,
152
+ "step": 200
153
+ },
154
+ {
155
+ "epoch": 2.53,
156
+ "learning_rate": 2.7479892761394104e-05,
157
+ "loss": 0.2306,
158
+ "step": 210
159
+ },
160
+ {
161
+ "epoch": 2.64,
162
+ "learning_rate": 2.6139410187667563e-05,
163
+ "loss": 0.2442,
164
+ "step": 220
165
+ },
166
+ {
167
+ "epoch": 2.76,
168
+ "learning_rate": 2.479892761394102e-05,
169
+ "loss": 0.2885,
170
+ "step": 230
171
+ },
172
+ {
173
+ "epoch": 2.88,
174
+ "learning_rate": 2.3458445040214477e-05,
175
+ "loss": 0.252,
176
+ "step": 240
177
+ },
178
+ {
179
+ "epoch": 2.99,
180
+ "eval_accuracy": 0.9486963835155593,
181
+ "eval_f1": 0.9486963835155594,
182
+ "eval_loss": 0.14449167251586914,
183
+ "eval_precision": 0.9486963835155593,
184
+ "eval_recall": 0.9486963835155593,
185
+ "eval_runtime": 64.3986,
186
+ "eval_samples_per_second": 18.463,
187
+ "eval_steps_per_second": 0.59,
188
+ "step": 249
189
+ },
190
+ {
191
+ "epoch": 3.01,
192
+ "learning_rate": 2.211796246648794e-05,
193
+ "loss": 0.2939,
194
+ "step": 250
195
+ },
196
+ {
197
+ "epoch": 3.13,
198
+ "learning_rate": 2.0777479892761395e-05,
199
+ "loss": 0.2099,
200
+ "step": 260
201
+ },
202
+ {
203
+ "epoch": 3.25,
204
+ "learning_rate": 1.9436997319034853e-05,
205
+ "loss": 0.2477,
206
+ "step": 270
207
+ },
208
+ {
209
+ "epoch": 3.37,
210
+ "learning_rate": 1.8096514745308312e-05,
211
+ "loss": 0.2674,
212
+ "step": 280
213
+ },
214
+ {
215
+ "epoch": 3.49,
216
+ "learning_rate": 1.675603217158177e-05,
217
+ "loss": 0.2538,
218
+ "step": 290
219
+ },
220
+ {
221
+ "epoch": 3.61,
222
+ "learning_rate": 1.5415549597855227e-05,
223
+ "loss": 0.2201,
224
+ "step": 300
225
+ },
226
+ {
227
+ "epoch": 3.73,
228
+ "learning_rate": 1.4075067024128689e-05,
229
+ "loss": 0.2591,
230
+ "step": 310
231
+ },
232
+ {
233
+ "epoch": 3.85,
234
+ "learning_rate": 1.2734584450402146e-05,
235
+ "loss": 0.2087,
236
+ "step": 320
237
+ },
238
+ {
239
+ "epoch": 3.97,
240
+ "learning_rate": 1.1394101876675605e-05,
241
+ "loss": 0.2364,
242
+ "step": 330
243
+ },
244
+ {
245
+ "epoch": 3.99,
246
+ "eval_accuracy": 0.95878889823381,
247
+ "eval_f1": 0.95878889823381,
248
+ "eval_loss": 0.10286473482847214,
249
+ "eval_precision": 0.95878889823381,
250
+ "eval_recall": 0.95878889823381,
251
+ "eval_runtime": 65.1006,
252
+ "eval_samples_per_second": 18.264,
253
+ "eval_steps_per_second": 0.584,
254
+ "step": 332
255
+ },
256
+ {
257
+ "epoch": 4.1,
258
+ "learning_rate": 1.0053619302949062e-05,
259
+ "loss": 0.2312,
260
+ "step": 340
261
+ },
262
+ {
263
+ "epoch": 4.21,
264
+ "learning_rate": 8.71313672922252e-06,
265
+ "loss": 0.2011,
266
+ "step": 350
267
+ },
268
+ {
269
+ "epoch": 4.33,
270
+ "learning_rate": 7.372654155495978e-06,
271
+ "loss": 0.1961,
272
+ "step": 360
273
+ },
274
+ {
275
+ "epoch": 4.45,
276
+ "learning_rate": 6.032171581769437e-06,
277
+ "loss": 0.2277,
278
+ "step": 370
279
+ },
280
+ {
281
+ "epoch": 4.57,
282
+ "learning_rate": 4.691689008042896e-06,
283
+ "loss": 0.2084,
284
+ "step": 380
285
+ },
286
+ {
287
+ "epoch": 4.69,
288
+ "learning_rate": 3.351206434316354e-06,
289
+ "loss": 0.2478,
290
+ "step": 390
291
+ },
292
+ {
293
+ "epoch": 4.81,
294
+ "learning_rate": 2.0107238605898126e-06,
295
+ "loss": 0.1992,
296
+ "step": 400
297
+ },
298
+ {
299
+ "epoch": 4.93,
300
+ "learning_rate": 6.702412868632708e-07,
301
+ "loss": 0.2178,
302
+ "step": 410
303
+ },
304
+ {
305
+ "epoch": 4.99,
306
+ "eval_accuracy": 0.9646761984861227,
307
+ "eval_f1": 0.9646761984861227,
308
+ "eval_loss": 0.10117336362600327,
309
+ "eval_precision": 0.9646761984861227,
310
+ "eval_recall": 0.9646761984861227,
311
+ "eval_runtime": 66.0046,
312
+ "eval_samples_per_second": 18.014,
313
+ "eval_steps_per_second": 0.576,
314
+ "step": 415
315
+ },
316
+ {
317
+ "epoch": 4.99,
318
+ "step": 415,
319
+ "total_flos": 1.3278811358711808e+18,
320
+ "train_loss": 0.383831198531461,
321
+ "train_runtime": 7560.7227,
322
+ "train_samples_per_second": 7.075,
323
+ "train_steps_per_second": 0.055
324
+ }
325
+ ],
326
+ "max_steps": 415,
327
+ "num_train_epochs": 5,
328
+ "total_flos": 1.3278811358711808e+18,
329
+ "trial_name": null,
330
+ "trial_params": null
331
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a6cd87ae59bd56ba9839d7fcc3640bfd1702e1825502c61e4444a7236643cb9f
3
+ size 3375