resnet-50-ucSat / config.json
YKXBCi's picture
Training in progress epoch 0
a238f95
raw
history blame
1.79 kB
{
"_name_or_path": "microsoft/resnet-50",
"architectures": [
"ViTForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"depths": [
3,
4,
6,
3
],
"downsample_in_first_stage": false,
"embedding_size": 64,
"encoder_stride": 16,
"hidden_act": "relu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"hidden_sizes": [
256,
512,
1024,
2048
],
"id2label": {
"0": "agricultural",
"1": "airplane",
"10": "harbor",
"11": "intersection",
"12": "mediumresidential",
"13": "mobilehomepark",
"14": "overpass",
"15": "parkinglot",
"16": "river",
"17": "runway",
"18": "sparseresidential",
"19": "storagetanks",
"2": "baseballdiamond",
"20": "tenniscourt",
"3": "beach",
"4": "buildings",
"5": "chaparral",
"6": "denseresidential",
"7": "forest",
"8": "freeway",
"9": "golfcourse"
},
"image_size": 224,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"agricultural": "0",
"airplane": "1",
"baseballdiamond": "2",
"beach": "3",
"buildings": "4",
"chaparral": "5",
"denseresidential": "6",
"forest": "7",
"freeway": "8",
"golfcourse": "9",
"harbor": "10",
"intersection": "11",
"mediumresidential": "12",
"mobilehomepark": "13",
"overpass": "14",
"parkinglot": "15",
"river": "16",
"runway": "17",
"sparseresidential": "18",
"storagetanks": "19",
"tenniscourt": "20"
},
"layer_norm_eps": 1e-12,
"layer_type": "bottleneck",
"model_type": "vit",
"num_attention_heads": 12,
"num_channels": 3,
"num_hidden_layers": 12,
"patch_size": 16,
"qkv_bias": true,
"torch_dtype": "float32",
"transformers_version": "4.18.0"
}