CVPROJ25
/

FINETUNED

Image Classification

Generated from Trainer

Model card Files Files and versions

hanngao commited on Mar 7

Commit

e96cfc5

·

verified ·

1 Parent(s): 150bb1d

Training complete

Files changed (2) hide show

README.md +5 -5
model.safetensors +1 -1

README.md CHANGED Viewed

@@ -18,8 +18,8 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.8523
-- Accuracy: 0.785
 ## Model description
@@ -51,9 +51,9 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.7260        | 1.0   | 16   | 0.8205          | 0.789    |
-| 0.3611        | 2.0   | 32   | 0.8372          | 0.784    |
-| 0.2573        | 3.0   | 48   | 0.8523          | 0.785    |
 ### Framework versions

 This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on an unknown dataset.
 It achieves the following results on the evaluation set:
+- Loss: 0.9199
+- Accuracy: 0.774
 ## Model description
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 0.8241        | 1.0   | 16   | 0.8922          | 0.773    |
+| 0.3997        | 2.0   | 32   | 0.9019          | 0.7705   |
+| 0.2905        | 3.0   | 48   | 0.9199          | 0.774    |
 ### Framework versions

model.safetensors CHANGED Viewed

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

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