heado commited on
Commit
7abc132
1 Parent(s): 1922aff
README.md CHANGED
@@ -2,8 +2,9 @@
2
  license: apache-2.0
3
  base_model: google/vit-base-patch16-224-in21k
4
  tags:
5
- - image-classification
6
  - ViT
 
 
7
  - generated_from_trainer
8
  metrics:
9
  - accuracy
@@ -19,7 +20,7 @@ should probably proofread and complete it, then remove this comment. -->
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 beans dataset.
21
  It achieves the following results on the evaluation set:
22
- - Loss: 0.0205
23
  - Accuracy: 1.0
24
 
25
  ## Model description
@@ -52,13 +53,13 @@ The following hyperparameters were used during training:
52
 
53
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
54
  |:-------------:|:------:|:----:|:---------------:|:--------:|
55
- | 0.091 | 1.5385 | 100 | 0.0747 | 0.9774 |
56
- | 0.036 | 3.0769 | 200 | 0.0205 | 1.0 |
57
 
58
 
59
  ### Framework versions
60
 
61
- - Transformers 4.41.2
62
- - Pytorch 2.3.0+cu121
63
  - Datasets 2.20.0
64
  - Tokenizers 0.19.1
 
2
  license: apache-2.0
3
  base_model: google/vit-base-patch16-224-in21k
4
  tags:
 
5
  - ViT
6
+ - HHD
7
+ - image-classification
8
  - generated_from_trainer
9
  metrics:
10
  - accuracy
 
20
 
21
  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 beans dataset.
22
  It achieves the following results on the evaluation set:
23
+ - Loss: 0.0148
24
  - Accuracy: 1.0
25
 
26
  ## Model description
 
53
 
54
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
55
  |:-------------:|:------:|:----:|:---------------:|:--------:|
56
+ | 0.056 | 1.5385 | 100 | 0.0564 | 0.9850 |
57
+ | 0.0375 | 3.0769 | 200 | 0.0148 | 1.0 |
58
 
59
 
60
  ### Framework versions
61
 
62
+ - Transformers 4.42.4
63
+ - Pytorch 2.3.1+cu121
64
  - Datasets 2.20.0
65
  - Tokenizers 0.19.1
config.json CHANGED
@@ -30,5 +30,5 @@
30
  "problem_type": "single_label_classification",
31
  "qkv_bias": true,
32
  "torch_dtype": "float32",
33
- "transformers_version": "4.41.2"
34
  }
 
30
  "problem_type": "single_label_classification",
31
  "qkv_bias": true,
32
  "torch_dtype": "float32",
33
+ "transformers_version": "4.42.4"
34
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ef21ae29af6781b332bf4ae417d3e2c6af87f6dfcb14dccb2bb0a78d6e8925f6
3
  size 343227052
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:09dfe7f83923c7b277fb12e5fb0b0654a5589c52a83414e574fd7da1982231e3
3
  size 343227052
preprocessor_config.json CHANGED
@@ -1,18 +1,4 @@
1
  {
2
- "_valid_processor_keys": [
3
- "images",
4
- "do_resize",
5
- "size",
6
- "resample",
7
- "do_rescale",
8
- "rescale_factor",
9
- "do_normalize",
10
- "image_mean",
11
- "image_std",
12
- "return_tensors",
13
- "data_format",
14
- "input_data_format"
15
- ],
16
  "do_normalize": true,
17
  "do_rescale": true,
18
  "do_resize": true,
 
1
  {
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  "do_normalize": true,
3
  "do_rescale": true,
4
  "do_resize": true,
runs/Jul30_00-50-27_c1a0f3ceb19c/events.out.tfevents.1722301819.c1a0f3ceb19c.179.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5b53c53ccbd99780a4f47840d87b8de84b5377a557aa0d59048c2c8f079e256f
3
+ size 11349
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:87354a911506538377ec10b629caf2f0beaa9bfa97389a2982938b09900d7a15
3
  size 5112
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:76dbed148258d897fc7e99af9307e81dcb5b75de5c390d3bb285c454fd01df4f
3
  size 5112