File size: 900 Bytes
142b9ca beb6bf3 7020d94 142b9ca |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
{
"_name_or_path": "/datasets/work/hb-mlaifsp-mm/work/checkpoints/uniformer_base_tl_384",
"architectures": [
"UniFormerModel"
],
"attn_drop_rate": 0.0,
"auto_map": {
"AutoConfig": "configuration_uniformer.UniFormerWithProjectionHeadConfig",
"AutoModel": "modelling_uniformer.UniFormerModel"
},
"conv_stem": false,
"layer_scale": false,
"init_value": 1e-6,
"depth": [
5,
8,
20,
7
],
"drop_path_rate": 0.3,
"drop_rate": 0.0,
"embed_dim": [
64,
128,
320,
512
],
"head_dim": 64,
"image_size": 384,
"in_chans": 3,
"layer_norm_eps": 1e-06,
"mlp_ratio": 4,
"model_type": "uniformer",
"num_classes": 1000,
"patch_size": [
4,
2,
2,
2
],
"projection_size": null,
"qk_scale": null,
"qkv_bias": true,
"representation_size": null,
"torch_dtype": "float32",
"transformers_version": "4.39.3"
}
|