swing_birds_model / config.json
mic3333's picture
Upload SwinForImageClassification
3e693fe verified
{
"_name_or_path": "microsoft/swin-base-patch4-window7-224-in22k",
"architectures": [
"SwinForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"depths": [
2,
2,
18,
2
],
"drop_path_rate": 0.1,
"embed_dim": 128,
"encoder_stride": 32,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 1024,
"id2label": {
"0": "ALBATROSS",
"1": "AMERICAN BITTERN",
"2": "ANDEAN LAPWING",
"3": "ASHY STORM PETREL",
"4": "BARN SWALLOW",
"5": "BLACK VENTED SHEARWATER",
"6": "BREWERS BLACKBIRD",
"7": "CALIFORNIA GULL",
"8": "CASPIAN TERN",
"9": "DOUBLE EYED FIG PARROT",
"10": "EURASIAN MAGPIE",
"11": "FAIRY PENGUIN",
"12": "IMPERIAL SHAQ",
"13": "MALLARD DUCK",
"14": "NORTHERN GANNET",
"15": "OYSTER CATCHER",
"16": "PURPLE SWAMPHEN",
"17": "PYGMY KINGFISHER",
"18": "RED BROWED FINCH",
"19": "RED TAILED HAWK",
"20": "ROCK DOVE",
"21": "SNOW GOOSE",
"22": "SPOON BILED SANDPIPER",
"23": "STRIPED OWL",
"24": "TRUMPTER SWAN",
"25": "VIOLET CUCKOO",
"26": "WHITE BROWED CRAKE"
},
"image_size": 224,
"initializer_range": 0.02,
"label2id": {
"ALBATROSS": 0,
"AMERICAN BITTERN": 1,
"ANDEAN LAPWING": 2,
"ASHY STORM PETREL": 3,
"BARN SWALLOW": 4,
"BLACK VENTED SHEARWATER": 5,
"BREWERS BLACKBIRD": 6,
"CALIFORNIA GULL": 7,
"CASPIAN TERN": 8,
"DOUBLE EYED FIG PARROT": 9,
"EURASIAN MAGPIE": 10,
"FAIRY PENGUIN": 11,
"IMPERIAL SHAQ": 12,
"MALLARD DUCK": 13,
"NORTHERN GANNET": 14,
"OYSTER CATCHER": 15,
"PURPLE SWAMPHEN": 16,
"PYGMY KINGFISHER": 17,
"RED BROWED FINCH": 18,
"RED TAILED HAWK": 19,
"ROCK DOVE": 20,
"SNOW GOOSE": 21,
"SPOON BILED SANDPIPER": 22,
"STRIPED OWL": 23,
"TRUMPTER SWAN": 24,
"VIOLET CUCKOO": 25,
"WHITE BROWED CRAKE": 26
},
"layer_norm_eps": 1e-05,
"mlp_ratio": 4.0,
"model_type": "swin",
"num_channels": 3,
"num_heads": [
4,
8,
16,
32
],
"num_layers": 4,
"out_features": [
"stage4"
],
"out_indices": [
4
],
"patch_size": 4,
"path_norm": true,
"qkv_bias": true,
"stage_names": [
"stem",
"stage1",
"stage2",
"stage3",
"stage4"
],
"torch_dtype": "float32",
"transformers_version": "4.45.2",
"use_absolute_embeddings": false,
"window_size": 7
}