seena18's picture
Training in progress, epoch 1
294655a
{
"_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": "dam",
"1": "parking lot",
"10": "sparse shrub land",
"11": "works",
"12": "oil field",
"13": "meadow",
"14": "ground track field",
"15": "detached house",
"16": "golf course",
"17": "forest",
"18": "desert",
"19": "lake",
"2": "greenhouse",
"20": "beach",
"21": "paddy field",
"22": "ice land",
"23": "bare land",
"24": "storage tank",
"25": "basketball court",
"26": "island",
"27": "substation",
"28": "mobile home park",
"29": "cemetery",
"3": "pier",
"30": "quarry",
"31": "solar power plant",
"32": "helipad",
"33": "roundabout",
"34": "runway",
"35": "wastewater plant",
"36": "river",
"37": "apartment",
"38": "dry field",
"39": "intersection",
"4": "bridge",
"40": "swimming pool",
"41": "commercial area",
"42": "church",
"43": "road",
"44": "orchard",
"45": "terraced field",
"46": "stadium",
"47": "train station",
"48": "railway",
"49": "viaduct",
"5": "mine",
"50": "wind turbine",
"6": "rock land",
"7": "baseball field",
"8": "apron",
"9": "tennis court"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"apartment": "37",
"apron": "8",
"bare land": "23",
"baseball field": "7",
"basketball court": "25",
"beach": "20",
"bridge": "4",
"cemetery": "29",
"church": "42",
"commercial area": "41",
"dam": "0",
"desert": "18",
"detached house": "15",
"dry field": "38",
"forest": "17",
"golf course": "16",
"greenhouse": "2",
"ground track field": "14",
"helipad": "32",
"ice land": "22",
"intersection": "39",
"island": "26",
"lake": "19",
"meadow": "13",
"mine": "5",
"mobile home park": "28",
"oil field": "12",
"orchard": "44",
"paddy field": "21",
"parking lot": "1",
"pier": "3",
"quarry": "30",
"railway": "48",
"river": "36",
"road": "43",
"rock land": "6",
"roundabout": "33",
"runway": "34",
"solar power plant": "31",
"sparse shrub land": "10",
"stadium": "46",
"storage tank": "24",
"substation": "27",
"swimming pool": "40",
"tennis court": "9",
"terraced field": "45",
"train station": "47",
"viaduct": "49",
"wastewater plant": "35",
"wind turbine": "50",
"works": "11"
},
"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.28.1"
}