vincentiussgk commited on
Commit
7aba2ae
1 Parent(s): 1a529fb

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +23 -8
README.md CHANGED
@@ -3,11 +3,26 @@ license: apache-2.0
3
  base_model: google/vit-base-patch16-224-in21k
4
  tags:
5
  - generated_from_trainer
 
 
6
  metrics:
7
  - accuracy
8
  model-index:
9
  - name: vit-base-patch16-224-in21k-finetuned-eurosat
10
- results: []
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  ---
12
 
13
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -15,10 +30,10 @@ should probably proofread and complete it, then remove this comment. -->
15
 
16
  # vit-base-patch16-224-in21k-finetuned-eurosat
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: nan
21
- - Accuracy: 0.12
22
 
23
  ## Model description
24
 
@@ -52,14 +67,14 @@ The following hyperparameters were used during training:
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
- | No log | 0.97 | 7 | nan | 0.12 |
56
- | 0.0 | 1.93 | 14 | nan | 0.12 |
57
- | 0.0 | 2.9 | 21 | nan | 0.12 |
58
 
59
 
60
  ### Framework versions
61
 
62
  - Transformers 4.31.0
63
  - Pytorch 2.0.1+cu118
64
- - Datasets 2.14.0
65
  - Tokenizers 0.13.3
 
3
  base_model: google/vit-base-patch16-224-in21k
4
  tags:
5
  - generated_from_trainer
6
+ datasets:
7
+ - food101
8
  metrics:
9
  - accuracy
10
  model-index:
11
  - name: vit-base-patch16-224-in21k-finetuned-eurosat
12
+ results:
13
+ - task:
14
+ name: Image Classification
15
+ type: image-classification
16
+ dataset:
17
+ name: food101
18
+ type: food101
19
+ config: default
20
+ split: train[:5000]
21
+ args: default
22
+ metrics:
23
+ - name: Accuracy
24
+ type: accuracy
25
+ value: 0.927
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
30
 
31
  # vit-base-patch16-224-in21k-finetuned-eurosat
32
 
33
+ 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 food101 dataset.
34
  It achieves the following results on the evaluation set:
35
+ - Loss: 1.1055
36
+ - Accuracy: 0.927
37
 
38
  ## Model description
39
 
 
67
 
68
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
69
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
70
+ | 3.0689 | 0.99 | 31 | 2.6415 | 0.82 |
71
+ | 1.6615 | 1.98 | 62 | 1.4504 | 0.898 |
72
+ | 1.1467 | 2.98 | 93 | 1.1055 | 0.927 |
73
 
74
 
75
  ### Framework versions
76
 
77
  - Transformers 4.31.0
78
  - Pytorch 2.0.1+cu118
79
+ - Datasets 2.14.1
80
  - Tokenizers 0.13.3