Upload folder using huggingface_hub

README.md

@@ -0,0 +1,42 @@
+
+---
+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:
+- real-or-anime-age/autotrain-data
+---
+
+# Model Trained Using AutoTrain
+
+- Problem type: Image Classification
+
+## Validation Metricsg
+loss: 0.4898335635662079
+
+f1_macro: 0.7652092723432848
+
+f1_micro: 0.7869487485101311
+
+f1_weighted: 0.7873025165202922
+
+precision_macro: 0.7664681493316189
+
+precision_micro: 0.7869487485101311
+
+precision_weighted: 0.7878104453780785
+
+recall_macro: 0.7640948741007194
+
+recall_micro: 0.7869487485101311
+
+recall_weighted: 0.7869487485101311
+
+accuracy: 0.7869487485101311

checkpoint-3356/config.json

@@ -0,0 +1,37 @@
+{
+  "_name_or_path": "google/vit-base-patch16-224",
+  "_num_labels": 4,
+  "architectures": [
+    "ViTForImageClassification"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "anime_adult",
+    "1": "anime_contentious",
+    "2": "anime_non_adult",
+    "3": "non_anime"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "anime_adult": 0,
+    "anime_contentious": 1,
+    "anime_non_adult": 2,
+    "non_anime": 3
+  },
+  "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.38.0"
+}

checkpoint-3356/model.safetensors

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

checkpoint-3356/optimizer.pt

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

checkpoint-3356/rng_state.pth

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

checkpoint-3356/scheduler.pt

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

checkpoint-3356/trainer_state.json

@@ -0,0 +1,337 @@
+{
+  "best_metric": 0.4898335635662079,
+  "best_model_checkpoint": "real-or-anime-age/checkpoint-3356",
+  "epoch": 2.0,
+  "eval_steps": 500,
+  "global_step": 3356,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.05,
+      "grad_norm": 9.71189022064209,
+      "learning_rate": 8.035714285714286e-06,
+      "loss": 1.379,
+      "step": 83
+    },
+    {
+      "epoch": 0.1,
+      "grad_norm": 7.463231086730957,
+      "learning_rate": 1.626984126984127e-05,
+      "loss": 0.9415,
+      "step": 166
+    },
+    {
+      "epoch": 0.15,
+      "grad_norm": 9.226115226745605,
+      "learning_rate": 2.4503968253968254e-05,
+      "loss": 0.7884,
+      "step": 249
+    },
+    {
+      "epoch": 0.2,
+      "grad_norm": 8.683225631713867,
+      "learning_rate": 3.273809523809524e-05,
+      "loss": 0.7458,
+      "step": 332
+    },
+    {
+      "epoch": 0.25,
+      "grad_norm": 8.938767433166504,
+      "learning_rate": 4.0972222222222225e-05,
+      "loss": 0.7075,
+      "step": 415
+    },
+    {
+      "epoch": 0.3,
+      "grad_norm": 7.352237224578857,
+      "learning_rate": 4.9206349206349204e-05,
+      "loss": 0.7246,
+      "step": 498
+    },
+    {
+      "epoch": 0.35,
+      "grad_norm": 11.598906517028809,
+      "learning_rate": 4.917218543046358e-05,
+      "loss": 0.6635,
+      "step": 581
+    },
+    {
+      "epoch": 0.4,
+      "grad_norm": 5.454655170440674,
+      "learning_rate": 4.8256070640176604e-05,
+      "loss": 0.6538,
+      "step": 664
+    },
+    {
+      "epoch": 0.45,
+      "grad_norm": 4.765255451202393,
+      "learning_rate": 4.7339955849889624e-05,
+      "loss": 0.6486,
+      "step": 747
+    },
+    {
+      "epoch": 0.49,
+      "grad_norm": 9.992238998413086,
+      "learning_rate": 4.642384105960265e-05,
+      "loss": 0.7404,
+      "step": 830
+    },
+    {
+      "epoch": 0.54,
+      "grad_norm": 8.219937324523926,
+      "learning_rate": 4.5518763796909496e-05,
+      "loss": 0.6858,
+      "step": 913
+    },
+    {
+      "epoch": 0.59,
+      "grad_norm": 6.649818420410156,
+      "learning_rate": 4.460264900662252e-05,
+      "loss": 0.6612,
+      "step": 996
+    },
+    {
+      "epoch": 0.64,
+      "grad_norm": 5.724815845489502,
+      "learning_rate": 4.3686534216335544e-05,
+      "loss": 0.62,
+      "step": 1079
+    },
+    {
+      "epoch": 0.69,
+      "grad_norm": 3.1197288036346436,
+      "learning_rate": 4.278145695364239e-05,
+      "loss": 0.6113,
+      "step": 1162
+    },
+    {
+      "epoch": 0.74,
+      "grad_norm": 6.231707572937012,
+      "learning_rate": 4.186534216335541e-05,
+      "loss": 0.6624,
+      "step": 1245
+    },
+    {
+      "epoch": 0.79,
+      "grad_norm": 3.7672340869903564,
+      "learning_rate": 4.0949227373068436e-05,
+      "loss": 0.6205,
+      "step": 1328
+    },
+    {
+      "epoch": 0.84,
+      "grad_norm": 1.8633880615234375,
+      "learning_rate": 4.003311258278146e-05,
+      "loss": 0.6535,
+      "step": 1411
+    },
+    {
+      "epoch": 0.89,
+      "grad_norm": 5.430916786193848,
+      "learning_rate": 3.911699779249448e-05,
+      "loss": 0.6112,
+      "step": 1494
+    },
+    {
+      "epoch": 0.94,
+      "grad_norm": 6.332820415496826,
+      "learning_rate": 3.8200883002207506e-05,
+      "loss": 0.6444,
+      "step": 1577
+    },
+    {
+      "epoch": 0.99,
+      "grad_norm": 2.2511579990386963,
+      "learning_rate": 3.728476821192053e-05,
+      "loss": 0.6151,
+      "step": 1660
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.7377830750893921,
+      "eval_f1_macro": 0.717936749616658,
+      "eval_f1_micro": 0.7377830750893921,
+      "eval_f1_weighted": 0.7408818935906496,
+      "eval_loss": 0.5908161401748657,
+      "eval_precision_macro": 0.7272555863618316,
+      "eval_precision_micro": 0.7377830750893921,
+      "eval_precision_weighted": 0.7562177850512167,
+      "eval_recall_macro": 0.7239640287769784,
+      "eval_recall_micro": 0.7377830750893921,
+      "eval_recall_weighted": 0.7377830750893921,
+      "eval_runtime": 56.2486,
+      "eval_samples_per_second": 59.664,
+      "eval_steps_per_second": 3.733,
+      "step": 1678
+    },
+    {
+      "epoch": 1.04,
+      "grad_norm": 4.224277973175049,
+      "learning_rate": 3.636865342163356e-05,
+      "loss": 0.5562,
+      "step": 1743
+    },
+    {
+      "epoch": 1.09,
+      "grad_norm": 4.652990341186523,
+      "learning_rate": 3.545253863134658e-05,
+      "loss": 0.5764,
+      "step": 1826
+    },
+    {
+      "epoch": 1.14,
+      "grad_norm": 3.9056460857391357,
+      "learning_rate": 3.453642384105961e-05,
+      "loss": 0.6014,
+      "step": 1909
+    },
+    {
+      "epoch": 1.19,
+      "grad_norm": 6.200495719909668,
+      "learning_rate": 3.362030905077263e-05,
+      "loss": 0.5248,
+      "step": 1992
+    },
+    {
+      "epoch": 1.24,
+      "grad_norm": 5.531810283660889,
+      "learning_rate": 3.270419426048565e-05,
+      "loss": 0.5606,
+      "step": 2075
+    },
+    {
+      "epoch": 1.29,
+      "grad_norm": 7.862312316894531,
+      "learning_rate": 3.178807947019868e-05,
+      "loss": 0.5808,
+      "step": 2158
+    },
+    {
+      "epoch": 1.34,
+      "grad_norm": 6.051772117614746,
+      "learning_rate": 3.08719646799117e-05,
+      "loss": 0.5093,
+      "step": 2241
+    },
+    {
+      "epoch": 1.38,
+      "grad_norm": 4.436023712158203,
+      "learning_rate": 2.9955849889624728e-05,
+      "loss": 0.6025,
+      "step": 2324
+    },
+    {
+      "epoch": 1.43,
+      "grad_norm": 5.660228252410889,
+      "learning_rate": 2.903973509933775e-05,
+      "loss": 0.5558,
+      "step": 2407
+    },
+    {
+      "epoch": 1.48,
+      "grad_norm": 3.855027914047241,
+      "learning_rate": 2.8123620309050773e-05,
+      "loss": 0.5865,
+      "step": 2490
+    },
+    {
+      "epoch": 1.53,
+      "grad_norm": 3.9190945625305176,
+      "learning_rate": 2.7207505518763797e-05,
+      "loss": 0.5439,
+      "step": 2573
+    },
+    {
+      "epoch": 1.58,
+      "grad_norm": 5.566714763641357,
+      "learning_rate": 2.6291390728476822e-05,
+      "loss": 0.5622,
+      "step": 2656
+    },
+    {
+      "epoch": 1.63,
+      "grad_norm": 5.549644470214844,
+      "learning_rate": 2.5375275938189846e-05,
+      "loss": 0.5273,
+      "step": 2739
+    },
+    {
+      "epoch": 1.68,
+      "grad_norm": 4.814943790435791,
+      "learning_rate": 2.445916114790287e-05,
+      "loss": 0.5809,
+      "step": 2822
+    },
+    {
+      "epoch": 1.73,
+      "grad_norm": 5.472110271453857,
+      "learning_rate": 2.3543046357615895e-05,
+      "loss": 0.5928,
+      "step": 2905
+    },
+    {
+      "epoch": 1.78,
+      "grad_norm": 5.81935453414917,
+      "learning_rate": 2.262693156732892e-05,
+      "loss": 0.5379,
+      "step": 2988
+    },
+    {
+      "epoch": 1.83,
+      "grad_norm": 3.03950834274292,
+      "learning_rate": 2.1710816777041943e-05,
+      "loss": 0.545,
+      "step": 3071
+    },
+    {
+      "epoch": 1.88,
+      "grad_norm": 5.148889064788818,
+      "learning_rate": 2.0794701986754968e-05,
+      "loss": 0.585,
+      "step": 3154
+    },
+    {
+      "epoch": 1.93,
+      "grad_norm": 3.934765100479126,
+      "learning_rate": 1.9878587196467992e-05,
+      "loss": 0.5071,
+      "step": 3237
+    },
+    {
+      "epoch": 1.98,
+      "grad_norm": 5.576645374298096,
+      "learning_rate": 1.8962472406181016e-05,
+      "loss": 0.5581,
+      "step": 3320
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.7869487485101311,
+      "eval_f1_macro": 0.7652092723432848,
+      "eval_f1_micro": 0.7869487485101311,
+      "eval_f1_weighted": 0.7873025165202922,
+      "eval_loss": 0.4898335635662079,
+      "eval_precision_macro": 0.7664681493316189,
+      "eval_precision_micro": 0.7869487485101311,
+      "eval_precision_weighted": 0.7878104453780785,
+      "eval_recall_macro": 0.7640948741007194,
+      "eval_recall_micro": 0.7869487485101311,
+      "eval_recall_weighted": 0.7869487485101311,
+      "eval_runtime": 57.5629,
+      "eval_samples_per_second": 58.301,
+      "eval_steps_per_second": 3.648,
+      "step": 3356
+    }
+  ],
+  "logging_steps": 83,
+  "max_steps": 5034,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 3,
+  "save_steps": 500,
+  "total_flos": 2.0803872442305577e+18,
+  "train_batch_size": 8,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-3356/training_args.bin

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

config.json

@@ -0,0 +1,37 @@
+{
+  "_name_or_path": "google/vit-base-patch16-224",
+  "_num_labels": 4,
+  "architectures": [
+    "ViTForImageClassification"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "anime_adult",
+    "1": "anime_contentious",
+    "2": "anime_non_adult",
+    "3": "non_anime"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "anime_adult": 0,
+    "anime_contentious": 1,
+    "anime_non_adult": 2,
+    "non_anime": 3
+  },
+  "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.38.0"
+}

model.safetensors

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

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:a9624a1acf1caf112746e8b6c8308dcfd92635aa36074922449d4a2e1709e497
+size 4920

training_params.json

@@ -0,0 +1,29 @@
+{
+    "data_path": "real-or-anime-age/autotrain-data",
+    "model": "google/vit-base-patch16-224",
+    "username": "antonioglass",
+    "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": "real-or-anime-age",
+    "auto_find_batch_size": false,
+    "mixed_precision": "fp16",
+    "save_total_limit": 1,
+    "save_strategy": "epoch",
+    "push_to_hub": true,
+    "repo_id": "antonioglass/real-or-anime-age",
+    "evaluation_strategy": "epoch",
+    "image_column": "autotrain_image",
+    "target_column": "autotrain_label",
+    "log": "none"
+}