internimage-tiny_8xb128_in1k-224.py 179 Bytes
Newer Older
limm's avatar
limm committed
1
2
3
4
5
6
7
8
_base_ = './_base_.py'

model = dict(
    backbone=dict(
        stem_channels=64,
        drop_path_rate=0.1,
        stage_blocks=[4, 4, 18, 4],
        groups=[4, 8, 16, 32]))