support hrnet frozen stage

open-mmlab · Jul 31, 2021 · 7fae39c · 7fae39c
1 parent 0706a00
commit 7fae39c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmseg/models/backbones/hrnet.py b/mmseg/models/backbones/hrnet.py
@@ -287,7 +287,7 @@ def __init__(self,
                  norm_cfg=dict(type='BN', requires_grad=True),
                  norm_eval=False,
                  with_cp=False,
-                 frozen_stages=1,
+                 frozen_stages=-1,
                  zero_init_residual=False,
                  pretrained=None,
                  init_cfg=None):