Hugging Face's logo

corranm
/
square_run_second_vote

Image Classification
Transformers
TensorBoard
Safetensors
vit
Generated from Trainer
Model card Files
xet
Metrics Community
corranm commited on
Commit
ea94f0e
·
verified ·
1 Parent(s): 1961f4e

End of training

Browse files
Files changed (13) hide show
  1. README.md +100 -0
  2. all_results.json +22 -0
  3. config.json +42 -0
  4. eval_results.json +17 -0
  5. model.safetensors +3 -0
  6. preprocessor_config.json +23 -0
  7. runs/Feb03_18-45-28_modal/events.out.tfevents.1738608362.modal.2.0 +3 -0
  8. runs/Feb03_18-45-28_modal/events.out.tfevents.1738608362.modal.2.1 +3 -0
  9. runs/Feb03_18-45-28_modal/events.out.tfevents.1738608742.modal.2.2 +3 -0
  10. runs/Feb03_18-45-28_modal/events.out.tfevents.1738608742.modal.2.3 +3 -0
  11. train_results.json +8 -0
  12. trainer_state.json +0 -0
  13. training_args.bin +3 -0
README.md ADDED
@@ -0,0 +1,100 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: google/vit-base-patch16-224
5
+ tags:
6
+ - generated_from_trainer
7
+ metrics:
8
+ - accuracy
9
+ model-index:
10
+ - name: square_run_second_vote
11
+ results: []
12
+ ---
13
+
14
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
+ should probably proofread and complete it, then remove this comment. -->
16
+
17
+ # square_run_second_vote
18
+
19
+ This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on an unknown dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 1.1557
22
+ - F1 Macro: 0.5777
23
+ - F1 Micro: 0.6667
24
+ - F1 Weighted: 0.6629
25
+ - Precision Macro: 0.5756
26
+ - Precision Micro: 0.6667
27
+ - Precision Weighted: 0.6734
28
+ - Recall Macro: 0.5912
29
+ - Recall Micro: 0.6667
30
+ - Recall Weighted: 0.6667
31
+ - Accuracy: 0.6667
32
+
33
+ ## Model description
34
+
35
+ More information needed
36
+
37
+ ## Intended uses & limitations
38
+
39
+ More information needed
40
+
41
+ ## Training and evaluation data
42
+
43
+ More information needed
44
+
45
+ ## Training procedure
46
+
47
+ ### Training hyperparameters
48
+
49
+ The following hyperparameters were used during training:
50
+ - learning_rate: 0.0001
51
+ - train_batch_size: 8
52
+ - eval_batch_size: 8
53
+ - seed: 42
54
+ - optimizer: Use OptimizerNames.ADAMW_BNB with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
55
+ - lr_scheduler_type: linear
56
+ - lr_scheduler_warmup_ratio: 0.1
57
+ - num_epochs: 30
58
+
59
+ ### Training results
60
+
61
+ | Training Loss | Epoch | Step | Validation Loss | F1 Macro | F1 Micro | F1 Weighted | Precision Macro | Precision Micro | Precision Weighted | Recall Macro | Recall Micro | Recall Weighted | Accuracy |
62
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:--------:|:-----------:|:---------------:|:---------------:|:------------------:|:------------:|:------------:|:---------------:|:--------:|
63
+ | 1.8754 | 1.0 | 58 | 1.7961 | 0.1385 | 0.2803 | 0.1722 | 0.1426 | 0.2803 | 0.1603 | 0.2098 | 0.2803 | 0.2803 | 0.2803 |
64
+ | 2.0246 | 2.0 | 116 | 2.0138 | 0.2236 | 0.3106 | 0.2484 | 0.2558 | 0.3106 | 0.2692 | 0.2842 | 0.3106 | 0.3106 | 0.3106 |
65
+ | 1.6189 | 3.0 | 174 | 1.5039 | 0.2444 | 0.3864 | 0.3195 | 0.2633 | 0.3864 | 0.3301 | 0.2847 | 0.3864 | 0.3864 | 0.3864 |
66
+ | 1.3445 | 4.0 | 232 | 1.3982 | 0.3287 | 0.4394 | 0.3866 | 0.3186 | 0.4394 | 0.3730 | 0.3696 | 0.4394 | 0.4394 | 0.4394 |
67
+ | 1.3387 | 5.0 | 290 | 1.1920 | 0.4401 | 0.5758 | 0.5265 | 0.4315 | 0.5758 | 0.5031 | 0.4683 | 0.5758 | 0.5758 | 0.5758 |
68
+ | 1.1664 | 6.0 | 348 | 1.1778 | 0.4179 | 0.5076 | 0.4988 | 0.5068 | 0.5076 | 0.5862 | 0.4395 | 0.5076 | 0.5076 | 0.5076 |
69
+ | 1.1622 | 7.0 | 406 | 1.1723 | 0.4518 | 0.5379 | 0.5251 | 0.4514 | 0.5379 | 0.5526 | 0.4867 | 0.5379 | 0.5379 | 0.5379 |
70
+ | 0.9827 | 8.0 | 464 | 1.0619 | 0.5084 | 0.6212 | 0.6074 | 0.5037 | 0.6212 | 0.6140 | 0.5345 | 0.6212 | 0.6212 | 0.6212 |
71
+ | 1.3416 | 9.0 | 522 | 1.3995 | 0.3997 | 0.5 | 0.4690 | 0.4218 | 0.5 | 0.5024 | 0.4509 | 0.5 | 0.5 | 0.5 |
72
+ | 0.758 | 10.0 | 580 | 1.1693 | 0.5066 | 0.5985 | 0.5836 | 0.5262 | 0.5985 | 0.6031 | 0.5279 | 0.5985 | 0.5985 | 0.5985 |
73
+ | 0.7758 | 11.0 | 638 | 1.0800 | 0.5491 | 0.6515 | 0.6320 | 0.5729 | 0.6515 | 0.6501 | 0.5710 | 0.6515 | 0.6515 | 0.6515 |
74
+ | 0.2319 | 12.0 | 696 | 1.1553 | 0.5467 | 0.6742 | 0.6410 | 0.5816 | 0.6742 | 0.6699 | 0.5711 | 0.6742 | 0.6742 | 0.6742 |
75
+ | 0.3528 | 13.0 | 754 | 1.1685 | 0.5794 | 0.6894 | 0.6711 | 0.5887 | 0.6894 | 0.6752 | 0.5955 | 0.6894 | 0.6894 | 0.6894 |
76
+ | 0.6238 | 14.0 | 812 | 1.1781 | 0.5579 | 0.6439 | 0.6285 | 0.5451 | 0.6439 | 0.6278 | 0.5856 | 0.6439 | 0.6439 | 0.6439 |
77
+ | 0.1869 | 15.0 | 870 | 1.2305 | 0.5146 | 0.6061 | 0.5983 | 0.5032 | 0.6061 | 0.6013 | 0.5369 | 0.6061 | 0.6061 | 0.6061 |
78
+ | 0.1015 | 16.0 | 928 | 1.3576 | 0.5019 | 0.5909 | 0.5932 | 0.5440 | 0.5909 | 0.6312 | 0.4959 | 0.5909 | 0.5909 | 0.5909 |
79
+ | 0.3809 | 17.0 | 986 | 1.2998 | 0.5667 | 0.6591 | 0.6527 | 0.5828 | 0.6591 | 0.6885 | 0.5838 | 0.6591 | 0.6591 | 0.6591 |
80
+ | 0.0887 | 18.0 | 1044 | 1.4154 | 0.5572 | 0.6667 | 0.6489 | 0.5682 | 0.6667 | 0.6518 | 0.5683 | 0.6667 | 0.6667 | 0.6667 |
81
+ | 0.1422 | 19.0 | 1102 | 1.3989 | 0.5609 | 0.6667 | 0.6472 | 0.5672 | 0.6667 | 0.6420 | 0.5695 | 0.6667 | 0.6667 | 0.6667 |
82
+ | 0.0037 | 20.0 | 1160 | 1.5134 | 0.5242 | 0.6212 | 0.6078 | 0.5263 | 0.6212 | 0.6093 | 0.5374 | 0.6212 | 0.6212 | 0.6212 |
83
+ | 0.0602 | 21.0 | 1218 | 1.5349 | 0.5660 | 0.6667 | 0.6544 | 0.5710 | 0.6667 | 0.6503 | 0.5671 | 0.6667 | 0.6667 | 0.6667 |
84
+ | 0.0353 | 22.0 | 1276 | 1.4489 | 0.6137 | 0.7045 | 0.6919 | 0.6146 | 0.7045 | 0.6909 | 0.6242 | 0.7045 | 0.7045 | 0.7045 |
85
+ | 0.001 | 23.0 | 1334 | 1.4781 | 0.5715 | 0.6667 | 0.6541 | 0.5657 | 0.6667 | 0.6449 | 0.5805 | 0.6667 | 0.6667 | 0.6667 |
86
+ | 0.0007 | 24.0 | 1392 | 1.6326 | 0.5713 | 0.6591 | 0.6511 | 0.5871 | 0.6591 | 0.6648 | 0.5786 | 0.6591 | 0.6591 | 0.6591 |
87
+ | 0.0084 | 25.0 | 1450 | 1.5856 | 0.5684 | 0.6591 | 0.6569 | 0.5662 | 0.6591 | 0.6672 | 0.5802 | 0.6591 | 0.6591 | 0.6591 |
88
+ | 0.0008 | 26.0 | 1508 | 1.5799 | 0.5826 | 0.6818 | 0.6675 | 0.5849 | 0.6818 | 0.6632 | 0.5884 | 0.6818 | 0.6818 | 0.6818 |
89
+ | 0.0053 | 27.0 | 1566 | 1.5308 | 0.5719 | 0.6667 | 0.6556 | 0.5667 | 0.6667 | 0.6524 | 0.5843 | 0.6667 | 0.6667 | 0.6667 |
90
+ | 0.0004 | 28.0 | 1624 | 1.5639 | 0.5732 | 0.6667 | 0.6617 | 0.5684 | 0.6667 | 0.6673 | 0.5867 | 0.6667 | 0.6667 | 0.6667 |
91
+ | 0.0007 | 29.0 | 1682 | 1.5346 | 0.5835 | 0.6742 | 0.6678 | 0.5786 | 0.6742 | 0.6703 | 0.5965 | 0.6742 | 0.6742 | 0.6742 |
92
+ | 0.0004 | 30.0 | 1740 | 1.5232 | 0.5791 | 0.6742 | 0.6661 | 0.5707 | 0.6742 | 0.6628 | 0.5918 | 0.6742 | 0.6742 | 0.6742 |
93
+
94
+
95
+ ### Framework versions
96
+
97
+ - Transformers 4.48.2
98
+ - Pytorch 2.6.0+cu124
99
+ - Datasets 3.2.0
100
+ - Tokenizers 0.21.0
all_results.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 30.0,
3
+ "eval_accuracy": 0.6666666666666666,
4
+ "eval_f1_macro": 0.5776781856118128,
5
+ "eval_f1_micro": 0.6666666666666666,
6
+ "eval_f1_weighted": 0.6629079228897741,
7
+ "eval_loss": 1.1556771993637085,
8
+ "eval_precision_macro": 0.5755952380952382,
9
+ "eval_precision_micro": 0.6666666666666666,
10
+ "eval_precision_weighted": 0.6734217171717172,
11
+ "eval_recall_macro": 0.591248037676609,
12
+ "eval_recall_micro": 0.6666666666666666,
13
+ "eval_recall_weighted": 0.6666666666666666,
14
+ "eval_runtime": 1.7249,
15
+ "eval_samples_per_second": 38.263,
16
+ "eval_steps_per_second": 5.218,
17
+ "total_flos": 1.0740871074163507e+18,
18
+ "train_loss": 0.5156426236089975,
19
+ "train_runtime": 378.1612,
20
+ "train_samples_per_second": 36.651,
21
+ "train_steps_per_second": 4.601
22
+ }
config.json ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/vit-base-patch16-224",
3
+ "architectures": [
4
+ "ViTForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "encoder_stride": 16,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.0,
10
+ "hidden_size": 768,
11
+ "id2label": {
12
+ "0": "-",
13
+ "1": "0",
14
+ "2": "1",
15
+ "3": "2",
16
+ "4": "3",
17
+ "5": "4",
18
+ "6": "5"
19
+ },
20
+ "image_size": 224,
21
+ "initializer_range": 0.02,
22
+ "intermediate_size": 3072,
23
+ "label2id": {
24
+ "-": "0",
25
+ "0": "1",
26
+ "1": "2",
27
+ "2": "3",
28
+ "3": "4",
29
+ "4": "5",
30
+ "5": "6"
31
+ },
32
+ "layer_norm_eps": 1e-12,
33
+ "model_type": "vit",
34
+ "num_attention_heads": 12,
35
+ "num_channels": 3,
36
+ "num_hidden_layers": 12,
37
+ "patch_size": 16,
38
+ "problem_type": "single_label_classification",
39
+ "qkv_bias": true,
40
+ "torch_dtype": "float32",
41
+ "transformers_version": "4.48.2"
42
+ }
eval_results.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 30.0,
3
+ "eval_accuracy": 0.6666666666666666,
4
+ "eval_f1_macro": 0.5776781856118128,
5
+ "eval_f1_micro": 0.6666666666666666,
6
+ "eval_f1_weighted": 0.6629079228897741,
7
+ "eval_loss": 1.1556771993637085,
8
+ "eval_precision_macro": 0.5755952380952382,
9
+ "eval_precision_micro": 0.6666666666666666,
10
+ "eval_precision_weighted": 0.6734217171717172,
11
+ "eval_recall_macro": 0.591248037676609,
12
+ "eval_recall_micro": 0.6666666666666666,
13
+ "eval_recall_weighted": 0.6666666666666666,
14
+ "eval_runtime": 1.7249,
15
+ "eval_samples_per_second": 38.263,
16
+ "eval_steps_per_second": 5.218
17
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e87ccf5947f2c929fc967bf5ab416e2500420684a09880ad0629e3e9ab317a68
3
+ size 343239356
preprocessor_config.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_convert_rgb": null,
3
+ "do_normalize": true,
4
+ "do_rescale": true,
5
+ "do_resize": true,
6
+ "image_mean": [
7
+ 0.5,
8
+ 0.5,
9
+ 0.5
10
+ ],
11
+ "image_processor_type": "ViTImageProcessorFast",
12
+ "image_std": [
13
+ 0.5,
14
+ 0.5,
15
+ 0.5
16
+ ],
17
+ "resample": 2,
18
+ "rescale_factor": 0.00392156862745098,
19
+ "size": {
20
+ "height": 224,
21
+ "width": 224
22
+ }
23
+ }
runs/Feb03_18-45-28_modal/events.out.tfevents.1738608362.modal.2.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:902ed88d2d0760a9879a8da6cbb266bfb61f70ea8683e9bb088b2b6c0117142c
3
+ size 213806
runs/Feb03_18-45-28_modal/events.out.tfevents.1738608362.modal.2.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f5769b6ebec514c693adb8496372171773415707fb31d1784431de70f18313fc
3
+ size 213806
runs/Feb03_18-45-28_modal/events.out.tfevents.1738608742.modal.2.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:47062e45f4d77d5f724bc0c7d5cb55423cd4883dd05f901b76ce381116b7209d
3
+ size 921
runs/Feb03_18-45-28_modal/events.out.tfevents.1738608742.modal.2.3 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f73641760d929749bee768f91a45e6ae04d35a1524b10a829ec64649cd99e9fe
3
+ size 921
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 30.0,
3
+ "total_flos": 1.0740871074163507e+18,
4
+ "train_loss": 0.5156426236089975,
5
+ "train_runtime": 378.1612,
6
+ "train_samples_per_second": 36.651,
7
+ "train_steps_per_second": 4.601
8
+ }
trainer_state.json ADDED
The diff for this file is too large to render. See raw diff
 
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:743505198ea453ed51e2c886685528a2a4e477dd00eb7b56af52effe91f89d2b
3
+ size 5368