Hugging Face's logo

Raihan004
/
Hierarchical_Agent_Action

Image Classification
Transformers
TensorBoard
Safetensors
vit
Generated from Trainer
Eval Results
Model card Files
xet
Metrics Community
Raihan004 commited on
Commit
ce69246
·
1 Parent(s): dfafb38

Model save

Browse files
Files changed (5) hide show
  1. README.md +33 -28
  2. config.json +10 -10
  3. model.safetensors +1 -1
  4. runs/Dec27_06-44-35_fc6b8939f5d8/events.out.tfevents.1703659644.fc6b8939f5d8.47.0 +3 -0
  5. training_args.bin +1 -1
README.md CHANGED
@@ -1,8 +1,7 @@
1
  ---
2
  license: apache-2.0
3
- base_model: Raihan004/Hierarchical_Agent_Action
4
  tags:
5
- - image-classification
6
  - generated_from_trainer
7
  datasets:
8
  - image_folder
@@ -15,7 +14,7 @@ model-index:
15
  name: Image Classification
16
  type: image-classification
17
  dataset:
18
- name: agent_action_class
19
  type: image_folder
20
  config: hierarchical-action-agent
21
  split: train
@@ -23,7 +22,7 @@ model-index:
23
  metrics:
24
  - name: Accuracy
25
  type: accuracy
26
- value: 0.7856115107913669
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
  # Hierarchical_Agent_Action
33
 
34
- This model is a fine-tuned version of [Raihan004/Hierarchical_Agent_Action](https://huggingface.co/Raihan004/Hierarchical_Agent_Action) on the agent_action_class dataset.
35
  It achieves the following results on the evaluation set:
36
- - Loss: 0.8558
37
- - Accuracy: 0.7856
38
 
39
  ## Model description
40
 
@@ -53,37 +52,43 @@ More information needed
53
  ### Training hyperparameters
54
 
55
  The following hyperparameters were used during training:
56
- - learning_rate: 0.0002
57
- - train_batch_size: 64
58
  - eval_batch_size: 8
59
  - seed: 42
60
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
61
  - lr_scheduler_type: linear
62
- - num_epochs: 30
63
  - mixed_precision_training: Native AMP
64
 
65
  ### Training results
66
 
67
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
68
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
69
- | 0.2855 | 1.61 | 100 | 0.8656 | 0.7842 |
70
- | 0.2304 | 3.23 | 200 | 0.9937 | 0.7496 |
71
- | 0.3009 | 4.84 | 300 | 1.0167 | 0.7482 |
72
- | 0.2438 | 6.45 | 400 | 1.0789 | 0.7424 |
73
- | 0.222 | 8.06 | 500 | 0.8866 | 0.7698 |
74
- | 0.2174 | 9.68 | 600 | 0.8679 | 0.7799 |
75
- | 0.2177 | 11.29 | 700 | 0.9798 | 0.7655 |
76
- | 0.2471 | 12.9 | 800 | 1.0291 | 0.7727 |
77
- | 0.2782 | 14.52 | 900 | 0.8910 | 0.7799 |
78
- | 0.2149 | 16.13 | 1000 | 0.8712 | 0.7712 |
79
- | 0.2141 | 17.74 | 1100 | 0.9117 | 0.7741 |
80
- | 0.1863 | 19.35 | 1200 | 0.8968 | 0.7770 |
81
- | 0.2194 | 20.97 | 1300 | 0.8561 | 0.7727 |
82
- | 0.1635 | 22.58 | 1400 | 0.8560 | 0.7813 |
83
- | 0.1528 | 24.19 | 1500 | 0.8994 | 0.7885 |
84
- | 0.1416 | 25.81 | 1600 | 0.8558 | 0.7856 |
85
- | 0.1454 | 27.42 | 1700 | 0.8568 | 0.7957 |
86
- | 0.1642 | 29.03 | 1800 | 0.8794 | 0.7914 |
 
 
 
 
 
 
87
 
88
 
89
  ### Framework versions
 
1
  ---
2
  license: apache-2.0
3
+ base_model: google/vit-base-patch16-224-in21k
4
  tags:
 
5
  - generated_from_trainer
6
  datasets:
7
  - image_folder
 
14
  name: Image Classification
15
  type: image-classification
16
  dataset:
17
+ name: image_folder
18
  type: image_folder
19
  config: hierarchical-action-agent
20
  split: train
 
22
  metrics:
23
  - name: Accuracy
24
  type: accuracy
25
+ value: 0.823021582733813
26
  ---
27
 
28
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
30
 
31
  # Hierarchical_Agent_Action
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 image_folder dataset.
34
  It achieves the following results on the evaluation set:
35
+ - Loss: 0.6058
36
+ - Accuracy: 0.8230
37
 
38
  ## Model description
39
 
 
52
  ### Training hyperparameters
53
 
54
  The following hyperparameters were used during training:
55
+ - learning_rate: 0.0001
56
+ - train_batch_size: 32
57
  - eval_batch_size: 8
58
  - seed: 42
59
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
60
  - lr_scheduler_type: linear
61
+ - num_epochs: 20
62
  - mixed_precision_training: Native AMP
63
 
64
  ### Training results
65
 
66
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
67
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
68
+ | 2.4733 | 0.81 | 100 | 2.2958 | 0.6173 |
69
+ | 1.7952 | 1.61 | 200 | 1.6267 | 0.7252 |
70
+ | 1.409 | 2.42 | 300 | 1.2853 | 0.7525 |
71
+ | 1.1321 | 3.23 | 400 | 0.9899 | 0.7971 |
72
+ | 0.9094 | 4.03 | 500 | 0.8828 | 0.8029 |
73
+ | 0.8552 | 4.84 | 600 | 0.8249 | 0.7885 |
74
+ | 0.7082 | 5.65 | 700 | 0.8111 | 0.7755 |
75
+ | 0.5812 | 6.45 | 800 | 0.7087 | 0.8115 |
76
+ | 0.5714 | 7.26 | 900 | 0.6596 | 0.8158 |
77
+ | 0.4802 | 8.06 | 1000 | 0.7486 | 0.7813 |
78
+ | 0.5135 | 8.87 | 1100 | 0.6208 | 0.8259 |
79
+ | 0.4724 | 9.68 | 1200 | 0.6434 | 0.8129 |
80
+ | 0.3781 | 10.48 | 1300 | 0.6290 | 0.8187 |
81
+ | 0.457 | 11.29 | 1400 | 0.6152 | 0.8345 |
82
+ | 0.4174 | 12.1 | 1500 | 0.6348 | 0.8144 |
83
+ | 0.3778 | 12.9 | 1600 | 0.6418 | 0.8158 |
84
+ | 0.3391 | 13.71 | 1700 | 0.6457 | 0.8173 |
85
+ | 0.309 | 14.52 | 1800 | 0.6429 | 0.8101 |
86
+ | 0.3543 | 15.32 | 1900 | 0.6086 | 0.8201 |
87
+ | 0.3232 | 16.13 | 2000 | 0.6375 | 0.8245 |
88
+ | 0.2687 | 16.94 | 2100 | 0.6282 | 0.8273 |
89
+ | 0.2551 | 17.74 | 2200 | 0.6022 | 0.8216 |
90
+ | 0.2711 | 18.55 | 2300 | 0.6056 | 0.8187 |
91
+ | 0.2371 | 19.35 | 2400 | 0.6058 | 0.8230 |
92
 
93
 
94
  ### Framework versions
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "Raihan004/Hierarchical_Agent_Action",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
@@ -11,14 +11,6 @@
11
  "id2label": {
12
  "0": "\u0995\u09c1\u0995\u09c1\u09b0_\u0995\u09ae\u09cd\u09aa\u09bf\u0989\u099f\u09be\u09b0_\u09ac\u09cd\u09af\u09ac\u09b9\u09be\u09b0_\u0995\u09b0\u09be",
13
  "1": "\u0995\u09c1\u0995\u09c1\u09b0_\u0996\u09be\u0993\u09df\u09be",
14
- "2": "\u0995\u09c1\u0995\u09c1\u09b0_\u0996\u09c7\u09b2\u09be_\u0995\u09b0\u09be",
15
- "3": "\u0995\u09c1\u0995\u09c1\u09b0_\u0998\u09c1\u09ae\u09be\u09a8\u09cb",
16
- "4": "\u0995\u09c1\u0995\u09c1\u09b0_\u09aa\u09be\u09a8_\u0995\u09b0\u09be",
17
- "5": "\u0995\u09c1\u0995\u09c1\u09b0_\u09aa\u09dc\u09be",
18
- "6": "\u0995\u09c1\u0995\u09c1\u09b0_\u09b9\u09be\u0981\u099f\u09be",
19
- "7": "\u099b\u09c7\u09b2\u09c7_\u0995\u09a5\u09be_\u09ac\u09b2\u09be",
20
- "8": "\u099b\u09c7\u09b2\u09c7_\u0995\u09ae\u09cd\u09aa\u09bf\u0989\u099f\u09be\u09b0_\u09ac\u09cd\u09af\u09ac\u09b9\u09be\u09b0_\u0995\u09b0\u09be",
21
- "9": "\u099b\u09c7\u09b2\u09c7_\u0996\u09be\u0993\u09df\u09be",
22
  "10": "\u099b\u09c7\u09b2\u09c7_\u0996\u09c7\u09b2\u09be_\u0995\u09b0\u09be",
23
  "11": "\u099b\u09c7\u09b2\u09c7_\u0998\u09c1\u09ae\u09be\u09a8\u09cb",
24
  "12": "\u099b\u09c7\u09b2\u09c7_\u09aa\u09be\u09a8_\u0995\u09b0\u09be",
@@ -29,6 +21,7 @@
29
  "17": "\u09ac\u09bf\u09a1\u09bc\u09be\u09b2_\u0995\u09ae\u09cd\u09aa\u09bf\u0989\u099f\u09be\u09b0_\u09ac\u09cd\u09af\u09ac\u09b9\u09be\u09b0_\u0995\u09b0\u09be",
30
  "18": "\u09ac\u09bf\u09a1\u09bc\u09be\u09b2_\u0996\u09be\u0993\u09df\u09be",
31
  "19": "\u09ac\u09bf\u09a1\u09bc\u09be\u09b2_\u0996\u09c7\u09b2\u09be_\u0995\u09b0\u09be",
 
32
  "20": "\u09ac\u09bf\u09a1\u09bc\u09be\u09b2_\u0998\u09c1\u09ae\u09be\u09a8\u09cb",
33
  "21": "\u09ac\u09bf\u09a1\u09bc\u09be\u09b2_\u09aa\u09be\u09a8_\u0995\u09b0\u09be",
34
  "22": "\u09ac\u09bf\u09a1\u09bc\u09be\u09b2_\u09aa\u09dc\u09be",
@@ -39,10 +32,17 @@
39
  "27": "\u09ae\u09c7\u09af\u09bc\u09c7_\u0996\u09c7\u09b2\u09be_\u0995\u09b0\u09be",
40
  "28": "\u09ae\u09c7\u09af\u09bc\u09c7_\u0998\u09c1\u09ae\u09be\u09a8\u09cb",
41
  "29": "\u09ae\u09c7\u09af\u09bc\u09c7_\u09aa\u09be\u09a8_\u0995\u09b0\u09be",
 
42
  "30": "\u09ae\u09c7\u09af\u09bc\u09c7_\u09aa\u09dc\u09be",
43
  "31": "\u09ae\u09c7\u09af\u09bc\u09c7_\u09b0\u09be\u09a8\u09cd\u09a8\u09be_\u0995\u09b0\u09be",
44
  "32": "\u09ae\u09c7\u09af\u09bc\u09c7_\u09b2\u09c7\u0996\u09be",
45
- "33": "\u09ae\u09c7\u09af\u09bc\u09c7_\u09b9\u09be\u0981\u099f\u09be"
 
 
 
 
 
 
46
  },
47
  "image_size": 224,
48
  "initializer_range": 0.02,
 
1
  {
2
+ "_name_or_path": "google/vit-base-patch16-224-in21k",
3
  "architectures": [
4
  "ViTForImageClassification"
5
  ],
 
11
  "id2label": {
12
  "0": "\u0995\u09c1\u0995\u09c1\u09b0_\u0995\u09ae\u09cd\u09aa\u09bf\u0989\u099f\u09be\u09b0_\u09ac\u09cd\u09af\u09ac\u09b9\u09be\u09b0_\u0995\u09b0\u09be",
13
  "1": "\u0995\u09c1\u0995\u09c1\u09b0_\u0996\u09be\u0993\u09df\u09be",
 
 
 
 
 
 
 
 
14
  "10": "\u099b\u09c7\u09b2\u09c7_\u0996\u09c7\u09b2\u09be_\u0995\u09b0\u09be",
15
  "11": "\u099b\u09c7\u09b2\u09c7_\u0998\u09c1\u09ae\u09be\u09a8\u09cb",
16
  "12": "\u099b\u09c7\u09b2\u09c7_\u09aa\u09be\u09a8_\u0995\u09b0\u09be",
 
21
  "17": "\u09ac\u09bf\u09a1\u09bc\u09be\u09b2_\u0995\u09ae\u09cd\u09aa\u09bf\u0989\u099f\u09be\u09b0_\u09ac\u09cd\u09af\u09ac\u09b9\u09be\u09b0_\u0995\u09b0\u09be",
22
  "18": "\u09ac\u09bf\u09a1\u09bc\u09be\u09b2_\u0996\u09be\u0993\u09df\u09be",
23
  "19": "\u09ac\u09bf\u09a1\u09bc\u09be\u09b2_\u0996\u09c7\u09b2\u09be_\u0995\u09b0\u09be",
24
+ "2": "\u0995\u09c1\u0995\u09c1\u09b0_\u0996\u09c7\u09b2\u09be_\u0995\u09b0\u09be",
25
  "20": "\u09ac\u09bf\u09a1\u09bc\u09be\u09b2_\u0998\u09c1\u09ae\u09be\u09a8\u09cb",
26
  "21": "\u09ac\u09bf\u09a1\u09bc\u09be\u09b2_\u09aa\u09be\u09a8_\u0995\u09b0\u09be",
27
  "22": "\u09ac\u09bf\u09a1\u09bc\u09be\u09b2_\u09aa\u09dc\u09be",
 
32
  "27": "\u09ae\u09c7\u09af\u09bc\u09c7_\u0996\u09c7\u09b2\u09be_\u0995\u09b0\u09be",
33
  "28": "\u09ae\u09c7\u09af\u09bc\u09c7_\u0998\u09c1\u09ae\u09be\u09a8\u09cb",
34
  "29": "\u09ae\u09c7\u09af\u09bc\u09c7_\u09aa\u09be\u09a8_\u0995\u09b0\u09be",
35
+ "3": "\u0995\u09c1\u0995\u09c1\u09b0_\u0998\u09c1\u09ae\u09be\u09a8\u09cb",
36
  "30": "\u09ae\u09c7\u09af\u09bc\u09c7_\u09aa\u09dc\u09be",
37
  "31": "\u09ae\u09c7\u09af\u09bc\u09c7_\u09b0\u09be\u09a8\u09cd\u09a8\u09be_\u0995\u09b0\u09be",
38
  "32": "\u09ae\u09c7\u09af\u09bc\u09c7_\u09b2\u09c7\u0996\u09be",
39
+ "33": "\u09ae\u09c7\u09af\u09bc\u09c7_\u09b9\u09be\u0981\u099f\u09be",
40
+ "4": "\u0995\u09c1\u0995\u09c1\u09b0_\u09aa\u09be\u09a8_\u0995\u09b0\u09be",
41
+ "5": "\u0995\u09c1\u0995\u09c1\u09b0_\u09aa\u09dc\u09be",
42
+ "6": "\u0995\u09c1\u0995\u09c1\u09b0_\u09b9\u09be\u0981\u099f\u09be",
43
+ "7": "\u099b\u09c7\u09b2\u09c7_\u0995\u09a5\u09be_\u09ac\u09b2\u09be",
44
+ "8": "\u099b\u09c7\u09b2\u09c7_\u0995\u09ae\u09cd\u09aa\u09bf\u0989\u099f\u09be\u09b0_\u09ac\u09cd\u09af\u09ac\u09b9\u09be\u09b0_\u0995\u09b0\u09be",
45
+ "9": "\u099b\u09c7\u09b2\u09c7_\u0996\u09be\u0993\u09df\u09be"
46
  },
47
  "image_size": 224,
48
  "initializer_range": 0.02,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3ac0de1708c193b96eced8bc45f10807d72c1f63b0294f24012c13fa15622671
3
  size 343322416
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6a2afa109948b75175e5dc053da4e42436b7200d586e2eb95fcb56fef66185b3
3
  size 343322416
runs/Dec27_06-44-35_fc6b8939f5d8/events.out.tfevents.1703659644.fc6b8939f5d8.47.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e4f03d7d5018d4e7058611f296db419e281d47c8bbdc35532576892f33e01994
3
+ size 37643
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7f9a6ea9a164f2e7c1ded6b3e825f808f2c705f13f96479352292113abd86910
3
  size 4155
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5e7cc738c7078a34dd6a657ae45257e376565f461a356153c6401f80255e2de1
3
  size 4155