Rebecasarai commited on
Commit
309082d
1 Parent(s): fa8e4d6

End of training

Browse files
Files changed (1) hide show
  1. README.md +13 -0
README.md CHANGED
@@ -3,6 +3,8 @@ license: apache-2.0
3
  base_model: google/vit-base-patch16-224-in21k
4
  tags:
5
  - generated_from_trainer
 
 
6
  model-index:
7
  - name: my_awesome_food_model
8
  results: []
@@ -14,6 +16,9 @@ should probably proofread and complete it, then remove this comment. -->
14
  # my_awesome_food_model
15
 
16
  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.
 
 
 
17
 
18
  ## Model description
19
 
@@ -43,6 +48,14 @@ The following hyperparameters were used during training:
43
  - lr_scheduler_warmup_ratio: 0.1
44
  - num_epochs: 2
45
 
 
 
 
 
 
 
 
 
46
  ### Framework versions
47
 
48
  - Transformers 4.41.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: my_awesome_food_model
10
  results: []
 
16
  # my_awesome_food_model
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: 2.9031
21
+ - Accuracy: 0.815
22
 
23
  ## Model description
24
 
 
48
  - lr_scheduler_warmup_ratio: 0.1
49
  - num_epochs: 2
50
 
51
+ ### Training results
52
+
53
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|
55
+ | 3.4887 | 0.992 | 31 | 3.2930 | 0.798 |
56
+ | 2.9076 | 1.984 | 62 | 2.9031 | 0.815 |
57
+
58
+
59
  ### Framework versions
60
 
61
  - Transformers 4.41.0