transfer-learning / ViT-H /config.json
Dataset-and-Model's picture
Upload folder using huggingface_hub
c5d0d07 verified
Raw History Blame Contribute Delete
2.24 kB
{
"_name_or_path": "/home/yoru/.cache/huggingface/hub/models--google--vit-hybrid-base-bit-384/snapshots/e8cc9d5fe904240b3b0035837daeabff06202bf2",
"architectures": [
"ViTHybridForImageClassification"
],
"attention_probs_dropout_prob": 0.0,
"backbone": null,
"backbone_config": {
"depths": [
3,
4,
9
],
"embedding_dynamic_padding": true,
"global_padding": "SAME",
"layer_type": "bottleneck",
"model_type": "bit",
"out_features": [
"stage3"
],
"out_indices": [
3
],
"stage_names": [
"stem",
"stage1",
"stage2",
"stage3"
]
},
"backbone_featmap_shape": [
1,
1024,
24,
24
],
"backbone_kwargs": null,
"hidden_act": "gelu",
"hidden_dropout_prob": 0.0,
"hidden_size": 768,
"id2label": {
"0": "Civet",
"1": "Emerald_Dove",
"2": "Foemosan_Gem-faced_Civet",
"3": "Formosan_Ferret-badger",
"4": "Formosan_Sambar_Deer",
"5": "Formosan_Serow",
"6": "Formosan_Sika_Deer",
"7": "Formosan_Yellow-throated_Marten",
"8": "Formosan_macaque",
"9": "Formosan_wild_boar",
"10": "Mouse",
"11": "Reeves's_Muntjac",
"12": "Swinhoe's_Pheasant",
"13": "Taiwan_Partridge",
"14": "Urva",
"15": "cat",
"16": "dog",
"17": "squirrel"
},
"image_size": 384,
"initializer_range": 0.02,
"intermediate_size": 3072,
"label2id": {
"Civet": 0,
"Emerald_Dove": 1,
"Foemosan_Gem-faced_Civet": 2,
"Formosan_Ferret-badger": 3,
"Formosan_Sambar_Deer": 4,
"Formosan_Serow": 5,
"Formosan_Sika_Deer": 6,
"Formosan_Yellow-throated_Marten": 7,
"Formosan_macaque": 8,
"Formosan_wild_boar": 9,
"Mouse": 10,
"Reeves's_Muntjac": 11,
"Swinhoe's_Pheasant": 12,
"Taiwan_Partridge": 13,
"Urva": 14,
"cat": 15,
"dog": 16,
"squirrel": 17
},
"layer_norm_eps": 1e-12,
"model_type": "vit-hybrid",
"num_attention_heads": 12,
"num_channels": 3,
"num_hidden_layers": 12,
"patch_size": 1,
"problem_type": "single_label_classification",
"qkv_bias": true,
"torch_dtype": "float32",
"transformers_version": "4.40.2",
"use_pretrained_backbone": false,
"use_timm_backbone": false
}