wd-v1-4-swinv2-tagger-v2 / sw_jax_cv_config.json
SmilingWolf's picture
SwinV2 Tagger v2.1
cdb0c7f
raw
history blame
442 Bytes
{
"image_size": 448,
"model_name": "swinv2_base",
"model_args": {
"image_size": 448,
"patch_size": 4,
"in_chans": 3,
"num_classes": 9083,
"embed_dim": 128,
"window_size": 14,
"mlp_ratio": 4.0,
"qkv_bias": true,
"drop_rate": 0.0,
"attn_drop_rate": 0.0,
"drop_path_rate": 0.1,
"patch_norm": true,
"layer_norm_eps": 1e-05
}
}