Hugging Face's logo

Encore02
/
vit-weldclassifyv4

Image Classification
Transformers
TensorBoard
Safetensors
vit
Generated from Trainer
Eval Results
Model card Files
xet
Metrics Community
Encore02 commited on
Commit
f00c9de
·
verified ·
1 Parent(s): 2c0ae60

Model save

Browse files
Files changed (3) hide show
  1. README.md +21 -30
  2. data/events.out.tfevents.1730702085.22f267685eef.199.0 +2 -2
  3. model.safetensors +1 -1
README.md CHANGED
@@ -3,7 +3,6 @@ library_name: transformers
3
  license: apache-2.0
4
  base_model: google/vit-base-patch16-224-in21k
5
  tags:
6
- - image-classification
7
  - generated_from_trainer
8
  datasets:
9
  - imagefolder
@@ -24,7 +23,7 @@ model-index:
24
  metrics:
25
  - name: Accuracy
26
  type: accuracy
27
- value: 0.9244604316546763
28
  ---
29
 
30
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -34,8 +33,8 @@ should probably proofread and complete it, then remove this comment. -->
34
 
35
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the imagefolder dataset.
36
  It achieves the following results on the evaluation set:
37
- - Loss: 0.3162
38
- - Accuracy: 0.9245
39
 
40
  ## Model description
41
 
@@ -60,36 +59,28 @@ The following hyperparameters were used during training:
60
  - seed: 42
61
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
62
  - lr_scheduler_type: linear
63
- - num_epochs: 15
64
  - mixed_precision_training: Native AMP
65
 
66
  ### Training results
67
 
68
- | Training Loss | Epoch | Step | Validation Loss | Accuracy |
69
- |:-------------:|:-------:|:----:|:---------------:|:--------:|
70
- | 0.8146 | 0.6410 | 100 | 0.7349 | 0.6835 |
71
- | 0.6048 | 1.2821 | 200 | 0.6821 | 0.6978 |
72
- | 0.4796 | 1.9231 | 300 | 0.4833 | 0.8129 |
73
- | 0.4532 | 2.5641 | 400 | 0.5380 | 0.8022 |
74
- | 0.1242 | 3.2051 | 500 | 0.3899 | 0.8741 |
75
- | 0.124 | 3.8462 | 600 | 0.5237 | 0.8273 |
76
- | 0.1239 | 4.4872 | 700 | 0.4221 | 0.8849 |
77
- | 0.0785 | 5.1282 | 800 | 0.3683 | 0.9137 |
78
- | 0.093 | 5.7692 | 900 | 0.6376 | 0.8597 |
79
- | 0.0056 | 6.4103 | 1000 | 0.3162 | 0.9245 |
80
- | 0.0472 | 7.0513 | 1100 | 0.5225 | 0.8885 |
81
- | 0.0234 | 7.6923 | 1200 | 0.6096 | 0.8597 |
82
- | 0.0354 | 8.3333 | 1300 | 0.5520 | 0.8777 |
83
- | 0.026 | 8.9744 | 1400 | 0.4938 | 0.8993 |
84
- | 0.002 | 9.6154 | 1500 | 0.4350 | 0.9173 |
85
- | 0.0021 | 10.2564 | 1600 | 0.4224 | 0.9173 |
86
- | 0.0016 | 10.8974 | 1700 | 0.3838 | 0.9281 |
87
- | 0.0014 | 11.5385 | 1800 | 0.3943 | 0.9281 |
88
- | 0.0013 | 12.1795 | 1900 | 0.4012 | 0.9281 |
89
- | 0.0012 | 12.8205 | 2000 | 0.4067 | 0.9281 |
90
- | 0.0011 | 13.4615 | 2100 | 0.4101 | 0.9281 |
91
- | 0.0011 | 14.1026 | 2200 | 0.4124 | 0.9281 |
92
- | 0.0012 | 14.7436 | 2300 | 0.4136 | 0.9281 |
93
 
94
 
95
  ### Framework versions
 
3
  license: apache-2.0
4
  base_model: google/vit-base-patch16-224-in21k
5
  tags:
 
6
  - generated_from_trainer
7
  datasets:
8
  - imagefolder
 
23
  metrics:
24
  - name: Accuracy
25
  type: accuracy
26
+ value: 0.9388489208633094
27
  ---
28
 
29
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
33
 
34
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the imagefolder dataset.
35
  It achieves the following results on the evaluation set:
36
+ - Loss: 0.3090
37
+ - Accuracy: 0.9388
38
 
39
  ## Model description
40
 
 
59
  - seed: 42
60
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
61
  - lr_scheduler_type: linear
62
+ - num_epochs: 10
63
  - mixed_precision_training: Native AMP
64
 
65
  ### Training results
66
 
67
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
68
+ |:-------------:|:------:|:----:|:---------------:|:--------:|
69
+ | 0.8791 | 0.6410 | 100 | 0.7507 | 0.7014 |
70
+ | 0.7436 | 1.2821 | 200 | 0.5400 | 0.7698 |
71
+ | 0.4783 | 1.9231 | 300 | 0.4832 | 0.8345 |
72
+ | 0.3055 | 2.5641 | 400 | 0.2734 | 0.9101 |
73
+ | 0.2407 | 3.2051 | 500 | 0.2732 | 0.9173 |
74
+ | 0.1367 | 3.8462 | 600 | 0.4062 | 0.8525 |
75
+ | 0.0943 | 4.4872 | 700 | 0.3188 | 0.9137 |
76
+ | 0.0938 | 5.1282 | 800 | 0.3211 | 0.9173 |
77
+ | 0.0352 | 5.7692 | 900 | 0.3002 | 0.9281 |
78
+ | 0.0054 | 6.4103 | 1000 | 0.3863 | 0.9245 |
79
+ | 0.0397 | 7.0513 | 1100 | 0.3790 | 0.9317 |
80
+ | 0.0038 | 7.6923 | 1200 | 0.2968 | 0.9388 |
81
+ | 0.0035 | 8.3333 | 1300 | 0.2937 | 0.9353 |
82
+ | 0.003 | 8.9744 | 1400 | 0.3026 | 0.9388 |
83
+ | 0.0031 | 9.6154 | 1500 | 0.3090 | 0.9388 |
 
 
 
 
 
 
 
 
84
 
85
 
86
  ### Framework versions
data/events.out.tfevents.1730702085.22f267685eef.199.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:823a0787d7fcc292f458a756a343495974625e35ad92a76640716d7d1053a149
3
- size 41347
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:864bc61f257884e377411ea6ae2173bcf7dba5b1591ec64aeb72cc22925c35fc
3
+ size 42967
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:556fd05c163447aa259686603fb9622294a07f25301c52225329a85959cc4f11
3
  size 343230128
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5da1b3eb249cedd4aac6e60c83b75727801ed75a3a5c146d1a831af53e5f3368
3
  size 343230128