Upload folder using huggingface_hub

checkpoint-930/config.json

@@ -0,0 +1,328 @@
+{
+  "_name_or_path": "dima806/pokemon_types_image_detection",
+  "architectures": [
+    "ViTForImageClassification"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "Wartortle",
+    "1": "Arcanine",
+    "2": "Staryu",
+    "3": "Arbok",
+    "4": "Butterfree",
+    "5": "Geodude",
+    "6": "Seaking",
+    "7": "Diglett",
+    "8": "Jynx",
+    "9": "Sandslash",
+    "10": "Magneton",
+    "11": "Scyther",
+    "12": "Kabuto",
+    "13": "Cubone",
+    "14": "Golem",
+    "15": "Dewgong",
+    "16": "Pidgey",
+    "17": "Kadabra",
+    "18": "Ditto",
+    "19": "Venomoth",
+    "20": "Rattata",
+    "21": "Alakazam",
+    "22": "Machoke",
+    "23": "Farfetchd",
+    "24": "Omastar",
+    "25": "Machamp",
+    "26": "Jigglypuff",
+    "27": "Dragonite",
+    "28": "Weepinbell",
+    "29": "Sandshrew",
+    "30": "Dugtrio",
+    "31": "Mankey",
+    "32": "Hitmonchan",
+    "33": "Spearow",
+    "34": "Caterpie",
+    "35": "Dratini",
+    "36": "Bulbasaur",
+    "37": "Tentacool",
+    "38": "Gengar",
+    "39": "Machop",
+    "40": "Raichu",
+    "41": "Alolan Sandslash",
+    "42": "Eevee",
+    "43": "Abra",
+    "44": "Haunter",
+    "45": "Metapod",
+    "46": "Fearow",
+    "47": "Nidorina",
+    "48": "Zapdos",
+    "49": "Ninetales",
+    "50": "Chansey",
+    "51": "Kangaskhan",
+    "52": "Poliwrath",
+    "53": "Gyarados",
+    "54": "Charmeleon",
+    "55": "Vulpix",
+    "56": "Pidgeot",
+    "57": "Blastoise",
+    "58": "Porygon",
+    "59": "Psyduck",
+    "60": "Dragonair",
+    "61": "Raticate",
+    "62": "Squirtle",
+    "63": "Charizard",
+    "64": "Electrode",
+    "65": "Flareon",
+    "66": "Exeggutor",
+    "67": "Pikachu",
+    "68": "Wigglytuff",
+    "69": "Venusaur",
+    "70": "Mewtwo",
+    "71": "Clefable",
+    "72": "Oddish",
+    "73": "Ekans",
+    "74": "Shellder",
+    "75": "Marowak",
+    "76": "Kakuna",
+    "77": "Rapidash",
+    "78": "Rhydon",
+    "79": "Ivysaur",
+    "80": "Slowpoke",
+    "81": "Lapras",
+    "82": "Clefairy",
+    "83": "Hitmonlee",
+    "84": "Jolteon",
+    "85": "Growlithe",
+    "86": "Gastly",
+    "87": "Aerodactyl",
+    "88": "Weedle",
+    "89": "Tauros",
+    "90": "Seel",
+    "91": "Zubat",
+    "92": "Meowth",
+    "93": "Persian",
+    "94": "Articuno",
+    "95": "Weezing",
+    "96": "Magnemite",
+    "97": "Omanyte",
+    "98": "Mew",
+    "99": "Vileplume",
+    "100": "Nidoqueen",
+    "101": "Vaporeon",
+    "102": "Ponyta",
+    "103": "Moltres",
+    "104": "Voltorb",
+    "105": "Magikarp",
+    "106": "Beedrill",
+    "107": "Nidoking",
+    "108": "Paras",
+    "109": "Grimer",
+    "110": "Dodrio",
+    "111": "Charmander",
+    "112": "Muk",
+    "113": "Primeape",
+    "114": "Victreebel",
+    "115": "Golbat",
+    "116": "Horsea",
+    "117": "Goldeen",
+    "118": "Pidgeotto",
+    "119": "Koffing",
+    "120": "Seadra",
+    "121": "Tentacruel",
+    "122": "Pinsir",
+    "123": "Cloyster",
+    "124": "Gloom",
+    "125": "Graveler",
+    "126": "Magmar",
+    "127": "Krabby",
+    "128": "Electabuzz",
+    "129": "Poliwhirl",
+    "130": "Golduck",
+    "131": "Onix",
+    "132": "Nidorino",
+    "133": "Snorlax",
+    "134": "Starmie",
+    "135": "Slowbro",
+    "136": "MrMime",
+    "137": "Venonat",
+    "138": "Kabutops",
+    "139": "Drowzee",
+    "140": "Rhyhorn",
+    "141": "Tangela",
+    "142": "Doduo",
+    "143": "Exeggcute",
+    "144": "Poliwag",
+    "145": "Lickitung",
+    "146": "Hypno",
+    "147": "Bellsprout",
+    "148": "Parasect",
+    "149": "Kingler"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "Abra": 43,
+    "Aerodactyl": 87,
+    "Alakazam": 21,
+    "Alolan Sandslash": 41,
+    "Arbok": 3,
+    "Arcanine": 1,
+    "Articuno": 94,
+    "Beedrill": 106,
+    "Bellsprout": 147,
+    "Blastoise": 57,
+    "Bulbasaur": 36,
+    "Butterfree": 4,
+    "Caterpie": 34,
+    "Chansey": 50,
+    "Charizard": 63,
+    "Charmander": 111,
+    "Charmeleon": 54,
+    "Clefable": 71,
+    "Clefairy": 82,
+    "Cloyster": 123,
+    "Cubone": 13,
+    "Dewgong": 15,
+    "Diglett": 7,
+    "Ditto": 18,
+    "Dodrio": 110,
+    "Doduo": 142,
+    "Dragonair": 60,
+    "Dragonite": 27,
+    "Dratini": 35,
+    "Drowzee": 139,
+    "Dugtrio": 30,
+    "Eevee": 42,
+    "Ekans": 73,
+    "Electabuzz": 128,
+    "Electrode": 64,
+    "Exeggcute": 143,
+    "Exeggutor": 66,
+    "Farfetchd": 23,
+    "Fearow": 46,
+    "Flareon": 65,
+    "Gastly": 86,
+    "Gengar": 38,
+    "Geodude": 5,
+    "Gloom": 124,
+    "Golbat": 115,
+    "Goldeen": 117,
+    "Golduck": 130,
+    "Golem": 14,
+    "Graveler": 125,
+    "Grimer": 109,
+    "Growlithe": 85,
+    "Gyarados": 53,
+    "Haunter": 44,
+    "Hitmonchan": 32,
+    "Hitmonlee": 83,
+    "Horsea": 116,
+    "Hypno": 146,
+    "Ivysaur": 79,
+    "Jigglypuff": 26,
+    "Jolteon": 84,
+    "Jynx": 8,
+    "Kabuto": 12,
+    "Kabutops": 138,
+    "Kadabra": 17,
+    "Kakuna": 76,
+    "Kangaskhan": 51,
+    "Kingler": 149,
+    "Koffing": 119,
+    "Krabby": 127,
+    "Lapras": 81,
+    "Lickitung": 145,
+    "Machamp": 25,
+    "Machoke": 22,
+    "Machop": 39,
+    "Magikarp": 105,
+    "Magmar": 126,
+    "Magnemite": 96,
+    "Magneton": 10,
+    "Mankey": 31,
+    "Marowak": 75,
+    "Meowth": 92,
+    "Metapod": 45,
+    "Mew": 98,
+    "Mewtwo": 70,
+    "Moltres": 103,
+    "MrMime": 136,
+    "Muk": 112,
+    "Nidoking": 107,
+    "Nidoqueen": 100,
+    "Nidorina": 47,
+    "Nidorino": 132,
+    "Ninetales": 49,
+    "Oddish": 72,
+    "Omanyte": 97,
+    "Omastar": 24,
+    "Onix": 131,
+    "Paras": 108,
+    "Parasect": 148,
+    "Persian": 93,
+    "Pidgeot": 56,
+    "Pidgeotto": 118,
+    "Pidgey": 16,
+    "Pikachu": 67,
+    "Pinsir": 122,
+    "Poliwag": 144,
+    "Poliwhirl": 129,
+    "Poliwrath": 52,
+    "Ponyta": 102,
+    "Porygon": 58,
+    "Primeape": 113,
+    "Psyduck": 59,
+    "Raichu": 40,
+    "Rapidash": 77,
+    "Raticate": 61,
+    "Rattata": 20,
+    "Rhydon": 78,
+    "Rhyhorn": 140,
+    "Sandshrew": 29,
+    "Sandslash": 9,
+    "Scyther": 11,
+    "Seadra": 120,
+    "Seaking": 6,
+    "Seel": 90,
+    "Shellder": 74,
+    "Slowbro": 135,
+    "Slowpoke": 80,
+    "Snorlax": 133,
+    "Spearow": 33,
+    "Squirtle": 62,
+    "Starmie": 134,
+    "Staryu": 2,
+    "Tangela": 141,
+    "Tauros": 89,
+    "Tentacool": 37,
+    "Tentacruel": 121,
+    "Vaporeon": 101,
+    "Venomoth": 19,
+    "Venonat": 137,
+    "Venusaur": 69,
+    "Victreebel": 114,
+    "Vileplume": 99,
+    "Voltorb": 104,
+    "Vulpix": 55,
+    "Wartortle": 0,
+    "Weedle": 88,
+    "Weepinbell": 28,
+    "Weezing": 95,
+    "Wigglytuff": 68,
+    "Zapdos": 48,
+    "Zubat": 91
+  },
+  "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.39.2"
+}

checkpoint-930/model.safetensors

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

checkpoint-930/optimizer.pt

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

checkpoint-930/preprocessor_config.json

@@ -0,0 +1,36 @@
+{
+  "_valid_processor_keys": [
+    "images",
+    "do_resize",
+    "size",
+    "resample",
+    "do_rescale",
+    "rescale_factor",
+    "do_normalize",
+    "image_mean",
+    "image_std",
+    "return_tensors",
+    "data_format",
+    "input_data_format"
+  ],
+  "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
+  }
+}

checkpoint-930/rng_state.pth

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

checkpoint-930/scheduler.pt

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

checkpoint-930/trainer_state.json

@@ -0,0 +1,73 @@
+{
+  "best_metric": 0.9411983489990234,
+  "best_model_checkpoint": "pokemon_types_image_detection/checkpoint-930",
+  "epoch": 5.0,
+  "eval_steps": 500,
+  "global_step": 930,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.9583333333333334,
+      "eval_loss": 0.9651849269866943,
+      "eval_runtime": 51.6263,
+      "eval_samples_per_second": 76.705,
+      "eval_steps_per_second": 9.588,
+      "step": 186
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.9585858585858585,
+      "eval_loss": 0.9547479748725891,
+      "eval_runtime": 51.7206,
+      "eval_samples_per_second": 76.565,
+      "eval_steps_per_second": 9.571,
+      "step": 372
+    },
+    {
+      "epoch": 2.69,
+      "grad_norm": 1.6661704778671265,
+      "learning_rate": 4.886363636363636e-07,
+      "loss": 1.0008,
+      "step": 500
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy": 0.9588383838383838,
+      "eval_loss": 0.9472126364707947,
+      "eval_runtime": 51.7404,
+      "eval_samples_per_second": 76.536,
+      "eval_steps_per_second": 9.567,
+      "step": 558
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy": 0.9588383838383838,
+      "eval_loss": 0.9425832033157349,
+      "eval_runtime": 51.3176,
+      "eval_samples_per_second": 77.166,
+      "eval_steps_per_second": 9.646,
+      "step": 744
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy": 0.9588383838383838,
+      "eval_loss": 0.9411983489990234,
+      "eval_runtime": 51.6442,
+      "eval_samples_per_second": 76.678,
+      "eval_steps_per_second": 9.585,
+      "step": 930
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 930,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 5,
+  "save_steps": 500,
+  "total_flos": 2.3045649948389376e+18,
+  "train_batch_size": 32,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-930/training_args.bin

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

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "google/vit-base-patch16-224-in21k",
+  "_name_or_path": "dima806/pokemon_types_image_detection",
   "architectures": [
     "ViTForImageClassification"
   ],
@@ -324,5 +324,5 @@
   "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.33.3"
+  "transformers_version": "4.39.2"
 }

model.safetensors

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

preprocessor_config.json

@@ -1,4 +1,18 @@
 {
+  "_valid_processor_keys": [
+    "images",
+    "do_resize",
+    "size",
+    "resample",
+    "do_rescale",
+    "rescale_factor",
+    "do_normalize",
+    "image_mean",
+    "image_std",
+    "return_tensors",
+    "data_format",
+    "input_data_format"
+  ],
   "do_normalize": true,
   "do_rescale": true,
   "do_resize": true,

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5eb2aa0893f36bd4bccb5cedd2df63d8fd8a20363e40239439d362401df751c6
-size 4027
+oid sha256:9aa2e336d9398f0df1f7ee740816bdc17f00243e383bf6d349a087b7f7b38b73
+size 4411