From 7fae39ccf8f6301725de4c79a85cdd1ce5aebdb9 Mon Sep 17 00:00:00 2001 From: sshuair Date: Sat, 31 Jul 2021 19:01:30 +0800 Subject: [PATCH] support hrnet frozen stage --- mmseg/models/backbones/hrnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmseg/models/backbones/hrnet.py b/mmseg/models/backbones/hrnet.py index 6e92bb5023..0f064cff7d 100644 --- 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):