vivi3n789 commited on
Commit
d015b9f
1 Parent(s): 825f06b

Upload folder using huggingface_hub

Browse files
README.md ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+ tags:
4
+ - autotrain
5
+ - image-classification
6
+ base_model: dima806/deepfake_vs_real_image_detection
7
+ widget:
8
+ - src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/tiger.jpg
9
+ example_title: Tiger
10
+ - src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/teapot.jpg
11
+ example_title: Teapot
12
+ - src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/palace.jpg
13
+ example_title: Palace
14
+ ---
15
+
16
+ # Model Trained Using AutoTrain
17
+
18
+ - Problem type: Image Classification
19
+
20
+ ## Validation Metrics
21
+ loss: 0.06105376407504082
22
+
23
+ f1: 0.9850746268656716
24
+
25
+ precision: 0.9801980198019802
26
+
27
+ recall: 0.99
28
+
29
+ auc: 0.998925
30
+
31
+ accuracy: 0.985
checkpoint-400/config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "dima806/deepfake_vs_real_image_detection",
3
+ "_num_labels": 2,
4
+ "architectures": [
5
+ "ViTForImageClassification"
6
+ ],
7
+ "attention_probs_dropout_prob": 0.0,
8
+ "encoder_stride": 16,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.0,
11
+ "hidden_size": 768,
12
+ "id2label": {
13
+ "0": "fake",
14
+ "1": "real"
15
+ },
16
+ "image_size": 224,
17
+ "initializer_range": 0.02,
18
+ "intermediate_size": 3072,
19
+ "label2id": {
20
+ "fake": 0,
21
+ "real": 1
22
+ },
23
+ "layer_norm_eps": 1e-12,
24
+ "model_type": "vit",
25
+ "num_attention_heads": 12,
26
+ "num_channels": 3,
27
+ "num_hidden_layers": 12,
28
+ "patch_size": 16,
29
+ "problem_type": "single_label_classification",
30
+ "qkv_bias": true,
31
+ "torch_dtype": "float32",
32
+ "transformers_version": "4.46.2"
33
+ }
checkpoint-400/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4b1c40f8783844502d80da8460c5c7fd2cb66fb3e47d8de741e8bb5ff9341c9b
3
+ size 343223968
checkpoint-400/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:379b8dd580a97fd0a5d5c4bb6de55407c94bad12a84ea956cdc118a60e1508ff
3
+ size 686563258
checkpoint-400/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f56790c3ee491baab329d1fe00f10cb0b9f5114649fad9b4021437c861c9b810
3
+ size 13990
checkpoint-400/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b7d443b575f4a41f34baa27ec05975c054652690958a3dc51640fcf06060285
3
+ size 1064
checkpoint-400/trainer_state.json ADDED
@@ -0,0 +1,348 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.06105376407504082,
3
+ "best_model_checkpoint": "deepfake-custom/checkpoint-400",
4
+ "epoch": 2.0,
5
+ "eval_steps": 500,
6
+ "global_step": 400,
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.05,
13
+ "grad_norm": 26.546144485473633,
14
+ "learning_rate": 8.333333333333334e-06,
15
+ "loss": 4.7282,
16
+ "step": 10
17
+ },
18
+ {
19
+ "epoch": 0.1,
20
+ "grad_norm": 37.319374084472656,
21
+ "learning_rate": 1.6666666666666667e-05,
22
+ "loss": 4.2879,
23
+ "step": 20
24
+ },
25
+ {
26
+ "epoch": 0.15,
27
+ "grad_norm": 10.413372039794922,
28
+ "learning_rate": 2.5e-05,
29
+ "loss": 1.7524,
30
+ "step": 30
31
+ },
32
+ {
33
+ "epoch": 0.2,
34
+ "grad_norm": 6.332607269287109,
35
+ "learning_rate": 3.3333333333333335e-05,
36
+ "loss": 0.7878,
37
+ "step": 40
38
+ },
39
+ {
40
+ "epoch": 0.25,
41
+ "grad_norm": 11.610198020935059,
42
+ "learning_rate": 4.166666666666667e-05,
43
+ "loss": 0.5413,
44
+ "step": 50
45
+ },
46
+ {
47
+ "epoch": 0.3,
48
+ "grad_norm": 4.810694694519043,
49
+ "learning_rate": 5e-05,
50
+ "loss": 0.4014,
51
+ "step": 60
52
+ },
53
+ {
54
+ "epoch": 0.35,
55
+ "grad_norm": 2.429203987121582,
56
+ "learning_rate": 4.9074074074074075e-05,
57
+ "loss": 0.2954,
58
+ "step": 70
59
+ },
60
+ {
61
+ "epoch": 0.4,
62
+ "grad_norm": 0.562802791595459,
63
+ "learning_rate": 4.814814814814815e-05,
64
+ "loss": 0.2239,
65
+ "step": 80
66
+ },
67
+ {
68
+ "epoch": 0.45,
69
+ "grad_norm": 3.2474796772003174,
70
+ "learning_rate": 4.722222222222222e-05,
71
+ "loss": 0.3575,
72
+ "step": 90
73
+ },
74
+ {
75
+ "epoch": 0.5,
76
+ "grad_norm": 2.1597166061401367,
77
+ "learning_rate": 4.62962962962963e-05,
78
+ "loss": 0.1528,
79
+ "step": 100
80
+ },
81
+ {
82
+ "epoch": 0.55,
83
+ "grad_norm": 8.586223602294922,
84
+ "learning_rate": 4.5370370370370374e-05,
85
+ "loss": 0.2503,
86
+ "step": 110
87
+ },
88
+ {
89
+ "epoch": 0.6,
90
+ "grad_norm": 26.026302337646484,
91
+ "learning_rate": 4.4444444444444447e-05,
92
+ "loss": 0.114,
93
+ "step": 120
94
+ },
95
+ {
96
+ "epoch": 0.65,
97
+ "grad_norm": 0.38507279753685,
98
+ "learning_rate": 4.351851851851852e-05,
99
+ "loss": 0.2078,
100
+ "step": 130
101
+ },
102
+ {
103
+ "epoch": 0.7,
104
+ "grad_norm": 10.108200073242188,
105
+ "learning_rate": 4.259259259259259e-05,
106
+ "loss": 0.1432,
107
+ "step": 140
108
+ },
109
+ {
110
+ "epoch": 0.75,
111
+ "grad_norm": 2.9426777362823486,
112
+ "learning_rate": 4.166666666666667e-05,
113
+ "loss": 0.1231,
114
+ "step": 150
115
+ },
116
+ {
117
+ "epoch": 0.8,
118
+ "grad_norm": 37.13353729248047,
119
+ "learning_rate": 4.074074074074074e-05,
120
+ "loss": 0.1891,
121
+ "step": 160
122
+ },
123
+ {
124
+ "epoch": 0.85,
125
+ "grad_norm": 16.757221221923828,
126
+ "learning_rate": 3.981481481481482e-05,
127
+ "loss": 0.2966,
128
+ "step": 170
129
+ },
130
+ {
131
+ "epoch": 0.9,
132
+ "grad_norm": 17.378347396850586,
133
+ "learning_rate": 3.888888888888889e-05,
134
+ "loss": 0.2156,
135
+ "step": 180
136
+ },
137
+ {
138
+ "epoch": 0.95,
139
+ "grad_norm": 0.16039691865444183,
140
+ "learning_rate": 3.7962962962962964e-05,
141
+ "loss": 0.1215,
142
+ "step": 190
143
+ },
144
+ {
145
+ "epoch": 1.0,
146
+ "grad_norm": 32.3621940612793,
147
+ "learning_rate": 3.7037037037037037e-05,
148
+ "loss": 0.2481,
149
+ "step": 200
150
+ },
151
+ {
152
+ "epoch": 1.0,
153
+ "eval_accuracy": 0.95,
154
+ "eval_auc": 0.9906250000000001,
155
+ "eval_f1": 0.9512195121951219,
156
+ "eval_loss": 0.16212879121303558,
157
+ "eval_precision": 0.9285714285714286,
158
+ "eval_recall": 0.975,
159
+ "eval_runtime": 117.7512,
160
+ "eval_samples_per_second": 3.397,
161
+ "eval_steps_per_second": 0.212,
162
+ "step": 200
163
+ },
164
+ {
165
+ "epoch": 1.05,
166
+ "grad_norm": 0.0691157802939415,
167
+ "learning_rate": 3.611111111111111e-05,
168
+ "loss": 0.114,
169
+ "step": 210
170
+ },
171
+ {
172
+ "epoch": 1.1,
173
+ "grad_norm": 0.07614326477050781,
174
+ "learning_rate": 3.518518518518519e-05,
175
+ "loss": 0.1587,
176
+ "step": 220
177
+ },
178
+ {
179
+ "epoch": 1.15,
180
+ "grad_norm": 3.3094229698181152,
181
+ "learning_rate": 3.425925925925926e-05,
182
+ "loss": 0.1153,
183
+ "step": 230
184
+ },
185
+ {
186
+ "epoch": 1.2,
187
+ "grad_norm": 0.43230101466178894,
188
+ "learning_rate": 3.3333333333333335e-05,
189
+ "loss": 0.17,
190
+ "step": 240
191
+ },
192
+ {
193
+ "epoch": 1.25,
194
+ "grad_norm": 4.47321891784668,
195
+ "learning_rate": 3.240740740740741e-05,
196
+ "loss": 0.2543,
197
+ "step": 250
198
+ },
199
+ {
200
+ "epoch": 1.3,
201
+ "grad_norm": 13.483596801757812,
202
+ "learning_rate": 3.148148148148148e-05,
203
+ "loss": 0.3613,
204
+ "step": 260
205
+ },
206
+ {
207
+ "epoch": 1.35,
208
+ "grad_norm": 0.24584409594535828,
209
+ "learning_rate": 3.055555555555556e-05,
210
+ "loss": 0.1508,
211
+ "step": 270
212
+ },
213
+ {
214
+ "epoch": 1.4,
215
+ "grad_norm": 0.4941992163658142,
216
+ "learning_rate": 2.962962962962963e-05,
217
+ "loss": 0.2375,
218
+ "step": 280
219
+ },
220
+ {
221
+ "epoch": 1.45,
222
+ "grad_norm": 2.139481782913208,
223
+ "learning_rate": 2.8703703703703706e-05,
224
+ "loss": 0.1685,
225
+ "step": 290
226
+ },
227
+ {
228
+ "epoch": 1.5,
229
+ "grad_norm": 0.24849505722522736,
230
+ "learning_rate": 2.777777777777778e-05,
231
+ "loss": 0.1729,
232
+ "step": 300
233
+ },
234
+ {
235
+ "epoch": 1.55,
236
+ "grad_norm": 13.943930625915527,
237
+ "learning_rate": 2.6851851851851855e-05,
238
+ "loss": 0.2713,
239
+ "step": 310
240
+ },
241
+ {
242
+ "epoch": 1.6,
243
+ "grad_norm": 18.25007438659668,
244
+ "learning_rate": 2.5925925925925925e-05,
245
+ "loss": 0.2127,
246
+ "step": 320
247
+ },
248
+ {
249
+ "epoch": 1.65,
250
+ "grad_norm": 0.24169185757637024,
251
+ "learning_rate": 2.5e-05,
252
+ "loss": 0.0963,
253
+ "step": 330
254
+ },
255
+ {
256
+ "epoch": 1.7,
257
+ "grad_norm": 2.021479845046997,
258
+ "learning_rate": 2.4074074074074074e-05,
259
+ "loss": 0.1298,
260
+ "step": 340
261
+ },
262
+ {
263
+ "epoch": 1.75,
264
+ "grad_norm": 12.028373718261719,
265
+ "learning_rate": 2.314814814814815e-05,
266
+ "loss": 0.1902,
267
+ "step": 350
268
+ },
269
+ {
270
+ "epoch": 1.8,
271
+ "grad_norm": 0.25696495175361633,
272
+ "learning_rate": 2.2222222222222223e-05,
273
+ "loss": 0.163,
274
+ "step": 360
275
+ },
276
+ {
277
+ "epoch": 1.85,
278
+ "grad_norm": 28.349374771118164,
279
+ "learning_rate": 2.1296296296296296e-05,
280
+ "loss": 0.1161,
281
+ "step": 370
282
+ },
283
+ {
284
+ "epoch": 1.9,
285
+ "grad_norm": 18.065284729003906,
286
+ "learning_rate": 2.037037037037037e-05,
287
+ "loss": 0.2918,
288
+ "step": 380
289
+ },
290
+ {
291
+ "epoch": 1.95,
292
+ "grad_norm": 0.1438354104757309,
293
+ "learning_rate": 1.9444444444444445e-05,
294
+ "loss": 0.158,
295
+ "step": 390
296
+ },
297
+ {
298
+ "epoch": 2.0,
299
+ "grad_norm": 0.04646734893321991,
300
+ "learning_rate": 1.8518518518518518e-05,
301
+ "loss": 0.0268,
302
+ "step": 400
303
+ },
304
+ {
305
+ "epoch": 2.0,
306
+ "eval_accuracy": 0.985,
307
+ "eval_auc": 0.998925,
308
+ "eval_f1": 0.9850746268656716,
309
+ "eval_loss": 0.06105376407504082,
310
+ "eval_precision": 0.9801980198019802,
311
+ "eval_recall": 0.99,
312
+ "eval_runtime": 122.4824,
313
+ "eval_samples_per_second": 3.266,
314
+ "eval_steps_per_second": 0.204,
315
+ "step": 400
316
+ }
317
+ ],
318
+ "logging_steps": 10,
319
+ "max_steps": 600,
320
+ "num_input_tokens_seen": 0,
321
+ "num_train_epochs": 3,
322
+ "save_steps": 500,
323
+ "stateful_callbacks": {
324
+ "EarlyStoppingCallback": {
325
+ "args": {
326
+ "early_stopping_patience": 5,
327
+ "early_stopping_threshold": 0.01
328
+ },
329
+ "attributes": {
330
+ "early_stopping_patience_counter": 0
331
+ }
332
+ },
333
+ "TrainerControl": {
334
+ "args": {
335
+ "should_epoch_stop": false,
336
+ "should_evaluate": false,
337
+ "should_log": false,
338
+ "should_save": true,
339
+ "should_training_stop": false
340
+ },
341
+ "attributes": {}
342
+ }
343
+ },
344
+ "total_flos": 2.479743667666944e+17,
345
+ "train_batch_size": 8,
346
+ "trial_name": null,
347
+ "trial_params": null
348
+ }
checkpoint-400/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:123ac1cf673e1cd2af8990dbc08bdcbeb0133521747f7e12b39a27a0c6a1fa8f
3
+ size 5304
config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "dima806/deepfake_vs_real_image_detection",
3
+ "_num_labels": 2,
4
+ "architectures": [
5
+ "ViTForImageClassification"
6
+ ],
7
+ "attention_probs_dropout_prob": 0.0,
8
+ "encoder_stride": 16,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.0,
11
+ "hidden_size": 768,
12
+ "id2label": {
13
+ "0": "fake",
14
+ "1": "real"
15
+ },
16
+ "image_size": 224,
17
+ "initializer_range": 0.02,
18
+ "intermediate_size": 3072,
19
+ "label2id": {
20
+ "fake": 0,
21
+ "real": 1
22
+ },
23
+ "layer_norm_eps": 1e-12,
24
+ "model_type": "vit",
25
+ "num_attention_heads": 12,
26
+ "num_channels": 3,
27
+ "num_hidden_layers": 12,
28
+ "patch_size": 16,
29
+ "problem_type": "single_label_classification",
30
+ "qkv_bias": true,
31
+ "torch_dtype": "float32",
32
+ "transformers_version": "4.46.2"
33
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4b1c40f8783844502d80da8460c5c7fd2cb66fb3e47d8de741e8bb5ff9341c9b
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
+ }
runs/Nov24_13-38-24_r-vivi3n789-deepfakes-fbq79tfn-950cf-uvqn7/events.out.tfevents.1732455506.r-vivi3n789-deepfakes-fbq79tfn-950cf-uvqn7.271.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c944214de1ab2788639a0fbc7d267b6142f4ed4f075315dd5927c0c1665a2a4a
3
- size 17552
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c2c8af0f25b3e9c98546492a6c9ac211a6abcaa0cb6f3acccec87881d74eb1bd
3
+ size 19691
runs/Nov24_13-38-24_r-vivi3n789-deepfakes-fbq79tfn-950cf-uvqn7/events.out.tfevents.1732460615.r-vivi3n789-deepfakes-fbq79tfn-950cf-uvqn7.271.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:98f689c8f34c540f2d825260cbf64a019526db44189d1e00bf8bd07faf8f4695
3
+ size 607
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:123ac1cf673e1cd2af8990dbc08bdcbeb0133521747f7e12b39a27a0c6a1fa8f
3
+ size 5304
training_params.json ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "deepfake-custom/autotrain-data",
3
+ "model": "dima806/deepfake_vs_real_image_detection",
4
+ "username": "vivi3n789",
5
+ "lr": 5e-05,
6
+ "epochs": 3,
7
+ "batch_size": 8,
8
+ "warmup_ratio": 0.1,
9
+ "gradient_accumulation": 1,
10
+ "optimizer": "adamw_torch",
11
+ "scheduler": "linear",
12
+ "weight_decay": 0.0,
13
+ "max_grad_norm": 1.0,
14
+ "seed": 42,
15
+ "train_split": "train",
16
+ "valid_split": "validation",
17
+ "logging_steps": -1,
18
+ "project_name": "deepfake-custom",
19
+ "auto_find_batch_size": false,
20
+ "mixed_precision": "fp16",
21
+ "save_total_limit": 1,
22
+ "push_to_hub": true,
23
+ "eval_strategy": "epoch",
24
+ "image_column": "autotrain_image",
25
+ "target_column": "autotrain_label",
26
+ "log": "tensorboard",
27
+ "early_stopping_patience": 5,
28
+ "early_stopping_threshold": 0.01
29
+ }