thuyentruong commited on
Commit
75b7b5f
1 Parent(s): bcfe5b6

End of training

Browse files
Files changed (1) hide show
  1. README.md +8 -8
README.md CHANGED
@@ -8,7 +8,7 @@ datasets:
8
  metrics:
9
  - accuracy
10
  model-index:
11
- - name: google/vit-base-patch16-224-in21k-finetuned
12
  results:
13
  - task:
14
  name: Image Classification
@@ -19,18 +19,18 @@ model-index:
19
  metrics:
20
  - name: Accuracy
21
  type: accuracy
22
- value: 0.893
23
  ---
24
 
25
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
26
  should probably proofread and complete it, then remove this comment. -->
27
 
28
- # google/vit-base-patch16-224-in21k-finetuned
29
 
30
  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.
31
  It achieves the following results on the evaluation set:
32
- - Loss: 1.5688
33
- - Accuracy: 0.893
34
 
35
  ## Model description
36
 
@@ -65,9 +65,9 @@ The following hyperparameters were used during training:
65
 
66
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
67
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
68
- | 2.6743 | 0.992 | 62 | 2.5147 | 0.83 |
69
- | 1.8116 | 2.0 | 125 | 1.7366 | 0.863 |
70
- | 1.5732 | 2.976 | 186 | 1.5688 | 0.893 |
71
 
72
 
73
  ### Framework versions
 
8
  metrics:
9
  - accuracy
10
  model-index:
11
+ - name: google/vit-base-patch16-224-in21k-v2-finetuned
12
  results:
13
  - task:
14
  name: Image Classification
 
19
  metrics:
20
  - name: Accuracy
21
  type: accuracy
22
+ value: 0.7968976897689769
23
  ---
24
 
25
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
26
  should probably proofread and complete it, then remove this comment. -->
27
 
28
+ # google/vit-base-patch16-224-in21k-v2-finetuned
29
 
30
  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.
31
  It achieves the following results on the evaluation set:
32
+ - Loss: 1.0612
33
+ - Accuracy: 0.7969
34
 
35
  ## Model description
36
 
 
65
 
66
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
67
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
68
+ | 1.9201 | 1.0 | 947 | 1.9632 | 0.7297 |
69
+ | 1.2002 | 2.0 | 1894 | 1.2327 | 0.7805 |
70
+ | 0.9561 | 3.0 | 2841 | 1.0612 | 0.7969 |
71
 
72
 
73
  ### Framework versions