Raihan004 commited on
Commit
ceff2ec
1 Parent(s): 5b5e6cd

Model save

Browse files
README.md CHANGED
@@ -2,7 +2,6 @@
2
  license: apache-2.0
3
  base_model: google/vit-base-patch16-224-in21k
4
  tags:
5
- - image-classification
6
  - generated_from_trainer
7
  datasets:
8
  - imagefolder
@@ -15,7 +14,7 @@ model-index:
15
  name: Image Classification
16
  type: image-classification
17
  dataset:
18
- name: action_class
19
  type: imagefolder
20
  config: default
21
  split: train
@@ -23,7 +22,7 @@ model-index:
23
  metrics:
24
  - name: Accuracy
25
  type: accuracy
26
- value: 0.8576449912126538
27
  ---
28
 
29
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -31,10 +30,10 @@ should probably proofread and complete it, then remove this comment. -->
31
 
32
  # Action_model
33
 
34
- 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 action_class dataset.
35
  It achieves the following results on the evaluation set:
36
- - Loss: 0.4589
37
- - Accuracy: 0.8576
38
 
39
  ## Model description
40
 
@@ -59,44 +58,23 @@ The following hyperparameters were used during training:
59
  - seed: 42
60
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
61
  - lr_scheduler_type: linear
62
- - num_epochs: 10
63
  - mixed_precision_training: Native AMP
64
 
65
  ### Training results
66
 
67
- | Training Loss | Epoch | Step | Validation Loss | Accuracy |
68
- |:-------------:|:-----:|:----:|:---------------:|:--------:|
69
- | 1.2754 | 0.37 | 100 | 1.1163 | 0.7329 |
70
- | 0.9345 | 0.75 | 200 | 0.8296 | 0.7996 |
71
- | 0.8816 | 1.12 | 300 | 0.7156 | 0.8102 |
72
- | 0.7425 | 1.49 | 400 | 0.6529 | 0.8067 |
73
- | 0.6883 | 1.87 | 500 | 0.6079 | 0.8243 |
74
- | 0.5454 | 2.24 | 600 | 0.5605 | 0.8348 |
75
- | 0.5383 | 2.61 | 700 | 0.5571 | 0.8295 |
76
- | 0.5442 | 2.99 | 800 | 0.5864 | 0.8190 |
77
- | 0.3986 | 3.36 | 900 | 0.5632 | 0.8313 |
78
- | 0.3438 | 3.73 | 1000 | 0.5606 | 0.8366 |
79
- | 0.4345 | 4.1 | 1100 | 0.5354 | 0.8366 |
80
- | 0.4523 | 4.48 | 1200 | 0.4988 | 0.8576 |
81
- | 0.3162 | 4.85 | 1300 | 0.5099 | 0.8541 |
82
- | 0.3793 | 5.22 | 1400 | 0.5190 | 0.8436 |
83
- | 0.3228 | 5.6 | 1500 | 0.4589 | 0.8576 |
84
- | 0.1795 | 5.97 | 1600 | 0.5096 | 0.8489 |
85
- | 0.2626 | 6.34 | 1700 | 0.5403 | 0.8489 |
86
- | 0.3041 | 6.72 | 1800 | 0.4908 | 0.8489 |
87
- | 0.1831 | 7.09 | 1900 | 0.5721 | 0.8383 |
88
- | 0.2275 | 7.46 | 2000 | 0.5349 | 0.8313 |
89
- | 0.1762 | 7.84 | 2100 | 0.5204 | 0.8541 |
90
- | 0.2112 | 8.21 | 2200 | 0.5189 | 0.8629 |
91
- | 0.1242 | 8.58 | 2300 | 0.5377 | 0.8471 |
92
- | 0.1207 | 8.96 | 2400 | 0.5325 | 0.8559 |
93
- | 0.1806 | 9.33 | 2500 | 0.5150 | 0.8647 |
94
- | 0.1793 | 9.7 | 2600 | 0.5153 | 0.8664 |
95
 
96
 
97
  ### Framework versions
98
 
99
- - Transformers 4.39.3
100
  - Pytorch 2.1.2
101
- - Datasets 2.18.0
102
- - Tokenizers 0.15.2
 
2
  license: apache-2.0
3
  base_model: google/vit-base-patch16-224-in21k
4
  tags:
 
5
  - generated_from_trainer
6
  datasets:
7
  - imagefolder
 
14
  name: Image Classification
15
  type: image-classification
16
  dataset:
17
+ name: imagefolder
18
  type: imagefolder
19
  config: default
20
  split: train
 
22
  metrics:
23
  - name: Accuracy
24
  type: accuracy
25
+ value: 0.843585237258348
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
30
 
31
  # Action_model
32
 
33
+ 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 imagefolder dataset.
34
  It achieves the following results on the evaluation set:
35
+ - Loss: 0.6087
36
+ - Accuracy: 0.8436
37
 
38
  ## Model description
39
 
 
58
  - seed: 42
59
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
60
  - lr_scheduler_type: linear
61
+ - num_epochs: 2
62
  - mixed_precision_training: Native AMP
63
 
64
  ### Training results
65
 
66
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy |
67
+ |:-------------:|:------:|:----:|:---------------:|:--------:|
68
+ | 1.2783 | 0.3731 | 100 | 1.2065 | 0.7153 |
69
+ | 0.9907 | 0.7463 | 200 | 0.8331 | 0.8102 |
70
+ | 0.8428 | 1.1194 | 300 | 0.7278 | 0.8260 |
71
+ | 0.7442 | 1.4925 | 400 | 0.6576 | 0.8172 |
72
+ | 0.6749 | 1.8657 | 500 | 0.6087 | 0.8436 |
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
73
 
74
 
75
  ### Framework versions
76
 
77
+ - Transformers 4.41.1
78
  - Pytorch 2.1.2
79
+ - Datasets 2.19.1
80
+ - Tokenizers 0.19.1
config.json CHANGED
@@ -44,5 +44,5 @@
44
  "problem_type": "single_label_classification",
45
  "qkv_bias": true,
46
  "torch_dtype": "float32",
47
- "transformers_version": "4.39.3"
48
  }
 
44
  "problem_type": "single_label_classification",
45
  "qkv_bias": true,
46
  "torch_dtype": "float32",
47
+ "transformers_version": "4.41.1"
48
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8b0a199ad74ff65ea529793bc6fdbef69fb6f7b70ddf2146d94e4c125b807dd2
3
  size 343248584
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f500fee33cf67c06f93219817b2deb712e53e414d594bd9408c3b2249c73814d
3
  size 343248584
runs/Jun01_14-51-37_6c3c2d6efe19/events.out.tfevents.1717256949.6c3c2d6efe19.34.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0cfef251604163ab8f40037335c74848e16fac0f811217caa5d157bd7204d626
3
+ size 18913
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3c8c7d4932a995756b5081ceec90e99008b51ea7166830884a36b9004ee45428
3
- size 4920
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0fb77d5f97f2e5ab7fa7e8770b92c00a16f8ca99d73dd63690a3e5b26e5212e6
3
+ size 5048