saffin
/

vit_food101

Image Classification

generated_from_keras_callback

Inference Endpoints

Model card Files Files and versions Community

saffin commited on Apr 16

Commit

522f237

•

1 Parent(s): afeebe6

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.3922
-- Validation Loss: 0.3191
-- Train Accuracy: 0.926
 - Epoch: 4
 ## Model description
@@ -44,11 +44,11 @@ The following hyperparameters were used during training:
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
-| 2.8236     | 1.6373          | 0.82           | 0     |
-| 1.2319     | 0.8169          | 0.891          | 1     |
-| 0.7147     | 0.5474          | 0.901          | 2     |
-| 0.5088     | 0.3837          | 0.919          | 3     |
-| 0.3922     | 0.3191          | 0.926          | 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.3727
+- Validation Loss: 0.3312
+- Train Accuracy: 0.913
 - Epoch: 4
 ## Model description
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
+| 2.8006     | 1.6298          | 0.822          | 0     |
+| 1.2026     | 0.8151          | 0.887          | 1     |
+| 0.6893     | 0.5666          | 0.882          | 2     |
+| 0.4607     | 0.4027          | 0.911          | 3     |
+| 0.3727     | 0.3312          | 0.913          | 4     |
 ### Framework versions

tf_model.h5 CHANGED Viewed

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

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