提交 9e1332e6 编写于 作者: C chenguowei01

update deeplab.py

上级 0d9aa902
......@@ -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(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册