vigneshwar472's picture
Training in progress, epoch 0
22bef96 verified
{
"_name_or_path": "google/vit-large-patch16-224-in21k",
"architectures": [
"ViTForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"encoder_stride": 16,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 1024,
"id2label": {
"0": "airplane",
"1": "airport",
"10": "cloud",
"11": "containers",
"12": "crosswalk",
"13": "dense residential area",
"14": "desert",
"15": "dock",
"16": "factory",
"17": "field",
"18": "football field",
"19": "forest",
"2": "bare soil",
"20": "freeway",
"21": "golf course",
"22": "grass",
"23": "greenhouse",
"24": "gully",
"25": "habor",
"26": "intersection",
"27": "island",
"28": "lake",
"29": "mobile home",
"3": "baseball diamond",
"30": "mountain",
"31": "overpass",
"32": "park",
"33": "parking lot",
"34": "parkway",
"35": "pavement",
"36": "railway",
"37": "railway station",
"38": "river",
"39": "road",
"4": "basketball court",
"40": "roundabout",
"41": "runway",
"42": "sand",
"43": "sea",
"44": "ships",
"45": "snow",
"46": "snowberg",
"47": "sparse residential area",
"48": "stadium",
"49": "swimming pool",
"5": "beach",
"50": "tanks",
"51": "tennis court",
"52": "terrace",
"53": "track",
"54": "trail",
"55": "transmission tower",
"56": "trees",
"57": "water",
"58": "wetland",
"59": "wind turbine",
"6": "bridge",
"7": "buildings",
"8": "cars",
"9": "chaparral"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 4096,
"label2id": {
"airplane": "0",
"airport": "1",
"bare soil": "2",
"baseball diamond": "3",
"basketball court": "4",
"beach": "5",
"bridge": "6",
"buildings": "7",
"cars": "8",
"chaparral": "9",
"cloud": "10",
"containers": "11",
"crosswalk": "12",
"dense residential area": "13",
"desert": "14",
"dock": "15",
"factory": "16",
"field": "17",
"football field": "18",
"forest": "19",
"freeway": "20",
"golf course": "21",
"grass": "22",
"greenhouse": "23",
"gully": "24",
"habor": "25",
"intersection": "26",
"island": "27",
"lake": "28",
"mobile home": "29",
"mountain": "30",
"overpass": "31",
"park": "32",
"parking lot": "33",
"parkway": "34",
"pavement": "35",
"railway": "36",
"railway station": "37",
"river": "38",
"road": "39",
"roundabout": "40",
"runway": "41",
"sand": "42",
"sea": "43",
"ships": "44",
"snow": "45",
"snowberg": "46",
"sparse residential area": "47",
"stadium": "48",
"swimming pool": "49",
"tanks": "50",
"tennis court": "51",
"terrace": "52",
"track": "53",
"trail": "54",
"transmission tower": "55",
"trees": "56",
"water": "57",
"wetland": "58",
"wind turbine": "59"
},
"layer_norm_eps": 1e-12,
"model_type": "vit",
"num_attention_heads": 16,
"num_channels": 3,
"num_hidden_layers": 24,
"patch_size": 16,
"problem_type": "multi_label_classification",
"qkv_bias": true,
"torch_dtype": "float32",
"transformers_version": "4.42.3"
}