Upload CustomViltForVQA

config.json

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