🍻 cheers

README.md

@@ -1,6 +1,6 @@
 ---
 license: apache-2.0
-base_model: google/vit-base-patch16-224
+base_model: google/vit-base-patch16-224-in21k
 tags:
 - image-classification
 - generated_from_trainer
@@ -16,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # vit-base-plankton
 
-This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on the beans dataset.
+This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the plankotn dataset.
 
 ## Model description
 
@@ -43,10 +43,6 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 2
 
-### Training results
-
-
-
 ### Framework versions
 
 - Transformers 4.35.2

config.json

@@ -49,7 +49,6 @@
   "num_channels": 3,
   "num_hidden_layers": 12,
   "patch_size": 16,
-  "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
   "transformers_version": "4.35.2"

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:256ee6287ffdf601707d223db0ead870e3d022d40c0de9d83978e3d13c420fdc
+oid sha256:133b10f5e46dbc59da71c447a0bbe6cef3e08afe98eadc74876d5f4d180f2459
 size 343260888

runs/Nov18_14-51-07_e028cef4f67f/events.out.tfevents.1700319079.e028cef4f67f.34901.6

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7ecd5f80dfd3ee89f4632a3d6f4dd23f64fdfb91e0f9d204b16e71628eb2cab4
-size 6727
+oid sha256:8057f80b735c6b829a4667f0f1110c8af78dd88a1805ca047951c5ce53e45de4
+size 11757

training_args.bin

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