EngTig
/

food_classifier

Image Classification

generated_from_keras_callback

Inference Endpoints

Model card Files Files and versions Community

EngTig commited on Mar 28

Commit

5dfba4d

•

1 Parent(s): 430c166

End of training

Files changed (2) hide show

README.md +8 -8
tf_model.h5 +1 -1

README.md CHANGED Viewed

@@ -15,9 +15,9 @@ 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:
-- Train Loss: 0.3841
-- Validation Loss: 0.2931
-- Train Accuracy: 0.938
 - Epoch: 4
 ## Model description
@@ -44,11 +44,11 @@ The following hyperparameters were used during training:
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
-| 2.7672     | 1.6351          | 0.817          | 0     |
-| 1.2042     | 0.8144          | 0.896          | 1     |
-| 0.7118     | 0.5195          | 0.909          | 2     |
-| 0.4993     | 0.3816          | 0.927          | 3     |
-| 0.3841     | 0.2931          | 0.938          | 4     |
 ### 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:
+- Train Loss: 0.3706
+- Validation Loss: 0.3269
+- Train Accuracy: 0.915
 - Epoch: 4
 ## Model description
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
+| 2.7588     | 1.6105          | 0.825          | 0     |
+| 1.2020     | 0.7780          | 0.897          | 1     |
+| 0.6908     | 0.5141          | 0.896          | 2     |
+| 0.4975     | 0.4127          | 0.905          | 3     |
+| 0.3706     | 0.3269          | 0.915          | 4     |
 ### Framework versions

tf_model.h5 CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:037498765c877209eb8e69929463022dd87cc0206c21d17491dcebb3fc76daca
 size 343773880

 version https://git-lfs.github.com/spec/v1
+oid sha256:03a7402e16cdfd293db6bf76184db4702be6d87e1eba2b00bcd0ff6e3edc988b
 size 343773880