saffin
/

vit_food101

Image Classification

generated_from_keras_callback

Inference Endpoints

Model card Files Files and versions Community

saffin commited on May 7

Commit

6760ec0

•

1 Parent(s): 39994db

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.3940
-- Validation Loss: 0.3227
-- Train Accuracy: 0.928
 - Epoch: 4
 ## Model description
@@ -44,11 +44,11 @@ The following hyperparameters were used during training:
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
-| 2.7845     | 1.5966          | 0.851          | 0     |
-| 1.1973     | 0.8061          | 0.898          | 1     |
-| 0.6877     | 0.5038          | 0.914          | 2     |
-| 0.4909     | 0.4110          | 0.909          | 3     |
-| 0.3940     | 0.3227          | 0.928          | 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.3828
+- Validation Loss: 0.3210
+- Train Accuracy: 0.922
 - Epoch: 4
 ## Model description
 | 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

tf_model.h5 CHANGED Viewed

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

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