Model save

README.md

@@ -1,9 +1,8 @@
 ---
 library_name: transformers
 license: apache-2.0
-base_model: panda992/fish_disease_datasets
+base_model: google/vit-base-patch16-224-in21k
 tags:
-- image-classification
 - generated_from_trainer
 metrics:
 - accuracy
@@ -17,10 +16,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # fish_disease_datasets
 
-This model is a fine-tuned version of [panda992/fish_disease_datasets](https://huggingface.co/panda992/fish_disease_datasets) on the fish_disease_datasets 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 an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0675
-- Accuracy: 0.9810
+- Loss: 0.0643
+- Accuracy: 0.9864
 
 ## Model description
 
@@ -52,15 +51,16 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch  | Step | Validation Loss | Accuracy |
 |:-------------:|:------:|:----:|:---------------:|:--------:|
-| 0.2052        | 0.7634 | 100  | 0.1684          | 0.9484   |
-| 0.05          | 1.5267 | 200  | 0.1600          | 0.9592   |
-| 0.0579        | 2.2901 | 300  | 0.1619          | 0.9511   |
-| 0.1181        | 3.0534 | 400  | 0.0685          | 0.9810   |
-| 0.1321        | 3.8168 | 500  | 0.0675          | 0.9810   |
+| 0.3988        | 0.7634 | 100  | 0.3312          | 0.9457   |
+| 0.1719        | 1.5267 | 200  | 0.1810          | 0.9620   |
+| 0.0527        | 2.2901 | 300  | 0.1292          | 0.9647   |
+| 0.0559        | 3.0534 | 400  | 0.0772          | 0.9810   |
+| 0.0413        | 3.8168 | 500  | 0.0643          | 0.9864   |
 
 
 ### Framework versions
 
-- Transformers 4.51.3
-- Pytorch 2.6.0+cu124
+- Transformers 4.52.2
+- Pytorch 2.7.0+cu126
+- Datasets 3.6.0
 - Tokenizers 0.21.1

config.json

@@ -39,5 +39,5 @@
   "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.51.3"
+  "transformers_version": "4.52.2"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8eb4da8fc8334930d10845fd42541e8b2b928295354e1b0cf5434914f9df107e
+oid sha256:328a6a36c9c92fb460cb5a7f2f419757b0df59a180eb55d1b353eb3b0b7b3f23
 size 343239356

runs/May22_05-46-42_32dbd7354e1a/events.out.tfevents.1747892831.32dbd7354e1a.24930.0

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

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:031c314868105b9b7649e9338a226e1df2bf6d6fb42c74956cd274436f1f7a7e
-size 5304
+oid sha256:ab9d66d498244eda317350d5014a78451f1c79b097267b0ad2c7f25ae9eea20a
+size 5713