porkchopEnjoyer
/

food_classifier

Image Classification

generated_from_keras_callback

Inference Endpoints

Model card Files Files and versions Community

porkchopEnjoyer commited on Jan 20

Commit

a3e835a

•

1 Parent(s): 0a7dc3e

End of training

Files changed (2) hide show

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

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.3744
-- Validation Loss: 0.3587
-- Train Accuracy: 0.911
 - Epoch: 4
 ## Model description
@@ -44,11 +44,11 @@ The following hyperparameters were used during training:
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
-| 2.7892     | 1.6801          | 0.793          | 0     |
-| 1.2251     | 0.8180          | 0.897          | 1     |
-| 0.6916     | 0.5436          | 0.891          | 2     |
-| 0.4950     | 0.3739          | 0.924          | 3     |
-| 0.3744     | 0.3587          | 0.911          | 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.3909
+- Validation Loss: 0.3604
+- Train Accuracy: 0.908
 - Epoch: 4
 ## Model description
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
+| 2.8110     | 1.6545          | 0.805          | 0     |
+| 1.2528     | 0.8188          | 0.893          | 1     |
+| 0.7347     | 0.5696          | 0.891          | 2     |
+| 0.5125     | 0.3828          | 0.934          | 3     |
+| 0.3909     | 0.3604          | 0.908          | 4     |
 ### Framework versions

tf_model.h5 CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:239cc4d326375885b8613998b6715055a08250b1e485957997fe91da5893ac85
+size 343790264