Upload CustomViltForVQA

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "dandelin/vilt-b32-finetuned-vqa",
+  "_name_or_path": "phonghoccode/vilt-vqa-finetune",
   "architectures": [
     "CustomViltForVQA"
   ],
@@ -8,227 +8,227 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "boat",
-    "1": "above",
-    "2": "apple",
-    "3": "clock",
-    "4": "9",
-    "5": "wine glass",
-    "6": "mouse",
-    "7": "2",
-    "8": "sheep",
-    "9": "couch",
-    "10": "snowboard",
-    "11": "Yes",
-    "12": "spoon",
-    "13": "sports ball",
-    "14": "handbag",
-    "15": "Blue",
-    "16": "5",
-    "17": "laptop",
-    "18": "toothbrush",
-    "19": "carrot",
-    "20": "cake",
-    "21": "dog",
-    "22": "White",
-    "23": "motorcycle",
-    "24": "chair",
-    "25": "train",
-    "26": "zebra",
-    "27": "6",
-    "28": "person",
-    "29": "parking meter",
-    "30": "Brown",
-    "31": "car",
-    "32": "sink",
-    "33": "right",
-    "34": "bear",
-    "35": "Right",
-    "36": "sandwich",
-    "37": "Grey",
-    "38": "Red",
-    "39": "surfboard",
-    "40": "bowl",
-    "41": "airplane",
-    "42": "potted plant",
-    "43": "7",
-    "44": "tv",
-    "45": "suitcase",
-    "46": "kite",
-    "47": "tennis racket",
-    "48": "bench",
-    "49": "0",
-    "50": "bed",
-    "51": "dining table",
-    "52": "No",
-    "53": "refrigerator",
-    "54": "giraffe",
-    "55": "Purple",
-    "56": "hot dog",
-    "57": "truck",
-    "58": "vase",
-    "59": "Orange",
-    "60": "tie",
-    "61": "broccoli",
-    "62": "umbrella",
-    "63": "Green",
-    "64": "left",
-    "65": "stop sign",
-    "66": "cat",
-    "67": "teddy bear",
-    "68": "bicycle",
-    "69": "orange",
-    "70": "3",
-    "71": "scissors",
-    "72": "baseball glove",
-    "73": "frisbee",
-    "74": "4",
-    "75": "keyboard",
-    "76": "banana",
-    "77": "bus",
-    "78": "1",
-    "79": "Black",
-    "80": "8",
-    "81": "fork",
-    "82": "baseball bat",
-    "83": "donut",
-    "84": "book",
-    "85": "Above",
-    "86": "skis",
-    "87": "oven",
-    "88": "Below",
-    "89": "cell phone",
-    "90": "Left",
-    "91": "fire hydrant",
-    "92": "Behind",
-    "93": "cup",
-    "94": "elephant",
-    "95": "pizza",
-    "96": "bird",
-    "97": "Pink",
-    "98": "knife",
-    "99": "skateboard",
-    "100": "horse",
-    "101": "Front",
-    "102": "Yellow",
-    "103": "traffic light",
-    "104": "remote",
-    "105": "cow",
-    "106": "toilet",
-    "107": "bottle"
+    "0": "sink",
+    "1": "zebra",
+    "2": "Right",
+    "3": "toothbrush",
+    "4": "hot dog",
+    "5": "traffic light",
+    "6": "chair",
+    "7": "mouse",
+    "8": "bicycle",
+    "9": "2",
+    "10": "bus",
+    "11": "fork",
+    "12": "carrot",
+    "13": "car",
+    "14": "Blue",
+    "15": "pizza",
+    "16": "9",
+    "17": "bear",
+    "18": "banana",
+    "19": "kite",
+    "20": "6",
+    "21": "dining table",
+    "22": "above",
+    "23": "sheep",
+    "24": "truck",
+    "25": "5",
+    "26": "8",
+    "27": "Left",
+    "28": "0",
+    "29": "oven",
+    "30": "apple",
+    "31": "handbag",
+    "32": "cup",
+    "33": "umbrella",
+    "34": "couch",
+    "35": "stop sign",
+    "36": "Above",
+    "37": "refrigerator",
+    "38": "giraffe",
+    "39": "train",
+    "40": "remote",
+    "41": "Below",
+    "42": "cake",
+    "43": "3",
+    "44": "clock",
+    "45": "bottle",
+    "46": "tie",
+    "47": "knife",
+    "48": "dog",
+    "49": "White",
+    "50": "skateboard",
+    "51": "sandwich",
+    "52": "Front",
+    "53": "cat",
+    "54": "suitcase",
+    "55": "snowboard",
+    "56": "broccoli",
+    "57": "tv",
+    "58": "elephant",
+    "59": "tennis racket",
+    "60": "baseball glove",
+    "61": "motorcycle",
+    "62": "right",
+    "63": "bird",
+    "64": "Brown",
+    "65": "airplane",
+    "66": "1",
+    "67": "bench",
+    "68": "boat",
+    "69": "No",
+    "70": "book",
+    "71": "Orange",
+    "72": "donut",
+    "73": "Red",
+    "74": "Behind",
+    "75": "laptop",
+    "76": "orange",
+    "77": "teddy bear",
+    "78": "Black",
+    "79": "person",
+    "80": "surfboard",
+    "81": "7",
+    "82": "4",
+    "83": "Green",
+    "84": "scissors",
+    "85": "Grey",
+    "86": "left",
+    "87": "horse",
+    "88": "vase",
+    "89": "potted plant",
+    "90": "sports ball",
+    "91": "Purple",
+    "92": "spoon",
+    "93": "parking meter",
+    "94": "keyboard",
+    "95": "cow",
+    "96": "wine glass",
+    "97": "fire hydrant",
+    "98": "bed",
+    "99": "baseball bat",
+    "100": "skis",
+    "101": "Yes",
+    "102": "frisbee",
+    "103": "toilet",
+    "104": "Pink",
+    "105": "Yellow",
+    "106": "bowl",
+    "107": "cell phone"
   },
   "image_size": 384,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "0": 49,
-    "1": 78,
-    "2": 7,
-    "3": 70,
-    "4": 74,
-    "5": 16,
-    "6": 27,
-    "7": 43,
-    "8": 80,
-    "9": 4,
-    "Above": 85,
-    "Behind": 92,
-    "Below": 88,
-    "Black": 79,
-    "Blue": 15,
-    "Brown": 30,
-    "Front": 101,
-    "Green": 63,
-    "Grey": 37,
-    "Left": 90,
-    "No": 52,
-    "Orange": 59,
-    "Pink": 97,
-    "Purple": 55,
-    "Red": 38,
-    "Right": 35,
-    "White": 22,
-    "Yellow": 102,
-    "Yes": 11,
-    "above": 1,
-    "airplane": 41,
-    "apple": 2,
-    "banana": 76,
-    "baseball bat": 82,
-    "baseball glove": 72,
-    "bear": 34,
-    "bed": 50,
-    "bench": 48,
-    "bicycle": 68,
-    "bird": 96,
-    "boat": 0,
-    "book": 84,
-    "bottle": 107,
-    "bowl": 40,
-    "broccoli": 61,
-    "bus": 77,
-    "cake": 20,
-    "car": 31,
-    "carrot": 19,
-    "cat": 66,
-    "cell phone": 89,
-    "chair": 24,
-    "clock": 3,
-    "couch": 9,
-    "cow": 105,
-    "cup": 93,
-    "dining table": 51,
-    "dog": 21,
-    "donut": 83,
-    "elephant": 94,
-    "fire hydrant": 91,
-    "fork": 81,
-    "frisbee": 73,
-    "giraffe": 54,
-    "handbag": 14,
-    "horse": 100,
-    "hot dog": 56,
-    "keyboard": 75,
-    "kite": 46,
-    "knife": 98,
-    "laptop": 17,
-    "left": 64,
-    "motorcycle": 23,
-    "mouse": 6,
-    "orange": 69,
-    "oven": 87,
-    "parking meter": 29,
-    "person": 28,
-    "pizza": 95,
-    "potted plant": 42,
-    "refrigerator": 53,
-    "remote": 104,
-    "right": 33,
-    "sandwich": 36,
-    "scissors": 71,
-    "sheep": 8,
-    "sink": 32,
-    "skateboard": 99,
-    "skis": 86,
-    "snowboard": 10,
-    "spoon": 12,
-    "sports ball": 13,
-    "stop sign": 65,
-    "suitcase": 45,
-    "surfboard": 39,
-    "teddy bear": 67,
-    "tennis racket": 47,
-    "tie": 60,
-    "toilet": 106,
-    "toothbrush": 18,
-    "traffic light": 103,
-    "train": 25,
-    "truck": 57,
-    "tv": 44,
-    "umbrella": 62,
-    "vase": 58,
-    "wine glass": 5,
-    "zebra": 26
+    "0": 28,
+    "1": 66,
+    "2": 9,
+    "3": 43,
+    "4": 82,
+    "5": 25,
+    "6": 20,
+    "7": 81,
+    "8": 26,
+    "9": 16,
+    "Above": 36,
+    "Behind": 74,
+    "Below": 41,
+    "Black": 78,
+    "Blue": 14,
+    "Brown": 64,
+    "Front": 52,
+    "Green": 83,
+    "Grey": 85,
+    "Left": 27,
+    "No": 69,
+    "Orange": 71,
+    "Pink": 104,
+    "Purple": 91,
+    "Red": 73,
+    "Right": 2,
+    "White": 49,
+    "Yellow": 105,
+    "Yes": 101,
+    "above": 22,
+    "airplane": 65,
+    "apple": 30,
+    "banana": 18,
+    "baseball bat": 99,
+    "baseball glove": 60,
+    "bear": 17,
+    "bed": 98,
+    "bench": 67,
+    "bicycle": 8,
+    "bird": 63,
+    "boat": 68,
+    "book": 70,
+    "bottle": 45,
+    "bowl": 106,
+    "broccoli": 56,
+    "bus": 10,
+    "cake": 42,
+    "car": 13,
+    "carrot": 12,
+    "cat": 53,
+    "cell phone": 107,
+    "chair": 6,
+    "clock": 44,
+    "couch": 34,
+    "cow": 95,
+    "cup": 32,
+    "dining table": 21,
+    "dog": 48,
+    "donut": 72,
+    "elephant": 58,
+    "fire hydrant": 97,
+    "fork": 11,
+    "frisbee": 102,
+    "giraffe": 38,
+    "handbag": 31,
+    "horse": 87,
+    "hot dog": 4,
+    "keyboard": 94,
+    "kite": 19,
+    "knife": 47,
+    "laptop": 75,
+    "left": 86,
+    "motorcycle": 61,
+    "mouse": 7,
+    "orange": 76,
+    "oven": 29,
+    "parking meter": 93,
+    "person": 79,
+    "pizza": 15,
+    "potted plant": 89,
+    "refrigerator": 37,
+    "remote": 40,
+    "right": 62,
+    "sandwich": 51,
+    "scissors": 84,
+    "sheep": 23,
+    "sink": 0,
+    "skateboard": 50,
+    "skis": 100,
+    "snowboard": 55,
+    "spoon": 92,
+    "sports ball": 90,
+    "stop sign": 35,
+    "suitcase": 54,
+    "surfboard": 80,
+    "teddy bear": 77,
+    "tennis racket": 59,
+    "tie": 46,
+    "toilet": 103,
+    "toothbrush": 3,
+    "traffic light": 5,
+    "train": 39,
+    "truck": 24,
+    "tv": 57,
+    "umbrella": 33,
+    "vase": 88,
+    "wine glass": 96,
+    "zebra": 1
   },
   "layer_norm_eps": 1e-12,
   "max_image_length": -1,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b9286befde0cc5eeffadbf936aad99ed37d7ca481f2b7755ace4122b99f138f5
+oid sha256:69cbf36c74dfb571bf45dbc627923b779ab169d1b4761396c5022df7a4e3325d
 size 451805856