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
+---
+
+# Model Trained Using AutoTrain
+
+- Problem type: Image Classification
+
+## Validation Metricsg
+loss: 0.044754352420568466
+
+f1: 0.9852413242919825
+
+precision: 0.9975767366720517
+
+recall: 0.9732072498029944
+
+auc: 0.9994751008188858
+
+accuracy: 0.9864568081991215

checkpoint-1026/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-1026/model.safetensors

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

checkpoint-1026/optimizer.pt

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

checkpoint-1026/rng_state.pth

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

checkpoint-1026/scheduler.pt

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

checkpoint-1026/trainer_state.json

@@ -0,0 +1,150 @@
+{
+  "best_metric": 0.044754352420568466,
+  "best_model_checkpoint": "/tmp/model/checkpoint-1026",
+  "epoch": 3.0,
+  "eval_steps": 500,
+  "global_step": 1026,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.2,
+      "learning_rate": 3.155339805825243e-05,
+      "loss": 1.2096,
+      "step": 68
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 4.837486457204768e-05,
+      "loss": 0.2349,
+      "step": 136
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 4.469122426868906e-05,
+      "loss": 0.3095,
+      "step": 204
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 4.1007583965330447e-05,
+      "loss": 0.2824,
+      "step": 272
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 3.7323943661971835e-05,
+      "loss": 0.2215,
+      "step": 340
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.9751098096632503,
+      "eval_auc": 0.9967504189228713,
+      "eval_f1": 0.9735202492211837,
+      "eval_loss": 0.07951382547616959,
+      "eval_precision": 0.962278675904542,
+      "eval_recall": 0.9850275807722616,
+      "eval_runtime": 50.4793,
+      "eval_samples_per_second": 54.121,
+      "eval_steps_per_second": 3.388,
+      "step": 342
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 3.3640303358613216e-05,
+      "loss": 0.1574,
+      "step": 408
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 2.9956663055254608e-05,
+      "loss": 0.1602,
+      "step": 476
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 2.6273022751895993e-05,
+      "loss": 0.1773,
+      "step": 544
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 2.258938244853738e-05,
+      "loss": 0.1794,
+      "step": 612
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 1.8905742145178766e-05,
+      "loss": 0.1738,
+      "step": 680
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.9838945827232797,
+      "eval_auc": 0.9987853795244613,
+      "eval_f1": 0.9824281150159744,
+      "eval_loss": 0.04984632134437561,
+      "eval_precision": 0.9959514170040485,
+      "eval_recall": 0.9692671394799054,
+      "eval_runtime": 52.4203,
+      "eval_samples_per_second": 52.117,
+      "eval_steps_per_second": 3.262,
+      "step": 684
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 1.5222101841820153e-05,
+      "loss": 0.0957,
+      "step": 748
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 1.153846153846154e-05,
+      "loss": 0.1417,
+      "step": 816
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 7.854821235102926e-06,
+      "loss": 0.0833,
+      "step": 884
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 4.171180931744313e-06,
+      "loss": 0.0681,
+      "step": 952
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 4.875406283856988e-07,
+      "loss": 0.103,
+      "step": 1020
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.9864568081991215,
+      "eval_auc": 0.9994751008188858,
+      "eval_f1": 0.9852413242919825,
+      "eval_loss": 0.044754352420568466,
+      "eval_precision": 0.9975767366720517,
+      "eval_recall": 0.9732072498029944,
+      "eval_runtime": 52.4079,
+      "eval_samples_per_second": 52.13,
+      "eval_steps_per_second": 3.263,
+      "step": 1026
+    }
+  ],
+  "logging_steps": 68,
+  "max_steps": 1026,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 3,
+  "save_steps": 500,
+  "total_flos": 6.35124346881196e+17,
+  "train_batch_size": 8,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-1026/training_args.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9ab95df77aa73436641b5dbaf01dcf96ea93d00d8466ef621e283d5d829c6944
+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:95f614c565d449da2d2916798be7633e2c0167d050e474cb1f36a5094f97f5ec
+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:9ab95df77aa73436641b5dbaf01dcf96ea93d00d8466ef621e283d5d829c6944
+size 4728

training_params.json

@@ -0,0 +1,29 @@
+{
+    "data_path": "Zmu/autotrain-data-xcd_classification",
+    "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",
+    "evaluation_strategy": "epoch",
+    "image_column": "autotrain_image",
+    "target_column": "autotrain_label",
+    "log": "none"
+}