mmclassification/projects/internimage_classification/configs/internimage-base_8xb128_in1k-224.py
zzc98 3eaf719a64
[Feature] Add InternImage Classification project (#1569)
* [Feature] add internimage project

* [Feature] add internimage project

* update license

* [Feature] add internimage project

* [Feature] add internimage project

* [Feature] add internimage project

* [Feature] add internimage project

* [Feature] add internimage project

* [Feature] add internimage project

* update license

* [Feature] add internimage project

* [Feature] add internimage project

* [Feature] add internimage project

* [Feature] add internimage project

* update internimage configs

* support internimage project

* support internimage project

* support internimage project

* internimage
2023-06-13 19:11:54 +08:00

14 lines
313 B
Python

_base_ = './_base_.py'
model = dict(
backbone=dict(
stem_channels=112,
drop_path_rate=0.5,
stage_blocks=[4, 4, 21, 4],
groups=[7, 14, 28, 56],
layer_scale=1e-5,
post_norm=True),
head=dict(in_channels=1344))
optim_wrapper = dict(optimizer=dict(lr=0.0005))