add lr_mult_list

pdseg/models/modeling/deeplab.py

@@ -280,7 +280,9 @@ def resnet_vd(input):
         dilation_dict = {3: 2}
     else:
         raise Exception("deeplab only support stride 8 or 16")
-    model = resnet_vd_backbone(layers, stem='deeplab')
+    lr_mult_list = cfg.MODEL.DEEPLAB.RESNET_LR_MULT_LIST
+    model = resnet_vd_backbone(
+        layers, stem='deeplab', lr_mult_list=lr_mult_list)
     data, decode_shortcuts = model.net(
         input,
         end_points=end_points,

pdseg/utils/config.py

@@ -206,6 +206,8 @@ cfg.MODEL.DEEPLAB.ENABLE_DECODER = True
 cfg.MODEL.DEEPLAB.ASPP_WITH_SEP_CONV = True
 # 解码器是否使用可分离卷积
 cfg.MODEL.DEEPLAB.DECODER_USE_SEP_CONV = True
+# resnet_vd分阶段学习率
+cfg.MODEL.DEEPLAB.RESNET_LR_MULT_LIST = [1.0, 1.0, 1.0, 1.0, 1.0]
 
 ########################## UNET模型配置 #######################################
 # 上采样方式, 默认为双线性插值