deit-base-distilled_pt-16xb64_in1k.py

_base_ = './deit-small_pt-4xb256_in1k.py'

# model settings
model = dict(
    backbone=dict(type='DistilledVisionTransformer', arch='deit-base'),
    head=dict(type='DeiTClsHead', in_channels=768),
)

# dataset settings
train_dataloader = dict(batch_size=64)