custom_model / config.json
nateg10's picture
Training in progress, step 200
5040a0a verified
{
"_name_or_path": "google/owlvit-base-patch32",
"architectures": [
"OwlViTForObjectDetection"
],
"id2label": {
"0": 1,
"1": 3,
"10": 21,
"11": 22,
"12": 23,
"13": 24,
"14": 25,
"15": 26,
"16": 27,
"17": 28,
"18": 29,
"19": 30,
"2": 7,
"20": 31,
"21": 32,
"22": 33,
"23": 34,
"24": 35,
"25": 36,
"26": 37,
"27": 38,
"28": 39,
"29": 40,
"3": 9,
"30": 41,
"31": 42,
"32": 43,
"33": 44,
"34": 45,
"35": 46,
"36": 50,
"37": 51,
"38": 52,
"39": 53,
"4": 11,
"40": 54,
"41": 55,
"42": 56,
"43": 57,
"44": 58,
"45": 59,
"46": 60,
"47": 61,
"48": 62,
"49": 63,
"5": 13,
"50": 64,
"51": 66,
"52": 67,
"53": 68,
"54": 69,
"55": 70,
"56": 72,
"57": 73,
"58": 74,
"59": 75,
"6": 14,
"60": 76,
"61": 77,
"62": 78,
"63": 79,
"64": 80,
"65": 81,
"66": 82,
"67": 83,
"68": 84,
"69": 85,
"7": 16,
"70": 86,
"71": 87,
"72": 88,
"73": 89,
"74": 90,
"75": 91,
"76": 92,
"77": 93,
"78": 94,
"79": 97,
"8": 19,
"80": 98,
"81": 99,
"82": 100,
"83": 101,
"84": 102,
"85": 103,
"86": 104,
"87": 105,
"9": 20
},
"initializer_factor": 1.0,
"label2id": {
"1": "0",
"3": "1",
"7": "2",
"9": "3",
"11": "4",
"13": "5",
"14": "6",
"16": "7",
"19": "8",
"20": "9",
"21": "10",
"22": "11",
"23": "12",
"24": "13",
"25": "14",
"26": "15",
"27": "16",
"28": "17",
"29": "18",
"30": "19",
"31": "20",
"32": "21",
"33": "22",
"34": "23",
"35": "24",
"36": "25",
"37": "26",
"38": "27",
"39": "28",
"40": "29",
"41": "30",
"42": "31",
"43": "32",
"44": "33",
"45": "34",
"46": "35",
"50": "36",
"51": "37",
"52": "38",
"53": "39",
"54": "40",
"55": "41",
"56": "42",
"57": "43",
"58": "44",
"59": "45",
"60": "46",
"61": "47",
"62": "48",
"63": "49",
"64": "50",
"66": "51",
"67": "52",
"68": "53",
"69": "54",
"70": "55",
"72": "56",
"73": "57",
"74": "58",
"75": "59",
"76": "60",
"77": "61",
"78": "62",
"79": "63",
"80": "64",
"81": "65",
"82": "66",
"83": "67",
"84": "68",
"85": "69",
"86": "70",
"87": "71",
"88": "72",
"89": "73",
"90": "74",
"91": "75",
"92": "76",
"93": "77",
"94": "78",
"97": "79",
"98": "80",
"99": "81",
"100": "82",
"101": "83",
"102": "84",
"103": "85",
"104": "86",
"105": "87"
},
"logit_scale_init_value": 2.6592,
"model_type": "owlvit",
"projection_dim": 512,
"text_config": {
"bos_token_id": 0,
"dropout": 0.0,
"eos_token_id": 2,
"max_length": 16,
"model_type": "owlvit_text_model",
"pad_token_id": 1
},
"text_config_dict": null,
"torch_dtype": "float32",
"transformers_version": "4.40.1",
"vision_config": {
"dropout": 0.0,
"model_type": "owlvit_vision_model"
},
"vision_config_dict": null
}