Upload folder using huggingface_hub

README.md

@@ -0,0 +1,32 @@
+
+---
+tags:
+- autotrain
+- image-classification
+widget:
+- src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/tiger.jpg
+  example_title: Tiger
+- src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/teapot.jpg
+  example_title: Teapot
+- src: https://huggingface.co/datasets/mishig/sample_images/resolve/main/palace.jpg
+  example_title: Palace
+datasets:
+- Zmu/autotrain-data-xcd_classification_v3
+---
+
+# Model Trained Using AutoTrain
+
+- Problem type: Image Classification
+
+## Validation Metricsg
+loss: 0.053496960550546646
+
+f1: 0.9824561403508772
+
+precision: 0.984182776801406
+
+recall: 0.9807355516637478
+
+auc: 0.9982940020845161
+
+accuracy: 0.9837133550488599

checkpoint-822/config.json

@@ -0,0 +1,33 @@
+{
+  "_name_or_path": "Falconsai/nsfw_image_detection",
+  "_num_labels": 2,
+  "architectures": [
+    "ViTForImageClassification"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "normal",
+    "1": "nsfw"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "normal": 0,
+    "nsfw": 1
+  },
+  "layer_norm_eps": 1e-12,
+  "model_type": "vit",
+  "num_attention_heads": 12,
+  "num_channels": 3,
+  "num_hidden_layers": 12,
+  "patch_size": 16,
+  "problem_type": "single_label_classification",
+  "qkv_bias": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.36.1"
+}

checkpoint-822/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4e7a453c30745c840bbf8fa0121268865c5ebd5730f10a4d74bcebee2fa0e34d
+size 343223968

checkpoint-822/optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:90296f6e4500993b78b2154cec5bd53d076772259d4d8586eacb25bd7187c8d4
+size 686568890

checkpoint-822/rng_state.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0cfd8968425398e6c4e30a086103bc348f08be517a952ca244a6aa581d0ddc9f
+size 14244

checkpoint-822/scheduler.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5a9099135d76dce235f5a4419261188303401431d8c1de0366b57fdde977d883
+size 1064

checkpoint-822/trainer_state.json

@@ -0,0 +1,222 @@
+{
+  "best_metric": 0.053496960550546646,
+  "best_model_checkpoint": "/tmp/model/checkpoint-822",
+  "epoch": 3.0,
+  "eval_steps": 500,
+  "global_step": 822,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.11,
+      "learning_rate": 1.6867469879518073e-05,
+      "loss": 1.7276,
+      "step": 30
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 3.4939759036144585e-05,
+      "loss": 0.4579,
+      "step": 60
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 4.96617050067659e-05,
+      "loss": 0.3088,
+      "step": 90
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 4.76319350473613e-05,
+      "loss": 0.35,
+      "step": 120
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 4.56021650879567e-05,
+      "loss": 0.2649,
+      "step": 150
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 4.35723951285521e-05,
+      "loss": 0.2502,
+      "step": 180
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 4.15426251691475e-05,
+      "loss": 0.3374,
+      "step": 210
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 3.95128552097429e-05,
+      "loss": 0.1944,
+      "step": 240
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 3.7483085250338296e-05,
+      "loss": 0.2182,
+      "step": 270
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.9739413680781759,
+      "eval_auc": 0.9947087408402573,
+      "eval_f1": 0.9719298245614035,
+      "eval_loss": 0.09450644999742508,
+      "eval_precision": 0.9736379613356766,
+      "eval_recall": 0.9702276707530648,
+      "eval_runtime": 22.7023,
+      "eval_samples_per_second": 54.091,
+      "eval_steps_per_second": 3.392,
+      "step": 274
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 3.5453315290933695e-05,
+      "loss": 0.2388,
+      "step": 300
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 3.3423545331529095e-05,
+      "loss": 0.1592,
+      "step": 330
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 3.1393775372124494e-05,
+      "loss": 0.1703,
+      "step": 360
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 2.9364005412719893e-05,
+      "loss": 0.2019,
+      "step": 390
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 2.7334235453315295e-05,
+      "loss": 0.2313,
+      "step": 420
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 2.530446549391069e-05,
+      "loss": 0.1362,
+      "step": 450
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 2.327469553450609e-05,
+      "loss": 0.1414,
+      "step": 480
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 2.124492557510149e-05,
+      "loss": 0.1077,
+      "step": 510
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 1.9215155615696888e-05,
+      "loss": 0.1434,
+      "step": 540
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.9723127035830619,
+      "eval_auc": 0.9975742842139215,
+      "eval_f1": 0.9707401032702238,
+      "eval_loss": 0.08003830909729004,
+      "eval_precision": 0.9543147208121827,
+      "eval_recall": 0.9877408056042032,
+      "eval_runtime": 23.305,
+      "eval_samples_per_second": 52.692,
+      "eval_steps_per_second": 3.304,
+      "step": 548
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 1.7185385656292287e-05,
+      "loss": 0.2318,
+      "step": 570
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 1.5155615696887688e-05,
+      "loss": 0.1714,
+      "step": 600
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 1.3125845737483087e-05,
+      "loss": 0.1123,
+      "step": 630
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 1.1096075778078486e-05,
+      "loss": 0.1017,
+      "step": 660
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 9.066305818673885e-06,
+      "loss": 0.1417,
+      "step": 690
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 7.036535859269283e-06,
+      "loss": 0.0908,
+      "step": 720
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 5.006765899864682e-06,
+      "loss": 0.1057,
+      "step": 750
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 2.9769959404600813e-06,
+      "loss": 0.1653,
+      "step": 780
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 1.0148849797023005e-06,
+      "loss": 0.0897,
+      "step": 810
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.9837133550488599,
+      "eval_auc": 0.9982940020845161,
+      "eval_f1": 0.9824561403508772,
+      "eval_loss": 0.053496960550546646,
+      "eval_precision": 0.984182776801406,
+      "eval_recall": 0.9807355516637478,
+      "eval_runtime": 22.6402,
+      "eval_samples_per_second": 54.24,
+      "eval_steps_per_second": 3.401,
+      "step": 822
+    }
+  ],
+  "logging_steps": 30,
+  "max_steps": 822,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 3,
+  "save_steps": 500,
+  "total_flos": 5.084249438613381e+17,
+  "train_batch_size": 8,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-822/training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aec843f599d276e6459acb5892c370f6c2c4cb17930075c7a54afe49ae46ac06
+size 4728

config.json

@@ -0,0 +1,33 @@
+{
+  "_name_or_path": "Falconsai/nsfw_image_detection",
+  "_num_labels": 2,
+  "architectures": [
+    "ViTForImageClassification"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "normal",
+    "1": "nsfw"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "normal": 0,
+    "nsfw": 1
+  },
+  "layer_norm_eps": 1e-12,
+  "model_type": "vit",
+  "num_attention_heads": 12,
+  "num_channels": 3,
+  "num_hidden_layers": 12,
+  "patch_size": 16,
+  "problem_type": "single_label_classification",
+  "qkv_bias": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.36.1"
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4e7a453c30745c840bbf8fa0121268865c5ebd5730f10a4d74bcebee2fa0e34d
+size 343223968

preprocessor_config.json

@@ -0,0 +1,22 @@
+{
+  "do_normalize": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "image_mean": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "image_processor_type": "ViTImageProcessor",
+  "image_std": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "resample": 2,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "height": 224,
+    "width": 224
+  }
+}

training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aec843f599d276e6459acb5892c370f6c2c4cb17930075c7a54afe49ae46ac06
+size 4728

training_params.json

@@ -0,0 +1,29 @@
+{
+    "data_path": "Zmu/autotrain-data-xcd_classification_v3",
+    "model": "Falconsai/nsfw_image_detection",
+    "username": "Zmu",
+    "lr": 5e-05,
+    "epochs": 3,
+    "batch_size": 8,
+    "warmup_ratio": 0.1,
+    "gradient_accumulation": 1,
+    "optimizer": "adamw_torch",
+    "scheduler": "linear",
+    "weight_decay": 0.0,
+    "max_grad_norm": 1.0,
+    "seed": 42,
+    "train_split": "train",
+    "valid_split": "validation",
+    "logging_steps": -1,
+    "project_name": "/tmp/model",
+    "auto_find_batch_size": false,
+    "mixed_precision": "fp16",
+    "save_total_limit": 1,
+    "save_strategy": "epoch",
+    "push_to_hub": true,
+    "repo_id": "Zmu/xcd_classification_v3",
+    "evaluation_strategy": "epoch",
+    "image_column": "autotrain_image",
+    "target_column": "autotrain_label",
+    "log": "none"
+}