larimei
/

food-classification-ai

Image Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

larimei commited on May 10

Commit

4ae2ecb

•

1 Parent(s): 51bcb41

End of training

Files changed (1) hide show

README.md +5 -5

README.md CHANGED Viewed

@@ -17,8 +17,8 @@ should probably proofread and complete it, then remove this comment. -->
 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.
 It achieves the following results on the evaluation set:
-- Loss: 1.5937
-- Accuracy: 0.893
 ## Model description
@@ -52,9 +52,9 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 2.7425        | 0.992 | 62   | 2.5168          | 0.833    |
-| 1.8395        | 2.0   | 125  | 1.7492          | 0.886    |
-| 1.6151        | 2.976 | 186  | 1.5937          | 0.893    |
 ### Framework versions

 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.
 It achieves the following results on the evaluation set:
+- Loss: 1.0685
+- Accuracy: 0.7966
 ## Model description
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 1.8963        | 1.0   | 947  | 1.9484          | 0.7290   |
+| 1.2315        | 2.0   | 1894 | 1.2305          | 0.7825   |
+| 0.9834        | 3.0   | 2841 | 1.0685          | 0.7966   |
 ### Framework versions