mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:53:55 +08:00
8 lines
200 B
Python
8 lines
200 B
Python
_base_ = './deit-small_pt-4xb256_in1k.py'
|
|
|
|
# model settings
|
|
model = dict(
|
|
backbone=dict(type='DistilledVisionTransformer', arch='deit-tiny'),
|
|
head=dict(type='DeiTClsHead', in_channels=192),
|
|
)
|