Schram03 commited on
Commit
6efe5e6
·
verified ·
1 Parent(s): f541eee

Model save

Browse files
README.md CHANGED
@@ -1,27 +1,52 @@
1
  ---
 
2
  license: apache-2.0
 
 
 
 
 
3
  metrics:
4
  - accuracy
5
- base_model:
6
- - google/vit-base-patch16-224
7
- library_name: transformers
8
- tags:
9
- - image-classification
 
 
 
 
 
 
 
 
 
 
 
10
  ---
11
 
12
- # oxford-pets-zero-shot
 
13
 
14
- This model achieves the following results:
15
- - Accuracy: 0.8800
16
- - Precision: 0.8768
17
- - Recall: 0.8800
18
 
19
- # vit-base-oxford-iiit-pets (Transfer learning)
20
-
21
- 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.
22
  It achieves the following results on the evaluation set:
23
- - Loss: 0.1814
24
- - Accuracy: 0.9499
 
 
 
 
 
 
 
 
 
 
 
 
25
 
26
  ## Training procedure
27
 
@@ -40,8 +65,16 @@ The following hyperparameters were used during training:
40
 
41
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
42
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
43
- | 0.3645 | 1.0 | 370 | 0.3259 | 0.9296 |
44
- | 0.2118 | 2.0 | 740 | 0.2660 | 0.9350 |
45
- | 0.1643 | 3.0 | 1110 | 0.2436 | 0.9323 |
46
- | 0.1482 | 4.0 | 1480 | 0.2364 | 0.9364 |
47
- | 0.1412 | 5.0 | 1850 | 0.2357 | 0.9350 |
 
 
 
 
 
 
 
 
 
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
+ datasets:
8
+ - imagefolder
9
  metrics:
10
  - accuracy
11
+ model-index:
12
+ - name: fruits-classification
13
+ results:
14
+ - task:
15
+ name: Image Classification
16
+ type: image-classification
17
+ dataset:
18
+ name: imagefolder
19
+ type: imagefolder
20
+ config: default
21
+ split: train
22
+ args: default
23
+ metrics:
24
+ - name: Accuracy
25
+ type: accuracy
26
+ value: 1.0
27
  ---
28
 
29
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
30
+ should probably proofread and complete it, then remove this comment. -->
31
 
32
+ # fruits-classification
 
 
 
33
 
34
+ This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on the imagefolder dataset.
 
 
35
  It achieves the following results on the evaluation set:
36
+ - Loss: 0.0007
37
+ - Accuracy: 1.0
38
+
39
+ ## Model description
40
+
41
+ More information needed
42
+
43
+ ## Intended uses & limitations
44
+
45
+ More information needed
46
+
47
+ ## Training and evaluation data
48
+
49
+ More information needed
50
 
51
  ## Training procedure
52
 
 
65
 
66
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
67
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
68
+ | 0.0063 | 1.0 | 1070 | 0.0053 | 1.0 |
69
+ | 0.0023 | 2.0 | 2140 | 0.0020 | 1.0 |
70
+ | 0.0012 | 3.0 | 3210 | 0.0011 | 1.0 |
71
+ | 0.001 | 4.0 | 4280 | 0.0008 | 1.0 |
72
+ | 0.0007 | 5.0 | 5350 | 0.0007 | 1.0 |
73
+
74
+
75
+ ### Framework versions
76
+
77
+ - Transformers 4.52.3
78
+ - Pytorch 2.7.0+cu128
79
+ - Datasets 3.6.0
80
+ - Tokenizers 0.21.1
runs/May30_09-34-57_ip-10-192-12-141/events.out.tfevents.1748598852.ip-10-192-12-141.92810.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:81c31b8fa83b7c319c3555a82f5794b8b84ee1cbb9b0339da9b9b2dec168e8ce
3
+ size 411