Model card for davit-base-ws4-cav2-in1k-160

DaViT-Base (ws=5, channel_attn_v2) trained from scratch on ImageNet-1K @160x160.

Training Recipe:

• DaViT supervised recipe -- Table 5.1 (https://arxiv.org/pdf/2204.03645)
• BF16 AMP

Model Details

• Model: davit_base
• Model Kwargs:
  • channel_attn_v2: True
  • window_size: 5
• Input Size:
  • 3x160x160
• Top-1 Accuracy: 83.46999993804931

Model Usage

import timm
model = timm.create_model('hf-hub:PRadecki/davit-base-ws4-cav2-in1k-128', pretrained=True)
model.eval()