Model card for davit-small-ws4-cav2-in1k-128

Training Recipe:

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

Model Details

• Model: davit_small
• Model Kwargs:
  • channel_attn_v2: True
  • window_size: 4
• Input Size:
  • 3x128x128
• Top-1 Accuracy: 81.39399999572754

Model Usage