mirror of
https://github.com/open-mmlab/mmpretrain.git
synced 2025-06-03 14:59:18 +08:00
10 lines
274 B
Python
10 lines
274 B
Python
_base_ = './deit-base_ft-16xb32_in1k-384px.py'
|
|
|
|
# model settings
|
|
model = dict(
|
|
backbone=dict(type='DistilledVisionTransformer'),
|
|
head=dict(type='DeiTClsHead'),
|
|
# Change to the path of the pretrained model
|
|
# init_cfg=dict(type='Pretrained', checkpoint=''),
|
|
)
|