hsinyen5 commited on
Commit
aca46b9
·
verified ·
1 Parent(s): efbc349

Model save

Browse files
README.md CHANGED
@@ -3,7 +3,6 @@ library_name: transformers
3
  license: apache-2.0
4
  base_model: google/vit-base-patch16-224
5
  tags:
6
- - image-classification
7
  - generated_from_trainer
8
  metrics:
9
  - accuracy
@@ -17,10 +16,10 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  # vit-base-oxford-iiit-pets
19
 
20
- This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on the pcuenq/oxford-pets dataset.
21
  It achieves the following results on the evaluation set:
22
- - Loss: 0.2277
23
- - Accuracy: 0.9323
24
 
25
  ## Model description
26
 
@@ -51,16 +50,16 @@ The following hyperparameters were used during training:
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
53
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
54
- | 0.3834 | 1.0 | 370 | 0.2994 | 0.9269 |
55
- | 0.207 | 2.0 | 740 | 0.2263 | 0.9364 |
56
- | 0.1794 | 3.0 | 1110 | 0.2058 | 0.9445 |
57
- | 0.143 | 4.0 | 1480 | 0.1970 | 0.9432 |
58
- | 0.1244 | 5.0 | 1850 | 0.1962 | 0.9445 |
59
 
60
 
61
  ### Framework versions
62
 
63
  - Transformers 4.48.3
64
- - Pytorch 2.5.1+cu124
65
- - Datasets 3.3.2
66
  - Tokenizers 0.21.0
 
3
  license: apache-2.0
4
  base_model: google/vit-base-patch16-224
5
  tags:
 
6
  - generated_from_trainer
7
  metrics:
8
  - accuracy
 
16
 
17
  # vit-base-oxford-iiit-pets
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: 0.2073
22
+ - Accuracy: 0.9296
23
 
24
  ## Model description
25
 
 
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
52
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
53
+ | 0.3825 | 1.0 | 370 | 0.2863 | 0.9269 |
54
+ | 0.2196 | 2.0 | 740 | 0.2195 | 0.9310 |
55
+ | 0.1563 | 3.0 | 1110 | 0.1977 | 0.9364 |
56
+ | 0.1598 | 4.0 | 1480 | 0.1906 | 0.9364 |
57
+ | 0.1444 | 5.0 | 1850 | 0.1876 | 0.9378 |
58
 
59
 
60
  ### Framework versions
61
 
62
  - Transformers 4.48.3
63
+ - Pytorch 2.6.0+cu124
64
+ - Datasets 3.4.0
65
  - Tokenizers 0.21.0
runs/Mar17_08-05-00_51f30ed1a00e/events.out.tfevents.1742199642.51f30ed1a00e.564.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d89f8f8fc8818f6ebd5c093594d341884cf110e573941a772e3a88689895c234
3
+ size 411