initial_ViT_model / config.json
dhruvilHV's picture
Training in progress, epoch 0
6a63513
{
"_name_or_path": "google/vit-base-patch16-224-in21k",
"architectures": [
"ViTForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"encoder_stride": 16,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"id2label": {
"0": "East Asian_Male_0-2",
"1": "East Asian_Male_3-9",
"2": "East Asian_Male_10-19",
"3": "East Asian_Male_20-29",
"4": "East Asian_Male_30-39",
"5": "East Asian_Male_40-49",
"6": "East Asian_Male_50-59",
"7": "East Asian_Male_60-69",
"8": "East Asian_Male_more than 70",
"9": "East Asian_Female_0-2",
"10": "East Asian_Female_3-9",
"11": "East Asian_Female_10-19",
"12": "East Asian_Female_20-29",
"13": "East Asian_Female_30-39",
"14": "East Asian_Female_40-49",
"15": "East Asian_Female_50-59",
"16": "East Asian_Female_60-69",
"17": "East Asian_Female_more than 70",
"18": "Indian_Male_0-2",
"19": "Indian_Male_3-9",
"20": "Indian_Male_10-19",
"21": "Indian_Male_20-29",
"22": "Indian_Male_30-39",
"23": "Indian_Male_40-49",
"24": "Indian_Male_50-59",
"25": "Indian_Male_60-69",
"26": "Indian_Male_more than 70",
"27": "Indian_Female_0-2",
"28": "Indian_Female_3-9",
"29": "Indian_Female_10-19",
"30": "Indian_Female_20-29",
"31": "Indian_Female_30-39",
"32": "Indian_Female_40-49",
"33": "Indian_Female_50-59",
"34": "Indian_Female_60-69",
"35": "Indian_Female_more than 70",
"36": "Black_Male_0-2",
"37": "Black_Male_3-9",
"38": "Black_Male_10-19",
"39": "Black_Male_20-29",
"40": "Black_Male_30-39",
"41": "Black_Male_40-49",
"42": "Black_Male_50-59",
"43": "Black_Male_60-69",
"44": "Black_Male_more than 70",
"45": "Black_Female_0-2",
"46": "Black_Female_3-9",
"47": "Black_Female_10-19",
"48": "Black_Female_20-29",
"49": "Black_Female_30-39",
"50": "Black_Female_40-49",
"51": "Black_Female_50-59",
"52": "Black_Female_60-69",
"53": "Black_Female_more than 70",
"54": "White_Male_0-2",
"55": "White_Male_3-9",
"56": "White_Male_10-19",
"57": "White_Male_20-29",
"58": "White_Male_30-39",
"59": "White_Male_40-49",
"60": "White_Male_50-59",
"61": "White_Male_60-69",
"62": "White_Male_more than 70",
"63": "White_Female_0-2",
"64": "White_Female_3-9",
"65": "White_Female_10-19",
"66": "White_Female_20-29",
"67": "White_Female_30-39",
"68": "White_Female_40-49",
"69": "White_Female_50-59",
"70": "White_Female_60-69",
"71": "White_Female_more than 70",
"72": "Middle Eastern_Male_0-2",
"73": "Middle Eastern_Male_3-9",
"74": "Middle Eastern_Male_10-19",
"75": "Middle Eastern_Male_20-29",
"76": "Middle Eastern_Male_30-39",
"77": "Middle Eastern_Male_40-49",
"78": "Middle Eastern_Male_50-59",
"79": "Middle Eastern_Male_60-69",
"80": "Middle Eastern_Male_more than 70",
"81": "Middle Eastern_Female_0-2",
"82": "Middle Eastern_Female_3-9",
"83": "Middle Eastern_Female_10-19",
"84": "Middle Eastern_Female_20-29",
"85": "Middle Eastern_Female_30-39",
"86": "Middle Eastern_Female_40-49",
"87": "Middle Eastern_Female_50-59",
"88": "Middle Eastern_Female_60-69",
"89": "Middle Eastern_Female_more than 70",
"90": "Latino_Hispanic_Male_0-2",
"91": "Latino_Hispanic_Male_3-9",
"92": "Latino_Hispanic_Male_10-19",
"93": "Latino_Hispanic_Male_20-29",
"94": "Latino_Hispanic_Male_30-39",
"95": "Latino_Hispanic_Male_40-49",
"96": "Latino_Hispanic_Male_50-59",
"97": "Latino_Hispanic_Male_60-69",
"98": "Latino_Hispanic_Male_more than 70",
"99": "Latino_Hispanic_Female_0-2",
"100": "Latino_Hispanic_Female_3-9",
"101": "Latino_Hispanic_Female_10-19",
"102": "Latino_Hispanic_Female_20-29",
"103": "Latino_Hispanic_Female_30-39",
"104": "Latino_Hispanic_Female_40-49",
"105": "Latino_Hispanic_Female_50-59",
"106": "Latino_Hispanic_Female_60-69",
"107": "Latino_Hispanic_Female_more than 70",
"108": "Southeast Asian_Male_0-2",
"109": "Southeast Asian_Male_3-9",
"110": "Southeast Asian_Male_10-19",
"111": "Southeast Asian_Male_20-29",
"112": "Southeast Asian_Male_30-39",
"113": "Southeast Asian_Male_40-49",
"114": "Southeast Asian_Male_50-59",
"115": "Southeast Asian_Male_60-69",
"116": "Southeast Asian_Male_more than 70",
"117": "Southeast Asian_Female_0-2",
"118": "Southeast Asian_Female_3-9",
"119": "Southeast Asian_Female_10-19",
"120": "Southeast Asian_Female_20-29",
"121": "Southeast Asian_Female_30-39",
"122": "Southeast Asian_Female_40-49",
"123": "Southeast Asian_Female_50-59",
"124": "Southeast Asian_Female_60-69",
"125": "Southeast Asian_Female_more than 70"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"Black_Female_0-2": 45,
"Black_Female_10-19": 47,
"Black_Female_20-29": 48,
"Black_Female_3-9": 46,
"Black_Female_30-39": 49,
"Black_Female_40-49": 50,
"Black_Female_50-59": 51,
"Black_Female_60-69": 52,
"Black_Female_more than 70": 53,
"Black_Male_0-2": 36,
"Black_Male_10-19": 38,
"Black_Male_20-29": 39,
"Black_Male_3-9": 37,
"Black_Male_30-39": 40,
"Black_Male_40-49": 41,
"Black_Male_50-59": 42,
"Black_Male_60-69": 43,
"Black_Male_more than 70": 44,
"East Asian_Female_0-2": 9,
"East Asian_Female_10-19": 11,
"East Asian_Female_20-29": 12,
"East Asian_Female_3-9": 10,
"East Asian_Female_30-39": 13,
"East Asian_Female_40-49": 14,
"East Asian_Female_50-59": 15,
"East Asian_Female_60-69": 16,
"East Asian_Female_more than 70": 17,
"East Asian_Male_0-2": 0,
"East Asian_Male_10-19": 2,
"East Asian_Male_20-29": 3,
"East Asian_Male_3-9": 1,
"East Asian_Male_30-39": 4,
"East Asian_Male_40-49": 5,
"East Asian_Male_50-59": 6,
"East Asian_Male_60-69": 7,
"East Asian_Male_more than 70": 8,
"Indian_Female_0-2": 27,
"Indian_Female_10-19": 29,
"Indian_Female_20-29": 30,
"Indian_Female_3-9": 28,
"Indian_Female_30-39": 31,
"Indian_Female_40-49": 32,
"Indian_Female_50-59": 33,
"Indian_Female_60-69": 34,
"Indian_Female_more than 70": 35,
"Indian_Male_0-2": 18,
"Indian_Male_10-19": 20,
"Indian_Male_20-29": 21,
"Indian_Male_3-9": 19,
"Indian_Male_30-39": 22,
"Indian_Male_40-49": 23,
"Indian_Male_50-59": 24,
"Indian_Male_60-69": 25,
"Indian_Male_more than 70": 26,
"Latino_Hispanic_Female_0-2": 99,
"Latino_Hispanic_Female_10-19": 101,
"Latino_Hispanic_Female_20-29": 102,
"Latino_Hispanic_Female_3-9": 100,
"Latino_Hispanic_Female_30-39": 103,
"Latino_Hispanic_Female_40-49": 104,
"Latino_Hispanic_Female_50-59": 105,
"Latino_Hispanic_Female_60-69": 106,
"Latino_Hispanic_Female_more than 70": 107,
"Latino_Hispanic_Male_0-2": 90,
"Latino_Hispanic_Male_10-19": 92,
"Latino_Hispanic_Male_20-29": 93,
"Latino_Hispanic_Male_3-9": 91,
"Latino_Hispanic_Male_30-39": 94,
"Latino_Hispanic_Male_40-49": 95,
"Latino_Hispanic_Male_50-59": 96,
"Latino_Hispanic_Male_60-69": 97,
"Latino_Hispanic_Male_more than 70": 98,
"Middle Eastern_Female_0-2": 81,
"Middle Eastern_Female_10-19": 83,
"Middle Eastern_Female_20-29": 84,
"Middle Eastern_Female_3-9": 82,
"Middle Eastern_Female_30-39": 85,
"Middle Eastern_Female_40-49": 86,
"Middle Eastern_Female_50-59": 87,
"Middle Eastern_Female_60-69": 88,
"Middle Eastern_Female_more than 70": 89,
"Middle Eastern_Male_0-2": 72,
"Middle Eastern_Male_10-19": 74,
"Middle Eastern_Male_20-29": 75,
"Middle Eastern_Male_3-9": 73,
"Middle Eastern_Male_30-39": 76,
"Middle Eastern_Male_40-49": 77,
"Middle Eastern_Male_50-59": 78,
"Middle Eastern_Male_60-69": 79,
"Middle Eastern_Male_more than 70": 80,
"Southeast Asian_Female_0-2": 117,
"Southeast Asian_Female_10-19": 119,
"Southeast Asian_Female_20-29": 120,
"Southeast Asian_Female_3-9": 118,
"Southeast Asian_Female_30-39": 121,
"Southeast Asian_Female_40-49": 122,
"Southeast Asian_Female_50-59": 123,
"Southeast Asian_Female_60-69": 124,
"Southeast Asian_Female_more than 70": 125,
"Southeast Asian_Male_0-2": 108,
"Southeast Asian_Male_10-19": 110,
"Southeast Asian_Male_20-29": 111,
"Southeast Asian_Male_3-9": 109,
"Southeast Asian_Male_30-39": 112,
"Southeast Asian_Male_40-49": 113,
"Southeast Asian_Male_50-59": 114,
"Southeast Asian_Male_60-69": 115,
"Southeast Asian_Male_more than 70": 116,
"White_Female_0-2": 63,
"White_Female_10-19": 65,
"White_Female_20-29": 66,
"White_Female_3-9": 64,
"White_Female_30-39": 67,
"White_Female_40-49": 68,
"White_Female_50-59": 69,
"White_Female_60-69": 70,
"White_Female_more than 70": 71,
"White_Male_0-2": 54,
"White_Male_10-19": 56,
"White_Male_20-29": 57,
"White_Male_3-9": 55,
"White_Male_30-39": 58,
"White_Male_40-49": 59,
"White_Male_50-59": 60,
"White_Male_60-69": 61,
"White_Male_more than 70": 62
},
"layer_norm_eps": 1e-12,
"model_type": "vit",
"num_attention_heads": 12,
"num_channels": 3,
"num_hidden_layers": 12,
"patch_size": 16,
"problem_type": "single_label_classification",
"qkv_bias": true,
"torch_dtype": "float32",
"transformers_version": "4.35.2"
}