CoAtNet-2 backbone (Dai et al., 2021).
Wider variant: blocks_per_stage=(2, 6, 14, 2),
dims=(128, 256, 512, 1024), stem_width=128,
attn_heads=(16, 32). Approximately 75M parameters, 84.1%
ImageNet-1k top-1 at 224×224 (Table 5).
CoAtNet-2 backbone (Dai et al., 2021).
Wider variant: blocks_per_stage=(2, 6, 14, 2),
dims=(128, 256, 512, 1024), stem_width=128,
attn_heads=(16, 32). Approximately 75M parameters, 84.1%
ImageNet-1k top-1 at 224×224 (Table 5).