From 4bc42bad35f1c794ee549acdb7f3f6a81387f90f Mon Sep 17 00:00:00 2001 From: BebDong Date: Sun, 14 Nov 2021 09:12:13 +0800 Subject: [PATCH] Update hrnet.py (#1713) Correct the trunk name in function "hrnet_w18_small_v2_s" --- gluoncv/model_zoo/hrnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gluoncv/model_zoo/hrnet.py b/gluoncv/model_zoo/hrnet.py index 20f04e7bd7..81d088d845 100644 --- a/gluoncv/model_zoo/hrnet.py +++ b/gluoncv/model_zoo/hrnet.py @@ -741,7 +741,7 @@ def hrnet_w18_small_v1_s(**kwargs): def hrnet_w18_small_v2_s(**kwargs): r"""hrnet_w18_small_v2 for cityscapes segmentation """ - return get_hrnet('w18_small_v1', stage_interp_type='bilinear_like', purpose='seg', + return get_hrnet('w18_small_v2', stage_interp_type='bilinear_like', purpose='seg', norm_kwargs={'momentum': 0.99}, num_classes=19, **kwargs) def hrnet_w48_s(**kwargs):