saffin
/

vit_food101

Image Classification

generated_from_keras_callback

Inference Endpoints

Model card Files Files and versions Community

saffin commited on May 8

Commit

b46b0ba

•

1 Parent(s): 6760ec0

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.3828
-- Validation Loss: 0.3210
-- Train Accuracy: 0.922
 - Epoch: 4
 ## Model description
@@ -44,11 +44,11 @@ The following hyperparameters were used during training:
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
-| 2.7414     | 1.6026          | 0.844          | 0     |
-| 1.1868     | 0.7884          | 0.896          | 1     |
-| 0.6536     | 0.5058          | 0.911          | 2     |
-| 0.4733     | 0.3832          | 0.92           | 3     |
-| 0.3828     | 0.3210          | 0.922          | 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.3672
+- Validation Loss: 0.3778
+- Train Accuracy: 0.899
 - Epoch: 4
 ## Model description
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
+| 2.7897     | 1.6408          | 0.818          | 0     |
+| 1.2215     | 0.8019          | 0.895          | 1     |
+| 0.6917     | 0.5261          | 0.898          | 2     |
+| 0.4765     | 0.4082          | 0.905          | 3     |
+| 0.3672     | 0.3778          | 0.899          | 4     |
 ### Framework versions

tf_model.h5 CHANGED Viewed

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

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