diff --git a/pdseg/models/modeling/deeplab.py b/pdseg/models/modeling/deeplab.py index 454a5c4700e6b68a1be177b735c1711dc69564cc..d05a0d049fdf548b0ed6cfb0f96c48b61e29de27 100644 --- a/pdseg/models/modeling/deeplab.py +++ b/pdseg/models/modeling/deeplab.py @@ -249,6 +249,8 @@ def resnet_vd(input): else: raise Exception("deeplab only support stride 8 or 16") lr_mult_list = cfg.MODEL.DEEPLAB.BACKBONE_LR_MULT_LIST + if lr_mult_list is None: + lr_mult_list = [1.0, 1.0, 1.0, 1.0, 1.0] model = resnet_vd_backbone( layers, stem='deeplab', lr_mult_list=lr_mult_list) data, decode_shortcuts = model.net(