commit files to HF hub

config.json

@@ -0,0 +1,226 @@
+{
+  "_name_or_path": "google/vit-base-patch16-224-in21k",
+  "architectures": [
+    "ViTForImageClassification"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "apple",
+    "1": "aquarium_fish",
+    "2": "baby",
+    "3": "bear",
+    "4": "beaver",
+    "5": "bed",
+    "6": "bee",
+    "7": "beetle",
+    "8": "bicycle",
+    "9": "bottle",
+    "10": "bowl",
+    "11": "boy",
+    "12": "bridge",
+    "13": "bus",
+    "14": "butterfly",
+    "15": "camel",
+    "16": "can",
+    "17": "castle",
+    "18": "caterpillar",
+    "19": "cattle",
+    "20": "chair",
+    "21": "chimpanzee",
+    "22": "clock",
+    "23": "cloud",
+    "24": "cockroach",
+    "25": "couch",
+    "26": "cra",
+    "27": "crocodile",
+    "28": "cup",
+    "29": "dinosaur",
+    "30": "dolphin",
+    "31": "elephant",
+    "32": "flatfish",
+    "33": "forest",
+    "34": "fox",
+    "35": "girl",
+    "36": "hamster",
+    "37": "house",
+    "38": "kangaroo",
+    "39": "keyboard",
+    "40": "lamp",
+    "41": "lawn_mower",
+    "42": "leopard",
+    "43": "lion",
+    "44": "lizard",
+    "45": "lobster",
+    "46": "man",
+    "47": "maple_tree",
+    "48": "motorcycle",
+    "49": "mountain",
+    "50": "mouse",
+    "51": "mushroom",
+    "52": "oak_tree",
+    "53": "orange",
+    "54": "orchid",
+    "55": "otter",
+    "56": "palm_tree",
+    "57": "pear",
+    "58": "pickup_truck",
+    "59": "pine_tree",
+    "60": "plain",
+    "61": "plate",
+    "62": "poppy",
+    "63": "porcupine",
+    "64": "possum",
+    "65": "rabbit",
+    "66": "raccoon",
+    "67": "ray",
+    "68": "road",
+    "69": "rocket",
+    "70": "rose",
+    "71": "sea",
+    "72": "seal",
+    "73": "shark",
+    "74": "shrew",
+    "75": "skunk",
+    "76": "skyscraper",
+    "77": "snail",
+    "78": "snake",
+    "79": "spider",
+    "80": "squirrel",
+    "81": "streetcar",
+    "82": "sunflower",
+    "83": "sweet_pepper",
+    "84": "table",
+    "85": "tank",
+    "86": "telephone",
+    "87": "television",
+    "88": "tiger",
+    "89": "tractor",
+    "90": "train",
+    "91": "trout",
+    "92": "tulip",
+    "93": "turtle",
+    "94": "wardrobe",
+    "95": "whale",
+    "96": "willow_tree",
+    "97": "wolf",
+    "98": "woman",
+    "99": "worm"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "apple": 0,
+    "aquarium_fish": 1,
+    "baby": 2,
+    "bear": 3,
+    "beaver": 4,
+    "bed": 5,
+    "bee": 6,
+    "beetle": 7,
+    "bicycle": 8,
+    "bottle": 9,
+    "bowl": 10,
+    "boy": 11,
+    "bridge": 12,
+    "bus": 13,
+    "butterfly": 14,
+    "camel": 15,
+    "can": 16,
+    "castle": 17,
+    "caterpillar": 18,
+    "cattle": 19,
+    "chair": 20,
+    "chimpanzee": 21,
+    "clock": 22,
+    "cloud": 23,
+    "cockroach": 24,
+    "couch": 25,
+    "cra": 26,
+    "crocodile": 27,
+    "cup": 28,
+    "dinosaur": 29,
+    "dolphin": 30,
+    "elephant": 31,
+    "flatfish": 32,
+    "forest": 33,
+    "fox": 34,
+    "girl": 35,
+    "hamster": 36,
+    "house": 37,
+    "kangaroo": 38,
+    "keyboard": 39,
+    "lamp": 40,
+    "lawn_mower": 41,
+    "leopard": 42,
+    "lion": 43,
+    "lizard": 44,
+    "lobster": 45,
+    "man": 46,
+    "maple_tree": 47,
+    "motorcycle": 48,
+    "mountain": 49,
+    "mouse": 50,
+    "mushroom": 51,
+    "oak_tree": 52,
+    "orange": 53,
+    "orchid": 54,
+    "otter": 55,
+    "palm_tree": 56,
+    "pear": 57,
+    "pickup_truck": 58,
+    "pine_tree": 59,
+    "plain": 60,
+    "plate": 61,
+    "poppy": 62,
+    "porcupine": 63,
+    "possum": 64,
+    "rabbit": 65,
+    "raccoon": 66,
+    "ray": 67,
+    "road": 68,
+    "rocket": 69,
+    "rose": 70,
+    "sea": 71,
+    "seal": 72,
+    "shark": 73,
+    "shrew": 74,
+    "skunk": 75,
+    "skyscraper": 76,
+    "snail": 77,
+    "snake": 78,
+    "spider": 79,
+    "squirrel": 80,
+    "streetcar": 81,
+    "sunflower": 82,
+    "sweet_pepper": 83,
+    "table": 84,
+    "tank": 85,
+    "telephone": 86,
+    "television": 87,
+    "tiger": 88,
+    "tractor": 89,
+    "train": 90,
+    "trout": 91,
+    "tulip": 92,
+    "turtle": 93,
+    "wardrobe": 94,
+    "whale": 95,
+    "willow_tree": 96,
+    "wolf": 97,
+    "woman": 98,
+    "worm": 99
+  },
+  "layer_norm_eps": 1e-12,
+  "model_type": "vit",
+  "num_attention_heads": 12,
+  "num_channels": 3,
+  "num_hidden_layers": 12,
+  "patch_size": 16,
+  "qkv_bias": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.12.3"
+}

eval_results.txt

@@ -0,0 +1,6 @@
+epoch = 5.0
+eval_accuracy = 0.9148
+eval_loss = 0.38199177384376526
+eval_runtime = 200.9642
+eval_samples_per_second = 49.76
+eval_steps_per_second = 0.781

pytorch_model.bin

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

training_args.bin

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