mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:53:55 +08:00
aaf127c5e1
* rebase * update filename * update URL * update UT * fix lint * update head * add efficientformer * update filename * update UT * fix lint * update configs * rebase * fix unit tests * Fix comments and docs. Co-authored-by: mzr1996 <mzr1996@163.com>
19 lines
598 B
Python
19 lines
598 B
Python
model = dict(
|
|
type='ImageClassifier',
|
|
backbone=dict(
|
|
type='EfficientFormer',
|
|
arch='l1',
|
|
drop_path_rate=0,
|
|
init_cfg=[
|
|
dict(
|
|
type='TruncNormal',
|
|
layer=['Conv2d', 'Linear'],
|
|
std=.02,
|
|
bias=0.),
|
|
dict(type='Constant', layer=['GroupNorm'], val=1., bias=0.),
|
|
dict(type='Constant', layer=['LayerScale'], val=1e-5)
|
|
]),
|
|
neck=dict(type='GlobalAveragePooling', dim=1),
|
|
head=dict(
|
|
type='EfficientFormerClsHead', in_channels=448, num_classes=1000))
|