rathi2023's picture
Training in progress, step 200
b2f71b6 verified
raw
history blame
1.76 kB
{
"_name_or_path": "google/owlvit-base-patch32",
"architectures": [
"OwlViTForObjectDetection"
],
"id2label": {
"0": 6,
"1": 7,
"10": 16,
"11": 17,
"12": 18,
"13": 19,
"14": 20,
"15": 24,
"16": 25,
"17": 26,
"18": 27,
"19": 28,
"2": 8,
"20": 29,
"21": 30,
"22": 31,
"23": 32,
"24": 33,
"25": 34,
"26": 36,
"27": 37,
"28": 38,
"29": 40,
"3": 9,
"30": 41,
"31": 42,
"32": 43,
"33": 44,
"34": 45,
"35": 46,
"36": 47,
"37": 48,
"4": 10,
"5": 11,
"6": 12,
"7": 13,
"8": 14,
"9": 15
},
"initializer_factor": 1.0,
"label2id": {
"6": "0",
"7": "1",
"8": "2",
"9": "3",
"10": "4",
"11": "5",
"12": "6",
"13": "7",
"14": "8",
"15": "9",
"16": "10",
"17": "11",
"18": "12",
"19": "13",
"20": "14",
"24": "15",
"25": "16",
"26": "17",
"27": "18",
"28": "19",
"29": "20",
"30": "21",
"31": "22",
"32": "23",
"33": "24",
"34": "25",
"36": "26",
"37": "27",
"38": "28",
"40": "29",
"41": "30",
"42": "31",
"43": "32",
"44": "33",
"45": "34",
"46": "35",
"47": "36",
"48": "37"
},
"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.38.2",
"vision_config": {
"dropout": 0.0,
"model_type": "owlvit_vision_model"
},
"vision_config_dict": null
}