Hugging Face's logo

prithivMLmods
/
vit-mini-explicit-content

Image Classification
Transformers
Safetensors
English
vit
explicit-content-detection
mini
art
sensual-content-detection
Anime
Model card Files
xet
Community
prithivMLmods commited on
Commit
44cec11
·
verified ·
1 Parent(s): ba864ff

Upload folder using huggingface_hub

Browse files
Files changed (20) hide show
  1. checkpoint-1800/config.json +39 -0
  2. checkpoint-1800/model.safetensors +3 -0
  3. checkpoint-1800/optimizer.pt +3 -0
  4. checkpoint-1800/preprocessor_config.json +23 -0
  5. checkpoint-1800/rng_state.pth +3 -0
  6. checkpoint-1800/scheduler.pt +3 -0
  7. checkpoint-1800/trainer_state.json +75 -0
  8. checkpoint-1800/training_args.bin +3 -0
  9. checkpoint-900/config.json +39 -0
  10. checkpoint-900/model.safetensors +3 -0
  11. checkpoint-900/optimizer.pt +3 -0
  12. checkpoint-900/preprocessor_config.json +23 -0
  13. checkpoint-900/rng_state.pth +3 -0
  14. checkpoint-900/scheduler.pt +3 -0
  15. checkpoint-900/trainer_state.json +51 -0
  16. checkpoint-900/training_args.bin +3 -0
  17. config.json +39 -0
  18. model.safetensors +3 -0
  19. preprocessor_config.json +23 -0
  20. training_args.bin +3 -0
checkpoint-1800/config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "ViTForImageClassification"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.0,
6
+ "encoder_stride": 16,
7
+ "hidden_act": "gelu",
8
+ "hidden_dropout_prob": 0.0,
9
+ "hidden_size": 768,
10
+ "id2label": {
11
+ "0": "Anime Picture",
12
+ "1": "Extincing & Sensual",
13
+ "2": "Hentai",
14
+ "3": "Pornography",
15
+ "4": "Safe for Work"
16
+ },
17
+ "image_size": 224,
18
+ "initializer_range": 0.02,
19
+ "intermediate_size": 3072,
20
+ "label2id": {
21
+ "Anime Picture": 0,
22
+ "Extincing & Sensual": 1,
23
+ "Hentai": 2,
24
+ "Pornography": 3,
25
+ "Safe for Work": 4
26
+ },
27
+ "layer_norm_eps": 1e-12,
28
+ "model_type": "vit",
29
+ "num_attention_heads": 12,
30
+ "num_channels": 3,
31
+ "num_hidden_layers": 12,
32
+ "patch_size": 16,
33
+ "pooler_act": "tanh",
34
+ "pooler_output_size": 768,
35
+ "problem_type": "single_label_classification",
36
+ "qkv_bias": true,
37
+ "torch_dtype": "float32",
38
+ "transformers_version": "4.50.0"
39
+ }
checkpoint-1800/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:16902c62b2bb49952d5e9a848e68cb03252ae0894e916cf17f4c74783ba12c84
3
+ size 343233204
checkpoint-1800/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9a56b49a84b8727e1bc911ddf160d984195cd7fc5c67940018b9a7bf63a1759b
3
+ size 686587322
checkpoint-1800/preprocessor_config.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_convert_rgb": null,
3
+ "do_normalize": true,
4
+ "do_rescale": true,
5
+ "do_resize": true,
6
+ "image_mean": [
7
+ 0.5,
8
+ 0.5,
9
+ 0.5
10
+ ],
11
+ "image_processor_type": "ViTImageProcessor",
12
+ "image_std": [
13
+ 0.5,
14
+ 0.5,
15
+ 0.5
16
+ ],
17
+ "resample": 2,
18
+ "rescale_factor": 0.00392156862745098,
19
+ "size": {
20
+ "height": 224,
21
+ "width": 224
22
+ }
23
+ }
checkpoint-1800/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eb14b3612b9a2297307cf6e9608f2377bf8a6b696c374c21ce4c8e29efdf3c17
3
+ size 14244
checkpoint-1800/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:55469babfe5d022c80a7048ffcc9b9b8021fc9dc2750ad6cd10fa8aa95d6685e
3
+ size 1064
checkpoint-1800/trainer_state.json ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 1800,
3
+ "best_metric": 0.2879721522331238,
4
+ "best_model_checkpoint": "siglip2-finetune-full/checkpoint-1800",
5
+ "epoch": 2.0,
6
+ "eval_steps": 500,
7
+ "global_step": 1800,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.5555555555555556,
14
+ "grad_norm": 2.6622252464294434,
15
+ "learning_rate": 1.4857142857142858e-05,
16
+ "loss": 0.6889,
17
+ "step": 500
18
+ },
19
+ {
20
+ "epoch": 1.0,
21
+ "eval_accuracy": 0.905307767125191,
22
+ "eval_loss": 0.340586394071579,
23
+ "eval_model_preparation_time": 0.0027,
24
+ "eval_runtime": 495.4956,
25
+ "eval_samples_per_second": 58.099,
26
+ "eval_steps_per_second": 7.263,
27
+ "step": 900
28
+ },
29
+ {
30
+ "epoch": 1.1111111111111112,
31
+ "grad_norm": 2.4411823749542236,
32
+ "learning_rate": 9.142857142857144e-06,
33
+ "loss": 0.3509,
34
+ "step": 1000
35
+ },
36
+ {
37
+ "epoch": 1.6666666666666665,
38
+ "grad_norm": 3.061096429824829,
39
+ "learning_rate": 3.428571428571429e-06,
40
+ "loss": 0.2866,
41
+ "step": 1500
42
+ },
43
+ {
44
+ "epoch": 2.0,
45
+ "eval_accuracy": 0.917118243712658,
46
+ "eval_loss": 0.2879721522331238,
47
+ "eval_model_preparation_time": 0.0027,
48
+ "eval_runtime": 496.6254,
49
+ "eval_samples_per_second": 57.967,
50
+ "eval_steps_per_second": 7.247,
51
+ "step": 1800
52
+ }
53
+ ],
54
+ "logging_steps": 500,
55
+ "max_steps": 1800,
56
+ "num_input_tokens_seen": 0,
57
+ "num_train_epochs": 2,
58
+ "save_steps": 500,
59
+ "stateful_callbacks": {
60
+ "TrainerControl": {
61
+ "args": {
62
+ "should_epoch_stop": false,
63
+ "should_evaluate": false,
64
+ "should_log": false,
65
+ "should_save": true,
66
+ "should_training_stop": true
67
+ },
68
+ "attributes": {}
69
+ }
70
+ },
71
+ "total_flos": 4.461798759897121e+18,
72
+ "train_batch_size": 32,
73
+ "trial_name": null,
74
+ "trial_params": null
75
+ }
checkpoint-1800/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:566011f5604bed061f658701cef40a0c715a83fe851a5dcfea9ade31ab9b8d4f
3
+ size 5304
checkpoint-900/config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "ViTForImageClassification"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.0,
6
+ "encoder_stride": 16,
7
+ "hidden_act": "gelu",
8
+ "hidden_dropout_prob": 0.0,
9
+ "hidden_size": 768,
10
+ "id2label": {
11
+ "0": "Anime Picture",
12
+ "1": "Extincing & Sensual",
13
+ "2": "Hentai",
14
+ "3": "Pornography",
15
+ "4": "Safe for Work"
16
+ },
17
+ "image_size": 224,
18
+ "initializer_range": 0.02,
19
+ "intermediate_size": 3072,
20
+ "label2id": {
21
+ "Anime Picture": 0,
22
+ "Extincing & Sensual": 1,
23
+ "Hentai": 2,
24
+ "Pornography": 3,
25
+ "Safe for Work": 4
26
+ },
27
+ "layer_norm_eps": 1e-12,
28
+ "model_type": "vit",
29
+ "num_attention_heads": 12,
30
+ "num_channels": 3,
31
+ "num_hidden_layers": 12,
32
+ "patch_size": 16,
33
+ "pooler_act": "tanh",
34
+ "pooler_output_size": 768,
35
+ "problem_type": "single_label_classification",
36
+ "qkv_bias": true,
37
+ "torch_dtype": "float32",
38
+ "transformers_version": "4.50.0"
39
+ }
checkpoint-900/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:97378aa3f46334af7d68b1df94dcec829898e99b919b46b23aa3f18c26605155
3
+ size 343233204
checkpoint-900/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:02dd94822770e7d0788cfb5348d4f94869d38c7f06e0bdb5e90ec4498199d621
3
+ size 686587322
checkpoint-900/preprocessor_config.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_convert_rgb": null,
3
+ "do_normalize": true,
4
+ "do_rescale": true,
5
+ "do_resize": true,
6
+ "image_mean": [
7
+ 0.5,
8
+ 0.5,
9
+ 0.5
10
+ ],
11
+ "image_processor_type": "ViTImageProcessor",
12
+ "image_std": [
13
+ 0.5,
14
+ 0.5,
15
+ 0.5
16
+ ],
17
+ "resample": 2,
18
+ "rescale_factor": 0.00392156862745098,
19
+ "size": {
20
+ "height": 224,
21
+ "width": 224
22
+ }
23
+ }
checkpoint-900/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6ad3e8f94a9f005f2eed6113dacd099e50cea541cae98c1ef3160d770074a25e
3
+ size 14244
checkpoint-900/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9abefc2b0e41f3c2f92b5629b4653c5abd8bb183516d1147056e7ce663fafe54
3
+ size 1064
checkpoint-900/trainer_state.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 900,
3
+ "best_metric": 0.340586394071579,
4
+ "best_model_checkpoint": "siglip2-finetune-full/checkpoint-900",
5
+ "epoch": 1.0,
6
+ "eval_steps": 500,
7
+ "global_step": 900,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.5555555555555556,
14
+ "grad_norm": 2.6622252464294434,
15
+ "learning_rate": 1.4857142857142858e-05,
16
+ "loss": 0.6889,
17
+ "step": 500
18
+ },
19
+ {
20
+ "epoch": 1.0,
21
+ "eval_accuracy": 0.905307767125191,
22
+ "eval_loss": 0.340586394071579,
23
+ "eval_model_preparation_time": 0.0027,
24
+ "eval_runtime": 495.4956,
25
+ "eval_samples_per_second": 58.099,
26
+ "eval_steps_per_second": 7.263,
27
+ "step": 900
28
+ }
29
+ ],
30
+ "logging_steps": 500,
31
+ "max_steps": 1800,
32
+ "num_input_tokens_seen": 0,
33
+ "num_train_epochs": 2,
34
+ "save_steps": 500,
35
+ "stateful_callbacks": {
36
+ "TrainerControl": {
37
+ "args": {
38
+ "should_epoch_stop": false,
39
+ "should_evaluate": false,
40
+ "should_log": false,
41
+ "should_save": true,
42
+ "should_training_stop": false
43
+ },
44
+ "attributes": {}
45
+ }
46
+ },
47
+ "total_flos": 2.2308993799485604e+18,
48
+ "train_batch_size": 32,
49
+ "trial_name": null,
50
+ "trial_params": null
51
+ }
checkpoint-900/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:566011f5604bed061f658701cef40a0c715a83fe851a5dcfea9ade31ab9b8d4f
3
+ size 5304
config.json ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "ViTForImageClassification"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.0,
6
+ "encoder_stride": 16,
7
+ "hidden_act": "gelu",
8
+ "hidden_dropout_prob": 0.0,
9
+ "hidden_size": 768,
10
+ "id2label": {
11
+ "0": "Anime Picture",
12
+ "1": "Extincing & Sensual",
13
+ "2": "Hentai",
14
+ "3": "Pornography",
15
+ "4": "Safe for Work"
16
+ },
17
+ "image_size": 224,
18
+ "initializer_range": 0.02,
19
+ "intermediate_size": 3072,
20
+ "label2id": {
21
+ "Anime Picture": 0,
22
+ "Extincing & Sensual": 1,
23
+ "Hentai": 2,
24
+ "Pornography": 3,
25
+ "Safe for Work": 4
26
+ },
27
+ "layer_norm_eps": 1e-12,
28
+ "model_type": "vit",
29
+ "num_attention_heads": 12,
30
+ "num_channels": 3,
31
+ "num_hidden_layers": 12,
32
+ "patch_size": 16,
33
+ "pooler_act": "tanh",
34
+ "pooler_output_size": 768,
35
+ "problem_type": "single_label_classification",
36
+ "qkv_bias": true,
37
+ "torch_dtype": "float32",
38
+ "transformers_version": "4.50.0"
39
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:16902c62b2bb49952d5e9a848e68cb03252ae0894e916cf17f4c74783ba12c84
3
+ size 343233204
preprocessor_config.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_convert_rgb": null,
3
+ "do_normalize": true,
4
+ "do_rescale": true,
5
+ "do_resize": true,
6
+ "image_mean": [
7
+ 0.5,
8
+ 0.5,
9
+ 0.5
10
+ ],
11
+ "image_processor_type": "ViTImageProcessor",
12
+ "image_std": [
13
+ 0.5,
14
+ 0.5,
15
+ 0.5
16
+ ],
17
+ "resample": 2,
18
+ "rescale_factor": 0.00392156862745098,
19
+ "size": {
20
+ "height": 224,
21
+ "width": 224
22
+ }
23
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:566011f5604bed061f658701cef40a0c715a83fe851a5dcfea9ade31ab9b8d4f
3
+ size 5304