mmsegmentation/configs/hrnet/fcn_hr18s_4xb4-80k_loveda-512x512.py
MengzhangLI bd1097ac02
[Fix] Fix several config file errors in 2.0 (New) (#1994)
* [Fix] Fix several config file errors in 2.0

* change _base_ config file name in configs
2022-08-30 20:20:05 +08:00

10 lines
372 B
Python

_base_ = './fcn_hr18_4xb4-80k_loveda-512x512.py'
model = dict(
pretrained='open-mmlab://msra/hrnetv2_w18_small',
backbone=dict(
extra=dict(
stage1=dict(num_blocks=(2, )),
stage2=dict(num_blocks=(2, 2)),
stage3=dict(num_modules=3, num_blocks=(2, 2, 2)),
stage4=dict(num_modules=2, num_blocks=(2, 2, 2, 2)))))