Hugging Face's logo

bluebird089
/
test-vit

Image Classification
Transformers
TensorBoard
Safetensors
vit
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
bluebird089 commited on
Commit
f3a8dfa
•
1 Parent(s): 40410ad

End of training

Browse files
Files changed (3) hide show
  1. README.md +13 -6
  2. model.safetensors +1 -1
  3. runs/Mar18_14-16-36_3d7c35467368/events.out.tfevents.1710771397.3d7c35467368.5146.8 +2 -2
README.md CHANGED
@@ -17,8 +17,8 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 1.3926
21
- - Accuracy: 0.8693
22
 
23
  ## Model description
24
 
@@ -46,15 +46,22 @@ The following hyperparameters were used during training:
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
  - lr_scheduler_warmup_ratio: 0.1
49
- - num_epochs: 3
50
 
51
  ### Training results
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
- | No log | 0.84 | 4 | 1.6451 | 0.6535 |
56
- | No log | 1.89 | 9 | 1.4371 | 0.8723 |
57
- | 1.6311 | 2.53 | 12 | 1.3926 | 0.8693 |
 
 
 
 
 
 
 
58
 
59
 
60
  ### Framework versions
 
17
 
18
  This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 0.4068
21
+ - Accuracy: 0.9909
22
 
23
  ## Model description
24
 
 
46
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
  - lr_scheduler_type: linear
48
  - lr_scheduler_warmup_ratio: 0.1
49
+ - num_epochs: 11
50
 
51
  ### Training results
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
+ | No log | 0.84 | 4 | 1.3347 | 0.9027 |
56
+ | No log | 1.89 | 9 | 1.0340 | 0.9605 |
57
+ | 1.2661 | 2.95 | 14 | 0.8152 | 0.9696 |
58
+ | 1.2661 | 4.0 | 19 | 0.6598 | 0.9878 |
59
+ | 0.7865 | 4.84 | 23 | 0.5723 | 0.9878 |
60
+ | 0.7865 | 5.89 | 28 | 0.4964 | 0.9909 |
61
+ | 0.5287 | 6.95 | 33 | 0.4474 | 0.9909 |
62
+ | 0.5287 | 8.0 | 38 | 0.4200 | 0.9939 |
63
+ | 0.4142 | 8.84 | 42 | 0.4085 | 0.9909 |
64
+ | 0.4142 | 9.26 | 44 | 0.4068 | 0.9909 |
65
 
66
 
67
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:607078619d955d7b1f90372e2a6469db6e60d6bf4c9847f5fe1fc959488c5515
3
  size 343236280
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6e0ee9ae2eb6bdb343c6f544325cb3c296b0af182d691470ad46baf5236edba4
3
  size 343236280
runs/Mar18_14-16-36_3d7c35467368/events.out.tfevents.1710771397.3d7c35467368.5146.8 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8b866fa3ddf1c6012cb98a1b4f1b7e3601c38441560150ce28ecfb00930a8106
3
- size 8523
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6bfeafe9e6579a3c8c94495c0303fdb68025abe74394f06d4d1a9db9f25df24a
3
+ size 9188