mirror of
https://github.com/open-mmlab/mmpretrain.git
synced 2025-06-03 14:59:18 +08:00
2eb27d0d4e
* add PoolFormer * fix some typos in PoolFormer * fix lint error * modify out_indices and gap * fix typo * fix lint * fix typo * fix typo in poolforemr README * fix lint * Update some paths * Refactor freeze_stages method * Add unit tests * Fix lint Co-authored-by: mzr1996 <mzr1996@163.com>
9 lines
252 B
Python
9 lines
252 B
Python
_base_ = [
|
|
'../_base_/models/poolformer/poolformer_m36.py',
|
|
'../_base_/datasets/imagenet_bs128_poolformer_medium_224.py',
|
|
'../_base_/schedules/imagenet_bs1024_adamw_swin.py',
|
|
'../_base_/default_runtime.py',
|
|
]
|
|
|
|
optimizer = dict(lr=4e-3)
|