mmclassification/configs/deit/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),
)

# data settings
data = dict(samples_per_gpu=64, workers_per_gpu=5)
[Feature] Add DeiT backbone and checkpoints. (#576) * Support DeiT backbone. * Use hook to automatically resize pos embed * Update ViT training setting * Add deit configs and update docs * Fix vit arch assertion * Remove useless init function * Add unit tests. * Fix resize_pos_embed for DeiT * Improve according to comments. 2021-12-15 22:44:57 +08:00			`_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),`
			`)`

			`# data settings`
			`data = dict(samples_per_gpu=64, workers_per_gpu=5)`