Update hrnet.py (#1713)

Correct the trunk name in function "hrnet_w18_small_v2_s"
dmlc · Nov 14, 2021 · 4bc42ba · 4bc42ba
1 parent f22650a
commit 4bc42ba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):