Hugging Face's logo

panda992
/
fish_disease_datasets

Image Classification
Transformers
TensorBoard
Safetensors
vit
vision-transformer
aquaculture
fish-disease
Generated from Trainer
Eval Results (legacy)
Model card Files
xet
Metrics Community
panda992 commited on
Commit
55d8909
·
verified ·
1 Parent(s): 13395df

Model save

Browse files
Files changed (5) hide show
  1. README.md +9 -10
  2. config.json +8 -8
  3. model.safetensors +1 -1
  4. runs/May23_06-49-46_1fcbf54ed8fc/events.out.tfevents.1747983009.1fcbf54ed8fc.6395.0 +3 -0
  5. training_args.bin +1 -1
README.md CHANGED
@@ -3,7 +3,6 @@ library_name: transformers
3
  license: apache-2.0
4
  base_model: google/vit-base-patch16-224-in21k
5
  tags:
6
- - image-classification
7
  - generated_from_trainer
8
  metrics:
9
  - accuracy
@@ -17,10 +16,10 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  # fish_disease_datasets
19
 
20
- 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 fish_disease_datasets dataset.
21
  It achieves the following results on the evaluation set:
22
- - Loss: 0.0824
23
- - Accuracy: 0.9783
24
 
25
  ## Model description
26
 
@@ -52,16 +51,16 @@ The following hyperparameters were used during training:
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:------:|:----:|:---------------:|:--------:|
55
- | 0.4914 | 0.7634 | 100 | 0.3797 | 0.9130 |
56
- | 0.0893 | 1.5267 | 200 | 0.1777 | 0.9565 |
57
- | 0.0733 | 2.2901 | 300 | 0.1245 | 0.9620 |
58
- | 0.0384 | 3.0534 | 400 | 0.1020 | 0.9701 |
59
- | 0.0189 | 3.8168 | 500 | 0.0824 | 0.9783 |
60
 
61
 
62
  ### Framework versions
63
 
64
- - Transformers 4.52.2
65
  - Pytorch 2.7.0+cu126
66
  - Datasets 3.6.0
67
  - Tokenizers 0.21.1
 
3
  license: apache-2.0
4
  base_model: google/vit-base-patch16-224-in21k
5
  tags:
 
6
  - generated_from_trainer
7
  metrics:
8
  - accuracy
 
16
 
17
  # fish_disease_datasets
18
 
19
+ 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.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.0704
22
+ - Accuracy: 0.9810
23
 
24
  ## Model description
25
 
 
51
 
52
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
53
  |:-------------:|:------:|:----:|:---------------:|:--------:|
54
+ | 0.4357 | 0.7634 | 100 | 0.3209 | 0.9375 |
55
+ | 0.1859 | 1.5267 | 200 | 0.1352 | 0.9701 |
56
+ | 0.0641 | 2.2901 | 300 | 0.0906 | 0.9783 |
57
+ | 0.0202 | 3.0534 | 400 | 0.1027 | 0.9755 |
58
+ | 0.0352 | 3.8168 | 500 | 0.0704 | 0.9810 |
59
 
60
 
61
  ### Framework versions
62
 
63
+ - Transformers 4.52.3
64
  - Pytorch 2.7.0+cu126
65
  - Datasets 3.6.0
66
  - Tokenizers 0.21.1
config.json CHANGED
@@ -20,13 +20,13 @@
20
  "initializer_range": 0.02,
21
  "intermediate_size": 3072,
22
  "label2id": {
23
- "Bacterial Red disease": "0",
24
- "Bacterial diseases - Aeromoniasis": "1",
25
- "Bacterial gill disease": "2",
26
- "Fungal diseases Saprolegniasis": "3",
27
- "Healthy Fish": "4",
28
- "Parasitic diseases": "5",
29
- "Viral diseases White tail disease": "6"
30
  },
31
  "layer_norm_eps": 1e-12,
32
  "model_type": "vit",
@@ -39,5 +39,5 @@
39
  "problem_type": "single_label_classification",
40
  "qkv_bias": true,
41
  "torch_dtype": "float32",
42
- "transformers_version": "4.52.2"
43
  }
 
20
  "initializer_range": 0.02,
21
  "intermediate_size": 3072,
22
  "label2id": {
23
+ "Bacterial Red disease": 0,
24
+ "Bacterial diseases - Aeromoniasis": 1,
25
+ "Bacterial gill disease": 2,
26
+ "Fungal diseases Saprolegniasis": 3,
27
+ "Healthy Fish": 4,
28
+ "Parasitic diseases": 5,
29
+ "Viral diseases White tail disease": 6
30
  },
31
  "layer_norm_eps": 1e-12,
32
  "model_type": "vit",
 
39
  "problem_type": "single_label_classification",
40
  "qkv_bias": true,
41
  "torch_dtype": "float32",
42
+ "transformers_version": "4.52.3"
43
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3e36ff03fd73310c47992654d930674745460651b87d1899da4fe402c57c4108
3
  size 343239356
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:480754fae42539ba993a7e2f005938d6baab67ba264f10918212d6f25c64541f
3
  size 343239356
runs/May23_06-49-46_1fcbf54ed8fc/events.out.tfevents.1747983009.1fcbf54ed8fc.6395.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5f96c84dd4967f459ec936b01588764241f2cbe48fe85c8e11157214f84451db
3
+ size 18256
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:531f41664913e383cdff9f4102451cc487f15c1f181516357d6889a68649fb6f
3
  size 5713
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1f3b9c14c7030249091db40395aafb65c5edd0828e0cbf47c9a115f5413550f2
3
  size 5713