arieg
/

food

Image Classification

Transformers

TensorFlow

vit

generated_from_keras_callback

Inference Endpoints

Model card Files Files and versions Community

arieg commited on Nov 9, 2023

Commit

118d2fe

1 Parent(s): 50b16b1

End of training

Browse files

Files changed (3) hide show

README.md +9 -9
config.json +160 -202
tf_model.h5 +2 -2

README.md CHANGED Viewed

@@ -15,9 +15,9 @@ probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Train Loss: 0.2861
-- Validation Loss: 0.2612
-- Train Accuracy: 0.929
 - Epoch: 4
 ## Model description
@@ -37,18 +37,18 @@ More information needed
 ### Training hyperparameters
 The following hyperparameters were used during training:
-- optimizer: {'name': 'AdamWeightDecay', 'clipnorm': 1.0, 'learning_rate': {'module': 'keras.optimizers.schedules', 'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 3e-05, 'decay_steps': 20000, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}, 'registered_name': None}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.01}
 - training_precision: float32
 ### Training results
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
-| 2.7216     | 1.6098          | 0.819          | 0     |
-| 1.1279     | 0.7218          | 0.899          | 1     |
-| 0.5720     | 0.4555          | 0.929          | 2     |
-| 0.3861     | 0.3125          | 0.923          | 3     |
-| 0.2861     | 0.2612          | 0.929          | 4     |
 ### Framework versions

 This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
 It achieves the following results on the evaluation set:
+- Train Loss: 2.2825
+- Validation Loss: 2.2431
+- Train Accuracy: 0.975
 - Epoch: 4
 ## Model description
 ### Training hyperparameters
 The following hyperparameters were used during training:
+- optimizer: {'name': 'AdamWeightDecay', 'clipnorm': 1.0, 'learning_rate': {'module': 'keras.optimizers.schedules', 'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 3e-05, 'decay_steps': 7200, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}, 'registered_name': None}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.01}
 - training_precision: float32
 ### Training results
 | Train Loss | Validation Loss | Train Accuracy | Epoch |
 |:----------:|:---------------:|:--------------:|:-----:|
+| 4.1953     | 3.9579          | 0.6562         | 0     |
+| 3.6362     | 3.4187          | 0.925          | 1     |
+| 3.1017     | 2.9427          | 0.9625         | 2     |
+| 2.6561     | 2.5775          | 0.9688         | 3     |
+| 2.2825     | 2.2431          | 0.975          | 4     |
 ### Framework versions

config.json CHANGED Viewed

@@ -9,213 +9,171 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "apple_pie",
-    "1": "baby_back_ribs",
-    "10": "bruschetta",
-    "100": "waffles",
-    "11": "caesar_salad",
-    "12": "cannoli",
-    "13": "caprese_salad",
-    "14": "carrot_cake",
-    "15": "ceviche",
-    "16": "cheesecake",
-    "17": "cheese_plate",
-    "18": "chicken_curry",
-    "19": "chicken_quesadilla",
-    "2": "baklava",
-    "20": "chicken_wings",
-    "21": "chocolate_cake",
-    "22": "chocolate_mousse",
-    "23": "churros",
-    "24": "clam_chowder",
-    "25": "club_sandwich",
-    "26": "crab_cakes",
-    "27": "creme_brulee",
-    "28": "croque_madame",
-    "29": "cup_cakes",
-    "3": "beef_carpaccio",
-    "30": "deviled_eggs",
-    "31": "donuts",
-    "32": "dumplings",
-    "33": "edamame",
-    "34": "eggs_benedict",
-    "35": "escargots",
-    "36": "falafel",
-    "37": "filet_mignon",
-    "38": "fish_and_chips",
-    "39": "foie_gras",
-    "4": "beef_tartare",
-    "40": "french_fries",
-    "41": "french_onion_soup",
-    "42": "french_toast",
-    "43": "fried_calamari",
-    "44": "fried_rice",
-    "45": "frozen_yogurt",
-    "46": "garlic_bread",
-    "47": "gnocchi",
-    "48": "greek_salad",
-    "49": "grilled_cheese_sandwich",
-    "5": "beet_salad",
-    "50": "grilled_salmon",
-    "51": "guacamole",
-    "52": "gyoza",
-    "53": "hamburger",
-    "54": "hot_and_sour_soup",
-    "55": "hot_dog",
-    "56": "huevos_rancheros",
-    "57": "hummus",
-    "58": "ice_cream",
-    "59": "lasagna",
-    "6": "beignets",
-    "60": "lobster_bisque",
-    "61": "lobster_roll_sandwich",
-    "62": "macaroni_and_cheese",
-    "63": "macarons",
-    "64": "miso_soup",
-    "65": "mussels",
-    "66": "nachos",
-    "67": "omelette",
-    "68": "onion_rings",
-    "69": "oysters",
-    "7": "bibimbap",
-    "70": "pad_thai",
-    "71": "paella",
-    "72": "pancakes",
-    "73": "panna_cotta",
-    "74": "peking_duck",
-    "75": "pho",
-    "76": "pizza",
-    "77": "pork_chop",
-    "78": "poutine",
-    "79": "prime_rib",
-    "8": "bread_pudding",
-    "80": "pulled_pork_sandwich",
-    "81": "ramen",
-    "82": "ravioli",
-    "83": "red_velvet_cake",
-    "84": "risotto",
-    "85": "samosa",
-    "86": "sashimi",
-    "87": "scallops",
-    "88": "seaweed_salad",
-    "89": "shrimp_and_grits",
-    "9": "breakfast_burrito",
-    "90": "spaghetti_bolognese",
-    "91": "spaghetti_carbonara",
-    "92": "spring_rolls",
-    "93": "steak",
-    "94": "strawberry_shortcake",
-    "95": "sushi",
-    "96": "tacos",
-    "97": "takoyaki",
-    "98": "tiramisu",
-    "99": "tuna_tartare"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "apple_pie": "0",
-    "baby_back_ribs": "1",
-    "baklava": "2",
-    "beef_carpaccio": "3",
-    "beef_tartare": "4",
-    "beet_salad": "5",
-    "beignets": "6",
-    "bibimbap": "7",
-    "bread_pudding": "8",
-    "breakfast_burrito": "9",
-    "bruschetta": "10",
-    "caesar_salad": "11",
-    "cannoli": "12",
-    "caprese_salad": "13",
-    "carrot_cake": "14",
-    "ceviche": "15",
-    "cheese_plate": "17",
-    "cheesecake": "16",
-    "chicken_curry": "18",
-    "chicken_quesadilla": "19",
-    "chicken_wings": "20",
-    "chocolate_cake": "21",
-    "chocolate_mousse": "22",
-    "churros": "23",
-    "clam_chowder": "24",
-    "club_sandwich": "25",
-    "crab_cakes": "26",
-    "creme_brulee": "27",
-    "croque_madame": "28",
-    "cup_cakes": "29",
-    "deviled_eggs": "30",
-    "donuts": "31",
-    "dumplings": "32",
-    "edamame": "33",
-    "eggs_benedict": "34",
-    "escargots": "35",
-    "falafel": "36",
-    "filet_mignon": "37",
-    "fish_and_chips": "38",
-    "foie_gras": "39",
-    "french_fries": "40",
-    "french_onion_soup": "41",
-    "french_toast": "42",
-    "fried_calamari": "43",
-    "fried_rice": "44",
-    "frozen_yogurt": "45",
-    "garlic_bread": "46",
-    "gnocchi": "47",
-    "greek_salad": "48",
-    "grilled_cheese_sandwich": "49",
-    "grilled_salmon": "50",
-    "guacamole": "51",
-    "gyoza": "52",
-    "hamburger": "53",
-    "hot_and_sour_soup": "54",
-    "hot_dog": "55",
-    "huevos_rancheros": "56",
-    "hummus": "57",
-    "ice_cream": "58",
-    "lasagna": "59",
-    "lobster_bisque": "60",
-    "lobster_roll_sandwich": "61",
-    "macaroni_and_cheese": "62",
-    "macarons": "63",
-    "miso_soup": "64",
-    "mussels": "65",
-    "nachos": "66",
-    "omelette": "67",
-    "onion_rings": "68",
-    "oysters": "69",
-    "pad_thai": "70",
-    "paella": "71",
-    "pancakes": "72",
-    "panna_cotta": "73",
-    "peking_duck": "74",
-    "pho": "75",
-    "pizza": "76",
-    "pork_chop": "77",
-    "poutine": "78",
-    "prime_rib": "79",
-    "pulled_pork_sandwich": "80",
-    "ramen": "81",
-    "ravioli": "82",
-    "red_velvet_cake": "83",
-    "risotto": "84",
-    "samosa": "85",
-    "sashimi": "86",
-    "scallops": "87",
-    "seaweed_salad": "88",
-    "shrimp_and_grits": "89",
-    "spaghetti_bolognese": "90",
-    "spaghetti_carbonara": "91",
-    "spring_rolls": "92",
-    "steak": "93",
-    "strawberry_shortcake": "94",
-    "sushi": "95",
-    "tacos": "96",
-    "takoyaki": "97",
-    "tiramisu": "98",
-    "tuna_tartare": "99",
-    "waffles": "100"
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",

   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
+    "0": "56248",
+    "1": "56249",
+    "10": "56470",
+    "11": "56471",
+    "12": "56472",
+    "13": "56474",
+    "14": "56493",
+    "15": "56495",
+    "16": "56496",
+    "17": "56497",
+    "18": "56498",
+    "19": "56499",
+    "2": "56273",
+    "20": "56516",
+    "21": "56517",
+    "22": "56518",
+    "23": "56519",
+    "24": "56520",
+    "25": "56521",
+    "26": "56639",
+    "27": "56640",
+    "28": "56641",
+    "29": "56645",
+    "3": "56274",
+    "30": "56646",
+    "31": "56648",
+    "32": "56649",
+    "33": "56650",
+    "34": "56651",
+    "35": "56686",
+    "36": "56687",
+    "37": "56688",
+    "38": "56689",
+    "39": "56690",
+    "4": "56275",
+    "40": "56691",
+    "41": "56692",
+    "42": "56693",
+    "43": "56694",
+    "44": "56695",
+    "45": "56696",
+    "46": "56795",
+    "47": "56796",
+    "48": "56797",
+    "49": "56798",
+    "5": "56465",
+    "50": "56799",
+    "51": "56800",
+    "52": "56801",
+    "53": "56802",
+    "54": "56803",
+    "55": "56804",
+    "56": "56805",
+    "57": "56888",
+    "58": "57164",
+    "59": "57175",
+    "6": "56466",
+    "60": "57176",
+    "61": "57177",
+    "62": "57178",
+    "63": "57179",
+    "64": "57180",
+    "65": "57344",
+    "66": "57360",
+    "67": "57371",
+    "68": "57417",
+    "69": "57418",
+    "7": "56467",
+    "70": "57440",
+    "71": "57442",
+    "72": "57500",
+    "73": "57569",
+    "74": "57626",
+    "75": "57627",
+    "76": "57628",
+    "77": "57629",
+    "78": "57630",
+    "79": "57639",
+    "8": "56468",
+    "9": "56469"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
+    "56248": "0",
+    "56249": "1",
+    "56273": "2",
+    "56274": "3",
+    "56275": "4",
+    "56465": "5",
+    "56466": "6",
+    "56467": "7",
+    "56468": "8",
+    "56469": "9",
+    "56470": "10",
+    "56471": "11",
+    "56472": "12",
+    "56474": "13",
+    "56493": "14",
+    "56495": "15",
+    "56496": "16",
+    "56497": "17",
+    "56498": "18",
+    "56499": "19",
+    "56516": "20",
+    "56517": "21",
+    "56518": "22",
+    "56519": "23",
+    "56520": "24",
+    "56521": "25",
+    "56639": "26",
+    "56640": "27",
+    "56641": "28",
+    "56645": "29",
+    "56646": "30",
+    "56648": "31",
+    "56649": "32",
+    "56650": "33",
+    "56651": "34",
+    "56686": "35",
+    "56687": "36",
+    "56688": "37",
+    "56689": "38",
+    "56690": "39",
+    "56691": "40",
+    "56692": "41",
+    "56693": "42",
+    "56694": "43",
+    "56695": "44",
+    "56696": "45",
+    "56795": "46",
+    "56796": "47",
+    "56797": "48",
+    "56798": "49",
+    "56799": "50",
+    "56800": "51",
+    "56801": "52",
+    "56802": "53",
+    "56803": "54",
+    "56804": "55",
+    "56805": "56",
+    "56888": "57",
+    "57164": "58",
+    "57175": "59",
+    "57176": "60",
+    "57177": "61",
+    "57178": "62",
+    "57179": "63",
+    "57180": "64",
+    "57344": "65",
+    "57360": "66",
+    "57371": "67",
+    "57417": "68",
+    "57418": "69",
+    "57440": "70",
+    "57442": "71",
+    "57500": "72",
+    "57569": "73",
+    "57626": "74",
+    "57627": "75",
+    "57628": "76",
+    "57629": "77",
+    "57630": "78",
+    "57639": "79"
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",

tf_model.h5 CHANGED Viewed

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

 version https://git-lfs.github.com/spec/v1
+oid sha256:8bcb0be5c078d818d2085cd3ef225fe498679fc4861d82aa543db711be6bfccd
+size 343725752