EasyCV/configs/classification/imagenet/hrnet/imagenet_hrnetw40_jpg.py
Chen Jiayu 3ee118f065
Expand cls model zoo (#55)
1. expand cls model zoo
2. uniform load_pretrained
2022-06-01 11:01:29 +08:00

6 lines
209 B
Python

_base_ = './hrnetw18_b32x8_100e_jpg.py'
# model settings
model = dict(
backbone=dict(type='HRNet', arch='w40', multi_scale_output=True),
neck=dict(type='HRFuseScales', in_channels=(40, 80, 160, 320)))