deit-tiny_pt-4xb256_in1k.py 204 Bytes
Newer Older
1
2
3
4
5
6
7
_base_ = './deit-small_pt-4xb256_in1k.py'

# model settings
model = dict(
    backbone=dict(type='VisionTransformer', arch='deit-tiny'),
    head=dict(type='VisionTransformerClsHead', in_channels=192),
)