Raihan004 commited on
Commit
d328d69
1 Parent(s): 008f4bc

Model save

Browse files
README.md CHANGED
@@ -1,8 +1,7 @@
1
  ---
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.789103690685413
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: 1.1399
37
- - Accuracy: 0.7891
38
 
39
  ## Model description
40
 
@@ -59,14 +58,26 @@ 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: 1
63
  - mixed_precision_training: Native AMP
64
 
65
  ### Training results
66
 
67
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
68
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
69
- | 1.2948 | 0.75 | 100 | 1.1399 | 0.7891 |
 
 
 
 
 
 
 
 
 
 
 
 
70
 
71
 
72
  ### Framework versions
 
1
  ---
2
  license: apache-2.0
3
+ base_model: Raihan004/Action_model
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.8488576449912126
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 [Raihan004/Action_model](https://huggingface.co/Raihan004/Action_model) on the imagefolder dataset.
34
  It achieves the following results on the evaluation set:
35
+ - Loss: 0.5034
36
+ - Accuracy: 0.8489
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: 10
62
  - mixed_precision_training: Native AMP
63
 
64
  ### Training results
65
 
66
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
67
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
68
+ | 0.7766 | 0.75 | 100 | 0.6780 | 0.8225 |
69
+ | 0.61 | 1.49 | 200 | 0.6279 | 0.8243 |
70
+ | 0.4734 | 2.24 | 300 | 0.5593 | 0.8278 |
71
+ | 0.5275 | 2.99 | 400 | 0.5148 | 0.8418 |
72
+ | 0.3767 | 3.73 | 500 | 0.5129 | 0.8436 |
73
+ | 0.3207 | 4.48 | 600 | 0.4966 | 0.8559 |
74
+ | 0.3155 | 5.22 | 700 | 0.5251 | 0.8453 |
75
+ | 0.2565 | 5.97 | 800 | 0.4790 | 0.8629 |
76
+ | 0.2791 | 6.72 | 900 | 0.5111 | 0.8524 |
77
+ | 0.1987 | 7.46 | 1000 | 0.5002 | 0.8453 |
78
+ | 0.2083 | 8.21 | 1100 | 0.5034 | 0.8629 |
79
+ | 0.2567 | 8.96 | 1200 | 0.4995 | 0.8576 |
80
+ | 0.2127 | 9.7 | 1300 | 0.5034 | 0.8489 |
81
 
82
 
83
  ### Framework versions
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "google/vit-base-patch16-224-in21k",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
 
1
  {
2
+ "_name_or_path": "Raihan004/Action_model",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:b07fa61508c7a346eaf09b62a85ae114960c850a6920d0fc26dd67faa7cbf714
3
  size 343248584
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:68d775d871b3b92e9f0e531c76b86b7e975139d8a195c23da80decccc0195047
3
  size 343248584
runs/May07_17-56-54_6cf576645707/events.out.tfevents.1715104615.6cf576645707.35.2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6d086b9d857cf192bb6c6bbad6093bc993fd5118d858eedf40b39e12d575e876
3
+ size 24345
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:148ee2272eda97b80560990604b5089c5ea7630b4eff20261a46564c1afa1241
3
  size 4920
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c3eff6ad79023ddf6c472f11a8acfce8080bad51a69350c3daf11b8ce31f6b31
3
  size 4920