Hugging Face's logo

muhammadummerrr
/
vit-watermark-detector-v1

Safetensors
vit
Model card Files
xet
 Community
muhammadummerrr commited on
Commit
69d625d
·
verified ·
1 Parent(s): 71b960c

Upload folder using huggingface_hub

Browse files
Files changed (20) hide show
  1. checkpoint-110/config.json +31 -0
  2. checkpoint-110/model.safetensors +3 -0
  3. checkpoint-110/optimizer.pt +3 -0
  4. checkpoint-110/preprocessor_config.json +22 -0
  5. checkpoint-110/rng_state.pth +3 -0
  6. checkpoint-110/scheduler.pt +3 -0
  7. checkpoint-110/trainer_state.json +167 -0
  8. checkpoint-110/training_args.bin +3 -0
  9. checkpoint-132/config.json +31 -0
  10. checkpoint-132/model.safetensors +3 -0
  11. checkpoint-132/optimizer.pt +3 -0
  12. checkpoint-132/preprocessor_config.json +22 -0
  13. checkpoint-132/rng_state.pth +3 -0
  14. checkpoint-132/scheduler.pt +3 -0
  15. checkpoint-132/trainer_state.json +191 -0
  16. checkpoint-132/training_args.bin +3 -0
  17. config.json +31 -0
  18. model.safetensors +3 -0
  19. preprocessor_config.json +22 -0
  20. training_args.bin +3 -0
checkpoint-110/config.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/vit-base-patch16-224-in21k",
3
+ "architectures": [
4
+ "ViTForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "encoder_stride": 16,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.0,
10
+ "hidden_size": 768,
11
+ "id2label": {
12
+ "0": "clean",
13
+ "1": "watermarked"
14
+ },
15
+ "image_size": 224,
16
+ "initializer_range": 0.02,
17
+ "intermediate_size": 3072,
18
+ "label2id": {
19
+ "clean": 0,
20
+ "watermarked": 1
21
+ },
22
+ "layer_norm_eps": 1e-12,
23
+ "model_type": "vit",
24
+ "num_attention_heads": 12,
25
+ "num_channels": 3,
26
+ "num_hidden_layers": 12,
27
+ "patch_size": 16,
28
+ "qkv_bias": true,
29
+ "torch_dtype": "float32",
30
+ "transformers_version": "4.44.2"
31
+ }
checkpoint-110/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5026ebc4c0e5ef2844a1cdb4f546978f62237817df7929a4f5f43f74c7837cb7
3
+ size 343223968
checkpoint-110/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:78fc32ccbe2a7efedb812bb2bdb9b68f24a23ef5119ceca78e01afaa36367468
3
+ size 686568890
checkpoint-110/preprocessor_config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_rescale": true,
4
+ "do_resize": true,
5
+ "image_mean": [
6
+ 0.5,
7
+ 0.5,
8
+ 0.5
9
+ ],
10
+ "image_processor_type": "ViTImageProcessor",
11
+ "image_std": [
12
+ 0.5,
13
+ 0.5,
14
+ 0.5
15
+ ],
16
+ "resample": 2,
17
+ "rescale_factor": 0.00392156862745098,
18
+ "size": {
19
+ "height": 224,
20
+ "width": 224
21
+ }
22
+ }
checkpoint-110/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9afbf853352cfbcfea61884ff6a2ddcd2aee1ce8618589cf5b56912c1b160011
3
+ size 14244
checkpoint-110/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:43df589a136f2bc1b4b39a65583a1e1ac6040a06981a14c05d1b6e6364f38319
3
+ size 1064
checkpoint-110/trainer_state.json ADDED
@@ -0,0 +1,167 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.9285714285714286,
3
+ "best_model_checkpoint": "/kaggle/working/wm_vit_advanced_out/checkpoint-110",
4
+ "epoch": 10.0,
5
+ "eval_steps": 500,
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": 1.0,
13
+ "eval_accuracy": 0.82,
14
+ "eval_f1": 0.8187919463087249,
15
+ "eval_loss": 0.5443980097770691,
16
+ "eval_precision": 0.9682539682539683,
17
+ "eval_recall": 0.7093023255813954,
18
+ "eval_runtime": 12.0414,
19
+ "eval_samples_per_second": 12.457,
20
+ "eval_steps_per_second": 0.249,
21
+ "step": 11
22
+ },
23
+ {
24
+ "epoch": 2.0,
25
+ "eval_accuracy": 0.8666666666666667,
26
+ "eval_f1": 0.8765432098765432,
27
+ "eval_loss": 0.4023720324039459,
28
+ "eval_precision": 0.9342105263157895,
29
+ "eval_recall": 0.8255813953488372,
30
+ "eval_runtime": 12.0898,
31
+ "eval_samples_per_second": 12.407,
32
+ "eval_steps_per_second": 0.248,
33
+ "step": 22
34
+ },
35
+ {
36
+ "epoch": 3.0,
37
+ "eval_accuracy": 0.8933333333333333,
38
+ "eval_f1": 0.9069767441860465,
39
+ "eval_loss": 0.33101674914360046,
40
+ "eval_precision": 0.9069767441860465,
41
+ "eval_recall": 0.9069767441860465,
42
+ "eval_runtime": 12.0605,
43
+ "eval_samples_per_second": 12.437,
44
+ "eval_steps_per_second": 0.249,
45
+ "step": 33
46
+ },
47
+ {
48
+ "epoch": 4.0,
49
+ "eval_accuracy": 0.86,
50
+ "eval_f1": 0.88268156424581,
51
+ "eval_loss": 0.34140676259994507,
52
+ "eval_precision": 0.8494623655913979,
53
+ "eval_recall": 0.9186046511627907,
54
+ "eval_runtime": 12.0711,
55
+ "eval_samples_per_second": 12.426,
56
+ "eval_steps_per_second": 0.249,
57
+ "step": 44
58
+ },
59
+ {
60
+ "epoch": 4.545454545454545,
61
+ "grad_norm": 98275.65625,
62
+ "learning_rate": 1.8636363636363638e-05,
63
+ "loss": 0.4091,
64
+ "step": 50
65
+ },
66
+ {
67
+ "epoch": 5.0,
68
+ "eval_accuracy": 0.92,
69
+ "eval_f1": 0.927710843373494,
70
+ "eval_loss": 0.23491248488426208,
71
+ "eval_precision": 0.9625,
72
+ "eval_recall": 0.8953488372093024,
73
+ "eval_runtime": 12.0071,
74
+ "eval_samples_per_second": 12.493,
75
+ "eval_steps_per_second": 0.25,
76
+ "step": 55
77
+ },
78
+ {
79
+ "epoch": 6.0,
80
+ "eval_accuracy": 0.8933333333333333,
81
+ "eval_f1": 0.8974358974358974,
82
+ "eval_loss": 0.2767026126384735,
83
+ "eval_precision": 1.0,
84
+ "eval_recall": 0.813953488372093,
85
+ "eval_runtime": 10.3605,
86
+ "eval_samples_per_second": 14.478,
87
+ "eval_steps_per_second": 0.29,
88
+ "step": 66
89
+ },
90
+ {
91
+ "epoch": 7.0,
92
+ "eval_accuracy": 0.8933333333333333,
93
+ "eval_f1": 0.9,
94
+ "eval_loss": 0.2597341537475586,
95
+ "eval_precision": 0.972972972972973,
96
+ "eval_recall": 0.8372093023255814,
97
+ "eval_runtime": 12.1419,
98
+ "eval_samples_per_second": 12.354,
99
+ "eval_steps_per_second": 0.247,
100
+ "step": 77
101
+ },
102
+ {
103
+ "epoch": 8.0,
104
+ "eval_accuracy": 0.92,
105
+ "eval_f1": 0.927710843373494,
106
+ "eval_loss": 0.24549703299999237,
107
+ "eval_precision": 0.9625,
108
+ "eval_recall": 0.8953488372093024,
109
+ "eval_runtime": 12.0766,
110
+ "eval_samples_per_second": 12.421,
111
+ "eval_steps_per_second": 0.248,
112
+ "step": 88
113
+ },
114
+ {
115
+ "epoch": 9.0,
116
+ "eval_accuracy": 0.92,
117
+ "eval_f1": 0.9268292682926831,
118
+ "eval_loss": 0.24640235304832458,
119
+ "eval_precision": 0.9743589743589743,
120
+ "eval_recall": 0.8837209302325582,
121
+ "eval_runtime": 11.0609,
122
+ "eval_samples_per_second": 13.561,
123
+ "eval_steps_per_second": 0.271,
124
+ "step": 99
125
+ },
126
+ {
127
+ "epoch": 9.090909090909092,
128
+ "grad_norm": 32005.05078125,
129
+ "learning_rate": 7.272727272727273e-06,
130
+ "loss": 0.1826,
131
+ "step": 100
132
+ },
133
+ {
134
+ "epoch": 10.0,
135
+ "eval_accuracy": 0.92,
136
+ "eval_f1": 0.9285714285714286,
137
+ "eval_loss": 0.23324181139469147,
138
+ "eval_precision": 0.9512195121951219,
139
+ "eval_recall": 0.9069767441860465,
140
+ "eval_runtime": 12.0736,
141
+ "eval_samples_per_second": 12.424,
142
+ "eval_steps_per_second": 0.248,
143
+ "step": 110
144
+ }
145
+ ],
146
+ "logging_steps": 50,
147
+ "max_steps": 132,
148
+ "num_input_tokens_seen": 0,
149
+ "num_train_epochs": 12,
150
+ "save_steps": 500,
151
+ "stateful_callbacks": {
152
+ "TrainerControl": {
153
+ "args": {
154
+ "should_epoch_stop": false,
155
+ "should_evaluate": false,
156
+ "should_log": false,
157
+ "should_save": true,
158
+ "should_training_stop": false
159
+ },
160
+ "attributes": {}
161
+ }
162
+ },
163
+ "total_flos": 5.42443927302144e+17,
164
+ "train_batch_size": 64,
165
+ "trial_name": null,
166
+ "trial_params": null
167
+ }
checkpoint-110/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fd9fe542da752b1d51b22235bff375ccfc30f8e5ff8dc485da0dc4ff2cd6a162
3
+ size 5176
checkpoint-132/config.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/vit-base-patch16-224-in21k",
3
+ "architectures": [
4
+ "ViTForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "encoder_stride": 16,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.0,
10
+ "hidden_size": 768,
11
+ "id2label": {
12
+ "0": "clean",
13
+ "1": "watermarked"
14
+ },
15
+ "image_size": 224,
16
+ "initializer_range": 0.02,
17
+ "intermediate_size": 3072,
18
+ "label2id": {
19
+ "clean": 0,
20
+ "watermarked": 1
21
+ },
22
+ "layer_norm_eps": 1e-12,
23
+ "model_type": "vit",
24
+ "num_attention_heads": 12,
25
+ "num_channels": 3,
26
+ "num_hidden_layers": 12,
27
+ "patch_size": 16,
28
+ "qkv_bias": true,
29
+ "torch_dtype": "float32",
30
+ "transformers_version": "4.44.2"
31
+ }
checkpoint-132/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7c98075e052eb2cf6f8b7bd8b36c211680e4da41faea0f53f48222c83cff0911
3
+ size 343223968
checkpoint-132/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e62468c62a2c88cf8bbc4a3c024576ef0242fe4a264e345730382b9311c3188c
3
+ size 686568890
checkpoint-132/preprocessor_config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_rescale": true,
4
+ "do_resize": true,
5
+ "image_mean": [
6
+ 0.5,
7
+ 0.5,
8
+ 0.5
9
+ ],
10
+ "image_processor_type": "ViTImageProcessor",
11
+ "image_std": [
12
+ 0.5,
13
+ 0.5,
14
+ 0.5
15
+ ],
16
+ "resample": 2,
17
+ "rescale_factor": 0.00392156862745098,
18
+ "size": {
19
+ "height": 224,
20
+ "width": 224
21
+ }
22
+ }
checkpoint-132/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d5f38f6446ba43bb9ae0be4911c150b41c5adfbe2712844ea3cb854b62ee2432
3
+ size 14244
checkpoint-132/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9c60c335588b8720f6070411dc45ff1ed1af385850001eb06906efd0af2d84b0
3
+ size 1064
checkpoint-132/trainer_state.json ADDED
@@ -0,0 +1,191 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.9285714285714286,
3
+ "best_model_checkpoint": "/kaggle/working/wm_vit_advanced_out/checkpoint-110",
4
+ "epoch": 12.0,
5
+ "eval_steps": 500,
6
+ "global_step": 132,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 1.0,
13
+ "eval_accuracy": 0.82,
14
+ "eval_f1": 0.8187919463087249,
15
+ "eval_loss": 0.5443980097770691,
16
+ "eval_precision": 0.9682539682539683,
17
+ "eval_recall": 0.7093023255813954,
18
+ "eval_runtime": 12.0414,
19
+ "eval_samples_per_second": 12.457,
20
+ "eval_steps_per_second": 0.249,
21
+ "step": 11
22
+ },
23
+ {
24
+ "epoch": 2.0,
25
+ "eval_accuracy": 0.8666666666666667,
26
+ "eval_f1": 0.8765432098765432,
27
+ "eval_loss": 0.4023720324039459,
28
+ "eval_precision": 0.9342105263157895,
29
+ "eval_recall": 0.8255813953488372,
30
+ "eval_runtime": 12.0898,
31
+ "eval_samples_per_second": 12.407,
32
+ "eval_steps_per_second": 0.248,
33
+ "step": 22
34
+ },
35
+ {
36
+ "epoch": 3.0,
37
+ "eval_accuracy": 0.8933333333333333,
38
+ "eval_f1": 0.9069767441860465,
39
+ "eval_loss": 0.33101674914360046,
40
+ "eval_precision": 0.9069767441860465,
41
+ "eval_recall": 0.9069767441860465,
42
+ "eval_runtime": 12.0605,
43
+ "eval_samples_per_second": 12.437,
44
+ "eval_steps_per_second": 0.249,
45
+ "step": 33
46
+ },
47
+ {
48
+ "epoch": 4.0,
49
+ "eval_accuracy": 0.86,
50
+ "eval_f1": 0.88268156424581,
51
+ "eval_loss": 0.34140676259994507,
52
+ "eval_precision": 0.8494623655913979,
53
+ "eval_recall": 0.9186046511627907,
54
+ "eval_runtime": 12.0711,
55
+ "eval_samples_per_second": 12.426,
56
+ "eval_steps_per_second": 0.249,
57
+ "step": 44
58
+ },
59
+ {
60
+ "epoch": 4.545454545454545,
61
+ "grad_norm": 98275.65625,
62
+ "learning_rate": 1.8636363636363638e-05,
63
+ "loss": 0.4091,
64
+ "step": 50
65
+ },
66
+ {
67
+ "epoch": 5.0,
68
+ "eval_accuracy": 0.92,
69
+ "eval_f1": 0.927710843373494,
70
+ "eval_loss": 0.23491248488426208,
71
+ "eval_precision": 0.9625,
72
+ "eval_recall": 0.8953488372093024,
73
+ "eval_runtime": 12.0071,
74
+ "eval_samples_per_second": 12.493,
75
+ "eval_steps_per_second": 0.25,
76
+ "step": 55
77
+ },
78
+ {
79
+ "epoch": 6.0,
80
+ "eval_accuracy": 0.8933333333333333,
81
+ "eval_f1": 0.8974358974358974,
82
+ "eval_loss": 0.2767026126384735,
83
+ "eval_precision": 1.0,
84
+ "eval_recall": 0.813953488372093,
85
+ "eval_runtime": 10.3605,
86
+ "eval_samples_per_second": 14.478,
87
+ "eval_steps_per_second": 0.29,
88
+ "step": 66
89
+ },
90
+ {
91
+ "epoch": 7.0,
92
+ "eval_accuracy": 0.8933333333333333,
93
+ "eval_f1": 0.9,
94
+ "eval_loss": 0.2597341537475586,
95
+ "eval_precision": 0.972972972972973,
96
+ "eval_recall": 0.8372093023255814,
97
+ "eval_runtime": 12.1419,
98
+ "eval_samples_per_second": 12.354,
99
+ "eval_steps_per_second": 0.247,
100
+ "step": 77
101
+ },
102
+ {
103
+ "epoch": 8.0,
104
+ "eval_accuracy": 0.92,
105
+ "eval_f1": 0.927710843373494,
106
+ "eval_loss": 0.24549703299999237,
107
+ "eval_precision": 0.9625,
108
+ "eval_recall": 0.8953488372093024,
109
+ "eval_runtime": 12.0766,
110
+ "eval_samples_per_second": 12.421,
111
+ "eval_steps_per_second": 0.248,
112
+ "step": 88
113
+ },
114
+ {
115
+ "epoch": 9.0,
116
+ "eval_accuracy": 0.92,
117
+ "eval_f1": 0.9268292682926831,
118
+ "eval_loss": 0.24640235304832458,
119
+ "eval_precision": 0.9743589743589743,
120
+ "eval_recall": 0.8837209302325582,
121
+ "eval_runtime": 11.0609,
122
+ "eval_samples_per_second": 13.561,
123
+ "eval_steps_per_second": 0.271,
124
+ "step": 99
125
+ },
126
+ {
127
+ "epoch": 9.090909090909092,
128
+ "grad_norm": 32005.05078125,
129
+ "learning_rate": 7.272727272727273e-06,
130
+ "loss": 0.1826,
131
+ "step": 100
132
+ },
133
+ {
134
+ "epoch": 10.0,
135
+ "eval_accuracy": 0.92,
136
+ "eval_f1": 0.9285714285714286,
137
+ "eval_loss": 0.23324181139469147,
138
+ "eval_precision": 0.9512195121951219,
139
+ "eval_recall": 0.9069767441860465,
140
+ "eval_runtime": 12.0736,
141
+ "eval_samples_per_second": 12.424,
142
+ "eval_steps_per_second": 0.248,
143
+ "step": 110
144
+ },
145
+ {
146
+ "epoch": 11.0,
147
+ "eval_accuracy": 0.92,
148
+ "eval_f1": 0.9268292682926831,
149
+ "eval_loss": 0.23514790832996368,
150
+ "eval_precision": 0.9743589743589743,
151
+ "eval_recall": 0.8837209302325582,
152
+ "eval_runtime": 12.0826,
153
+ "eval_samples_per_second": 12.414,
154
+ "eval_steps_per_second": 0.248,
155
+ "step": 121
156
+ },
157
+ {
158
+ "epoch": 12.0,
159
+ "eval_accuracy": 0.92,
160
+ "eval_f1": 0.9268292682926831,
161
+ "eval_loss": 0.23184794187545776,
162
+ "eval_precision": 0.9743589743589743,
163
+ "eval_recall": 0.8837209302325582,
164
+ "eval_runtime": 12.082,
165
+ "eval_samples_per_second": 12.415,
166
+ "eval_steps_per_second": 0.248,
167
+ "step": 132
168
+ }
169
+ ],
170
+ "logging_steps": 50,
171
+ "max_steps": 132,
172
+ "num_input_tokens_seen": 0,
173
+ "num_train_epochs": 12,
174
+ "save_steps": 500,
175
+ "stateful_callbacks": {
176
+ "TrainerControl": {
177
+ "args": {
178
+ "should_epoch_stop": false,
179
+ "should_evaluate": false,
180
+ "should_log": false,
181
+ "should_save": true,
182
+ "should_training_stop": true
183
+ },
184
+ "attributes": {}
185
+ }
186
+ },
187
+ "total_flos": 6.509327127625728e+17,
188
+ "train_batch_size": 64,
189
+ "trial_name": null,
190
+ "trial_params": null
191
+ }
checkpoint-132/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fd9fe542da752b1d51b22235bff375ccfc30f8e5ff8dc485da0dc4ff2cd6a162
3
+ size 5176
config.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/vit-base-patch16-224-in21k",
3
+ "architectures": [
4
+ "ViTForImageClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "encoder_stride": 16,
8
+ "hidden_act": "gelu",
9
+ "hidden_dropout_prob": 0.0,
10
+ "hidden_size": 768,
11
+ "id2label": {
12
+ "0": "clean",
13
+ "1": "watermarked"
14
+ },
15
+ "image_size": 224,
16
+ "initializer_range": 0.02,
17
+ "intermediate_size": 3072,
18
+ "label2id": {
19
+ "clean": 0,
20
+ "watermarked": 1
21
+ },
22
+ "layer_norm_eps": 1e-12,
23
+ "model_type": "vit",
24
+ "num_attention_heads": 12,
25
+ "num_channels": 3,
26
+ "num_hidden_layers": 12,
27
+ "patch_size": 16,
28
+ "qkv_bias": true,
29
+ "torch_dtype": "float32",
30
+ "transformers_version": "4.44.2"
31
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5026ebc4c0e5ef2844a1cdb4f546978f62237817df7929a4f5f43f74c7837cb7
3
+ size 343223968
preprocessor_config.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_normalize": true,
3
+ "do_rescale": true,
4
+ "do_resize": true,
5
+ "image_mean": [
6
+ 0.5,
7
+ 0.5,
8
+ 0.5
9
+ ],
10
+ "image_processor_type": "ViTImageProcessor",
11
+ "image_std": [
12
+ 0.5,
13
+ 0.5,
14
+ 0.5
15
+ ],
16
+ "resample": 2,
17
+ "rescale_factor": 0.00392156862745098,
18
+ "size": {
19
+ "height": 224,
20
+ "width": 224
21
+ }
22
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fd9fe542da752b1d51b22235bff375ccfc30f8e5ff8dc485da0dc4ff2cd6a162
3
+ size 5176