vikas117 commited on
Commit
0b0bc90
·
verified ·
1 Parent(s): 22a0e21

Model save

Browse files
README.md CHANGED
@@ -1,7 +1,7 @@
1
  ---
2
  library_name: transformers
3
  license: apache-2.0
4
- base_model: microsoft/beit-base-patch16-224-pt22k-ft22k
5
  tags:
6
  - generated_from_trainer
7
  datasets:
@@ -23,7 +23,7 @@ model-index:
23
  metrics:
24
  - name: Accuracy
25
  type: accuracy
26
- value: 0.9586776859504132
27
  ---
28
 
29
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -31,10 +31,10 @@ should probably proofread and complete it, then remove this comment. -->
31
 
32
  # finetuned-ai-real-beit
33
 
34
- This model is a fine-tuned version of [microsoft/beit-base-patch16-224-pt22k-ft22k](https://huggingface.co/microsoft/beit-base-patch16-224-pt22k-ft22k) on the imagefolder dataset.
35
  It achieves the following results on the evaluation set:
36
- - Loss: 0.1340
37
- - Accuracy: 0.9587
38
 
39
  ## Model description
40
 
@@ -59,15 +59,17 @@ The following hyperparameters were used during training:
59
  - seed: 42
60
  - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
61
  - lr_scheduler_type: linear
62
- - num_epochs: 5
63
  - mixed_precision_training: Native AMP
64
 
65
  ### Training results
66
 
67
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
68
  |:-------------:|:------:|:----:|:---------------:|:--------:|
69
- | 0.0724 | 2.2727 | 50 | 0.0964 | 0.9669 |
70
- | 0.0198 | 4.5455 | 100 | 0.1340 | 0.9587 |
 
 
71
 
72
 
73
  ### Framework versions
 
1
  ---
2
  library_name: transformers
3
  license: apache-2.0
4
+ base_model: vikas117/finetuned-ai-real-beit
5
  tags:
6
  - generated_from_trainer
7
  datasets:
 
23
  metrics:
24
  - name: Accuracy
25
  type: accuracy
26
+ value: 0.9338842975206612
27
  ---
28
 
29
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
31
 
32
  # finetuned-ai-real-beit
33
 
34
+ This model is a fine-tuned version of [vikas117/finetuned-ai-real-beit](https://huggingface.co/vikas117/finetuned-ai-real-beit) on the imagefolder dataset.
35
  It achieves the following results on the evaluation set:
36
+ - Loss: 0.2805
37
+ - Accuracy: 0.9339
38
 
39
  ## Model description
40
 
 
59
  - seed: 42
60
  - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
61
  - lr_scheduler_type: linear
62
+ - num_epochs: 2
63
  - mixed_precision_training: Native AMP
64
 
65
  ### Training results
66
 
67
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
68
  |:-------------:|:------:|:----:|:---------------:|:--------:|
69
+ | 0.4366 | 0.4545 | 10 | 0.0494 | 0.9752 |
70
+ | 0.0713 | 0.9091 | 20 | 0.1101 | 0.9587 |
71
+ | 0.0302 | 1.3636 | 30 | 0.2225 | 0.9587 |
72
+ | 0.0531 | 1.8182 | 40 | 0.2805 | 0.9339 |
73
 
74
 
75
  ### Framework versions
all_results.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 5.0,
3
+ "eval_accuracy": 0.9586776859504132,
4
+ "eval_loss": 0.13598860800266266,
5
+ "eval_runtime": 1.6577,
6
+ "eval_samples_per_second": 72.995,
7
+ "eval_steps_per_second": 9.652,
8
+ "total_flos": 2.633601533681664e+17,
9
+ "train_loss": 0.08520049764351412,
10
+ "train_runtime": 91.5555,
11
+ "train_samples_per_second": 37.136,
12
+ "train_steps_per_second": 1.201
13
+ }
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "microsoft/beit-base-patch16-224-pt22k-ft22k",
3
  "add_fpn": false,
4
  "architectures": [
5
  "BeitForImageClassification"
 
1
  {
2
+ "_name_or_path": "vikas117/finetuned-ai-real-beit",
3
  "add_fpn": false,
4
  "architectures": [
5
  "BeitForImageClassification"
eval_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 5.0,
3
+ "eval_accuracy": 0.9586776859504132,
4
+ "eval_loss": 0.13598860800266266,
5
+ "eval_runtime": 1.6577,
6
+ "eval_samples_per_second": 72.995,
7
+ "eval_steps_per_second": 9.652
8
+ }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7ad18403a2ad5373fe340f4c9ff19b38b3570be7aa1953761ae3b9bdc38643e5
3
  size 343080328
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:de5a6af65005273dfbc1c13fe92a3b73f2e0224aa4c89628570d94a8c26fd805
3
  size 343080328
preprocessor_config.json CHANGED
@@ -1,8 +1,11 @@
1
  {
2
- "crop_size": 224,
 
 
 
3
  "do_center_crop": false,
4
- "do_convert_rgb": null,
5
  "do_normalize": true,
 
6
  "do_rescale": true,
7
  "do_resize": true,
8
  "image_mean": [
@@ -10,7 +13,7 @@
10
  0.5,
11
  0.5
12
  ],
13
- "image_processor_type": "ViTFeatureExtractor",
14
  "image_std": [
15
  0.5,
16
  0.5,
 
1
  {
2
+ "crop_size": {
3
+ "height": 224,
4
+ "width": 224
5
+ },
6
  "do_center_crop": false,
 
7
  "do_normalize": true,
8
+ "do_reduce_labels": false,
9
  "do_rescale": true,
10
  "do_resize": true,
11
  "image_mean": [
 
13
  0.5,
14
  0.5
15
  ],
16
+ "image_processor_type": "BeitFeatureExtractor",
17
  "image_std": [
18
  0.5,
19
  0.5,
runs/Dec30_15-52-44_01976ac788fd/events.out.tfevents.1735574112.01976ac788fd.1392.11 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e5bc48f222058aa413376df93d28111efdaaf510c1ab8cd97631d0d4fa1277ea
3
+ size 405
runs/Dec30_16-01-36_01976ac788fd/events.out.tfevents.1735574504.01976ac788fd.1392.12 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8fdfdf3f5c411cd148b855fa5ecc91fc9350af0252dcde8cb2f58a8ccac9961e
3
+ size 8256
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 5.0,
3
+ "total_flos": 2.633601533681664e+17,
4
+ "train_loss": 0.08520049764351412,
5
+ "train_runtime": 91.5555,
6
+ "train_samples_per_second": 37.136,
7
+ "train_steps_per_second": 1.201
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,137 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.09644630551338196,
3
+ "best_model_checkpoint": "finetuned-ai-real-beit/checkpoint-50",
4
+ "epoch": 5.0,
5
+ "eval_steps": 50,
6
+ "global_step": 110,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.45454545454545453,
13
+ "grad_norm": 4.455732345581055,
14
+ "learning_rate": 0.00018363636363636366,
15
+ "loss": 0.3675,
16
+ "step": 10
17
+ },
18
+ {
19
+ "epoch": 0.9090909090909091,
20
+ "grad_norm": 6.04647159576416,
21
+ "learning_rate": 0.00016545454545454545,
22
+ "loss": 0.1004,
23
+ "step": 20
24
+ },
25
+ {
26
+ "epoch": 1.3636363636363638,
27
+ "grad_norm": 7.7111616134643555,
28
+ "learning_rate": 0.00014727272727272728,
29
+ "loss": 0.1074,
30
+ "step": 30
31
+ },
32
+ {
33
+ "epoch": 1.8181818181818183,
34
+ "grad_norm": 3.5139763355255127,
35
+ "learning_rate": 0.0001290909090909091,
36
+ "loss": 0.0934,
37
+ "step": 40
38
+ },
39
+ {
40
+ "epoch": 2.2727272727272725,
41
+ "grad_norm": 1.9594430923461914,
42
+ "learning_rate": 0.00011090909090909092,
43
+ "loss": 0.0724,
44
+ "step": 50
45
+ },
46
+ {
47
+ "epoch": 2.2727272727272725,
48
+ "eval_accuracy": 0.9669421487603306,
49
+ "eval_loss": 0.09644630551338196,
50
+ "eval_runtime": 1.506,
51
+ "eval_samples_per_second": 80.347,
52
+ "eval_steps_per_second": 10.624,
53
+ "step": 50
54
+ },
55
+ {
56
+ "epoch": 2.7272727272727275,
57
+ "grad_norm": 1.1000548601150513,
58
+ "learning_rate": 9.272727272727273e-05,
59
+ "loss": 0.0213,
60
+ "step": 60
61
+ },
62
+ {
63
+ "epoch": 3.1818181818181817,
64
+ "grad_norm": 1.2705051898956299,
65
+ "learning_rate": 7.454545454545455e-05,
66
+ "loss": 0.0563,
67
+ "step": 70
68
+ },
69
+ {
70
+ "epoch": 3.6363636363636362,
71
+ "grad_norm": 0.07004624605178833,
72
+ "learning_rate": 5.636363636363636e-05,
73
+ "loss": 0.0588,
74
+ "step": 80
75
+ },
76
+ {
77
+ "epoch": 4.090909090909091,
78
+ "grad_norm": 6.447866916656494,
79
+ "learning_rate": 3.818181818181819e-05,
80
+ "loss": 0.0296,
81
+ "step": 90
82
+ },
83
+ {
84
+ "epoch": 4.545454545454545,
85
+ "grad_norm": 0.036224253475666046,
86
+ "learning_rate": 2e-05,
87
+ "loss": 0.0198,
88
+ "step": 100
89
+ },
90
+ {
91
+ "epoch": 4.545454545454545,
92
+ "eval_accuracy": 0.9586776859504132,
93
+ "eval_loss": 0.13395561277866364,
94
+ "eval_runtime": 1.4421,
95
+ "eval_samples_per_second": 83.904,
96
+ "eval_steps_per_second": 11.095,
97
+ "step": 100
98
+ },
99
+ {
100
+ "epoch": 5.0,
101
+ "grad_norm": 0.009324366226792336,
102
+ "learning_rate": 1.818181818181818e-06,
103
+ "loss": 0.0103,
104
+ "step": 110
105
+ },
106
+ {
107
+ "epoch": 5.0,
108
+ "step": 110,
109
+ "total_flos": 2.633601533681664e+17,
110
+ "train_loss": 0.08520049764351412,
111
+ "train_runtime": 91.5555,
112
+ "train_samples_per_second": 37.136,
113
+ "train_steps_per_second": 1.201
114
+ }
115
+ ],
116
+ "logging_steps": 10,
117
+ "max_steps": 110,
118
+ "num_input_tokens_seen": 0,
119
+ "num_train_epochs": 5,
120
+ "save_steps": 100,
121
+ "stateful_callbacks": {
122
+ "TrainerControl": {
123
+ "args": {
124
+ "should_epoch_stop": false,
125
+ "should_evaluate": false,
126
+ "should_log": false,
127
+ "should_save": true,
128
+ "should_training_stop": true
129
+ },
130
+ "attributes": {}
131
+ }
132
+ },
133
+ "total_flos": 2.633601533681664e+17,
134
+ "train_batch_size": 32,
135
+ "trial_name": null,
136
+ "trial_params": null
137
+ }
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:45fa37e2ac23c6aefde3bb9d54dc6c7d8b03764c20c49eb1a9948e2a18f4dffe
3
  size 5304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1e8449390aa8c61fc3185c0d7e732e49620f1f71e94961cacae30b9a81189611
3
  size 5304