Upload folder using huggingface_hub

checkpoint-402/config.json

@@ -0,0 +1,118 @@
+{
+  "_name_or_path": "dima806/mammals_45_types_image_classification",
+  "architectures": [
+    "ViTForImageClassification"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "african_elephant",
+    "1": "alpaca",
+    "2": "american_bison",
+    "3": "anteater",
+    "4": "arctic_fox",
+    "5": "armadillo",
+    "6": "baboon",
+    "7": "badger",
+    "8": "blue_whale",
+    "9": "brown_bear",
+    "10": "camel",
+    "11": "dolphin",
+    "12": "giraffe",
+    "13": "groundhog",
+    "14": "highland_cattle",
+    "15": "horse",
+    "16": "jackal",
+    "17": "kangaroo",
+    "18": "koala",
+    "19": "manatee",
+    "20": "mongoose",
+    "21": "mountain_goat",
+    "22": "opossum",
+    "23": "orangutan",
+    "24": "otter",
+    "25": "polar_bear",
+    "26": "porcupine",
+    "27": "red_panda",
+    "28": "rhinoceros",
+    "29": "sea_lion",
+    "30": "seal",
+    "31": "snow_leopard",
+    "32": "squirrel",
+    "33": "sugar_glider",
+    "34": "tapir",
+    "35": "vampire_bat",
+    "36": "vicuna",
+    "37": "walrus",
+    "38": "warthog",
+    "39": "water_buffalo",
+    "40": "weasel",
+    "41": "wildebeest",
+    "42": "wombat",
+    "43": "yak",
+    "44": "zebra"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "african_elephant": 0,
+    "alpaca": 1,
+    "american_bison": 2,
+    "anteater": 3,
+    "arctic_fox": 4,
+    "armadillo": 5,
+    "baboon": 6,
+    "badger": 7,
+    "blue_whale": 8,
+    "brown_bear": 9,
+    "camel": 10,
+    "dolphin": 11,
+    "giraffe": 12,
+    "groundhog": 13,
+    "highland_cattle": 14,
+    "horse": 15,
+    "jackal": 16,
+    "kangaroo": 17,
+    "koala": 18,
+    "manatee": 19,
+    "mongoose": 20,
+    "mountain_goat": 21,
+    "opossum": 22,
+    "orangutan": 23,
+    "otter": 24,
+    "polar_bear": 25,
+    "porcupine": 26,
+    "red_panda": 27,
+    "rhinoceros": 28,
+    "sea_lion": 29,
+    "seal": 30,
+    "snow_leopard": 31,
+    "squirrel": 32,
+    "sugar_glider": 33,
+    "tapir": 34,
+    "vampire_bat": 35,
+    "vicuna": 36,
+    "walrus": 37,
+    "warthog": 38,
+    "water_buffalo": 39,
+    "weasel": 40,
+    "wildebeest": 41,
+    "wombat": 42,
+    "yak": 43,
+    "zebra": 44
+  },
+  "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.2"
+}

checkpoint-402/model.safetensors

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

checkpoint-402/optimizer.pt

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

checkpoint-402/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
+  }
+}

checkpoint-402/rng_state.pth

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

checkpoint-402/scheduler.pt

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

checkpoint-402/trainer_state.json

@@ -0,0 +1,39 @@
+{
+  "best_metric": 2.867035388946533,
+  "best_model_checkpoint": "mammals_45_types_image_classification/checkpoint-402",
+  "epoch": 2.0,
+  "eval_steps": 500,
+  "global_step": 402,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "eval_accuracy": 0.9572409488139825,
+      "eval_loss": 2.875110626220703,
+      "eval_runtime": 31.3806,
+      "eval_samples_per_second": 102.101,
+      "eval_steps_per_second": 3.219,
+      "step": 201
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy": 0.9572409488139825,
+      "eval_loss": 2.867035388946533,
+      "eval_runtime": 31.3219,
+      "eval_samples_per_second": 102.293,
+      "eval_steps_per_second": 3.225,
+      "step": 402
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 402,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 2,
+  "save_steps": 500,
+  "total_flos": 1.987040178923176e+18,
+  "train_batch_size": 64,
+  "trial_name": null,
+  "trial_params": null
+}

checkpoint-402/training_args.bin

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

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "google/vit-base-patch16-224-in21k",
+  "_name_or_path": "dima806/mammals_45_types_image_classification",
   "architectures": [
     "ViTForImageClassification"
   ],
@@ -114,5 +114,5 @@
   "problem_type": "single_label_classification",
   "qkv_bias": true,
   "torch_dtype": "float32",
-  "transformers_version": "4.35.2"
+  "transformers_version": "4.38.2"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e336fe2f266eac9c541f7cd98489c16682b0b349fa876705809f69df2e8da690
+oid sha256:ae3c72b717e991b7e3dc8f993732e330d0c05bf6b9c4b748297b380a0b159f63
 size 343356252

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5197112e7aafa3b6ccf7ea7d30b828ea9762f503efb192189d71522ef61f01e3
-size 4091
+oid sha256:f83e92d7633d75dd3622a9df2270651c30c01b34d5de896dc7e6559db58cea23
+size 4411