mmpretrain/configs/convmixer/convmixer-1536-20_10xb64_in1k.py
Hubert 04cb42a768
[Feature] Support ConvMixer. (#716)
* basic support for ConvMixer

* simplify

* add data pipeine config for timm

* Add model readme and metafile

* add unittest for convmixer

* add copyright

* modify

* add tests

* update model

* add conv2dAdaptivePadding replacement

* update model index

* fix comments

* Update checkpoint path

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-03-23 15:23:57 +08:00

11 lines
301 B
Python

_base_ = [
'../_base_/models/convmixer/convmixer-1536-20.py',
'../_base_/datasets/imagenet_bs64_convmixer_224.py',
'../_base_/schedules/imagenet_bs1024_adamw_swin.py',
'../_base_/default_runtime.py',
]
optimizer = dict(lr=0.01)
runner = dict(type='EpochBasedRunner', max_epochs=150)