{ "ape": false, "attn_drop_rate": 0.0, "depths": [ 2, 2, 18, 2 ], "drop_path_rate": 0.2, "drop_rate": 0.0, "embed_dim": 128, "img_size": 256, "in_chans": 3, "mlp_ratio": 4.0, "num_classes": 1000, "num_heads": [ 4, 8, 16, 32 ], "output_dim": 512, "p": 3.0, "patch_norm": true, "patch_size": 4, "pretrained": null, "pretrained_window_sizes": [ 12, 12, 12, 6 ], "qkv_bias": true, "use_checkpoint": false, "window_size": 16 }