{ | |
"architectures": [ | |
"DaViTModel" | |
], | |
"conv_at_attn": true, | |
"conv_at_ffn": true, | |
"depths": [ | |
1, | |
1, | |
9, | |
1 | |
], | |
"drop_path_rate": 0.1, | |
"embed_dims": [ | |
256, | |
512, | |
1024, | |
2048 | |
], | |
"enable_checkpoint": false, | |
"in_chans": 3, | |
"mlp_ratio": 4.0, | |
"model_type": "davit", | |
"norm_layer": "layer_norm", | |
"num_groups": [ | |
8, | |
16, | |
32, | |
64 | |
], | |
"num_heads": [ | |
8, | |
16, | |
32, | |
64 | |
], | |
"patch_padding": [ | |
3, | |
1, | |
1, | |
1 | |
], | |
"patch_prenorm": [ | |
false, | |
true, | |
true, | |
true | |
], | |
"patch_size": [ | |
7, | |
3, | |
3, | |
3 | |
], | |
"patch_stride": [ | |
4, | |
2, | |
2, | |
2 | |
], | |
"projection_dim": 1024, | |
"qkv_bias": true, | |
"torch_dtype": "float16", | |
"transformers_version": "4.43.3", | |
"window_size": 12 | |
} | |