Hugging Face's logo

xanore
/
results

Image Classification
Transformers
TensorBoard
Safetensors
vit
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
xanore commited on
Commit
c96bd89
•
1 Parent(s): 371c7a8

End of training

Browse files
Files changed (3) hide show
  1. README.md +10 -10
  2. model.safetensors +1 -1
  3. runs/Nov12_13-19-17_DESKTOP-4PF3H2T/events.out.tfevents.1699777157.DESKTOP-4PF3H2T.2684.0 +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: 0.0632
21
- - Accuracy: 0.9788
22
 
23
  ## Model description
24
 
@@ -38,11 +38,11 @@ More information needed
38
 
39
  The following hyperparameters were used during training:
40
  - learning_rate: 0.0002
41
- - train_batch_size: 8
42
- - eval_batch_size: 8
43
  - seed: 1337
44
  - gradient_accumulation_steps: 4
45
- - total_train_batch_size: 32
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
@@ -52,11 +52,11 @@ The following hyperparameters were used during training:
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
- | 0.1085 | 1.0 | 200 | 0.1348 | 0.9525 |
56
- | 0.1069 | 2.0 | 400 | 0.0919 | 0.96 |
57
- | 0.0662 | 3.0 | 600 | 0.0839 | 0.9706 |
58
- | 0.0356 | 4.0 | 800 | 0.0784 | 0.9731 |
59
- | 0.04 | 5.0 | 1000 | 0.0632 | 0.9788 |
60
 
61
 
62
  ### 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.0441
21
+ - Accuracy: 0.987
22
 
23
  ## Model description
24
 
 
38
 
39
  The following hyperparameters were used during training:
40
  - learning_rate: 0.0002
41
+ - train_batch_size: 32
42
+ - eval_batch_size: 32
43
  - seed: 1337
44
  - gradient_accumulation_steps: 4
45
+ - total_train_batch_size: 128
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
 
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
+ | 0.0923 | 0.99 | 62 | 0.0608 | 0.9805 |
56
+ | 0.0594 | 2.0 | 125 | 0.0612 | 0.978 |
57
+ | 0.0596 | 2.99 | 187 | 0.0538 | 0.98 |
58
+ | 0.0378 | 4.0 | 250 | 0.0440 | 0.9825 |
59
+ | 0.0246 | 4.96 | 310 | 0.0441 | 0.987 |
60
 
61
 
62
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4b19d84062a79e9874831a73c0c3c3d26e1fbd32fb7ebf994e921ec8348ffd32
3
  size 343223968
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5ce15c6e9706aab48c09dd63cd6f1e41012d0ea1294df7ee5f5c287f9aed27bf
3
  size 343223968
runs/Nov12_13-19-17_DESKTOP-4PF3H2T/events.out.tfevents.1699777157.DESKTOP-4PF3H2T.2684.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:77e4c0c85b08123d3fd1aae0ae43c2d7307f44523a5539a73be94982c43a6118
3
- size 9352
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:16abc750d7ed0d8face87b7b1824727b58b3f02dbc0f22dab465f038cbe1914e
3
+ size 10971