mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:50:04 +08:00
f9a2b04cee
* 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.
8 lines
201 B
Python
8 lines
201 B
Python
_base_ = './deit-small_pt-4xb256_in1k.py'
|
|
|
|
# model settings
|
|
model = dict(
|
|
backbone=dict(type='DistilledVisionTransformer', arch='deit-small'),
|
|
head=dict(type='DeiTClsHead', in_channels=384),
|
|
)
|