From bc5e0d7ab949c3b4ec9a5516b54140edca0a121e Mon Sep 17 00:00:00 2001 From: wuyefeilin <30919197+wuyefeilin@users.noreply.github.com> Date: Wed, 23 Sep 2020 18:01:58 +0800 Subject: [PATCH] update mobilenetv3.py (#405) --- .../deeplabv3p_mobilenetv3_large_cityscapes_769x769_160k.yml | 2 ++ .../deeplabv3p_xception65_cityscapes_769x769_160k.yml | 2 ++ dygraph/paddleseg/models/backbones/mobilenetv3.py | 3 ++- 3 files changed, 6 insertions(+), 1 deletion(-) diff --git a/dygraph/configs/deeplabv3p/deeplabv3p_mobilenetv3_large_cityscapes_769x769_160k.yml b/dygraph/configs/deeplabv3p/deeplabv3p_mobilenetv3_large_cityscapes_769x769_160k.yml index 82a0649a..9ae6093a 100644 --- a/dygraph/configs/deeplabv3p/deeplabv3p_mobilenetv3_large_cityscapes_769x769_160k.yml +++ b/dygraph/configs/deeplabv3p/deeplabv3p_mobilenetv3_large_cityscapes_769x769_160k.yml @@ -1,5 +1,7 @@ _base_: '../_base_/cityscapes.yml' +iters: 160000 + model: type: DeepLabV3 backbone: diff --git a/dygraph/configs/deeplabv3p/deeplabv3p_xception65_cityscapes_769x769_160k.yml b/dygraph/configs/deeplabv3p/deeplabv3p_xception65_cityscapes_769x769_160k.yml index c604ee7e..17d854d0 100644 --- a/dygraph/configs/deeplabv3p/deeplabv3p_xception65_cityscapes_769x769_160k.yml +++ b/dygraph/configs/deeplabv3p/deeplabv3p_xception65_cityscapes_769x769_160k.yml @@ -1,5 +1,7 @@ _base_: '../_base_/cityscapes.yml' +iters: 160000 + model: type: DeepLabV3 backbone: diff --git a/dygraph/paddleseg/models/backbones/mobilenetv3.py b/dygraph/paddleseg/models/backbones/mobilenetv3.py index 14baf3e9..8bbd0c5e 100644 --- a/dygraph/paddleseg/models/backbones/mobilenetv3.py +++ b/dygraph/paddleseg/models/backbones/mobilenetv3.py @@ -25,7 +25,8 @@ import paddle.nn as nn import paddle.nn.functional as F from paddle.nn import Conv2d, AdaptiveAvgPool2d from paddle.nn import SyncBatchNorm as BatchNorm -from paddle.regularizer import L2Decay +# from paddle.regularizer import L2Decay +from paddle.fluid.regularizer import L2Decay from paddle import ParamAttr from paddleseg.models.common import layer_libs, activation -- GitLab