From 05996d63c64b5185d1237ad7ee54f41dfe4c7eae Mon Sep 17 00:00:00 2001 From: michaelowenliu Date: Fri, 28 Aug 2020 19:08:24 +0800 Subject: [PATCH] add dim_reduction to PPModule --- dygraph/models/pspnet.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/dygraph/models/pspnet.py b/dygraph/models/pspnet.py index 9017484d..6ed08c19 100644 --- a/dygraph/models/pspnet.py +++ b/dygraph/models/pspnet.py @@ -209,23 +209,23 @@ class PPModule(fluid.dygraph.Layer): @manager.MODELS.add_component def pspnet_resnet101_vd(*args, **kwargs): - pretrained_model = "/mnt/liuyi22/PaddlePaddle/PaddleClas/pretrained/resnet101_vd_ssld_imagenet" + pretrained_model = None return PSPNet(backbone='ResNet101_vd', pretrained_model=pretrained_model, **kwargs) @manager.MODELS.add_component def pspnet_resnet101_vd_os8(*args, **kwargs): - pretrained_model = "/mnt/liuyi22/PaddlePaddle/PaddleClas/pretrained/resnet101_vd_ssld_imagenet" + pretrained_model = None return PSPNet(backbone='ResNet101_vd', output_stride=8, pretrained_model=pretrained_model, **kwargs) @manager.MODELS.add_component def pspnet_resnet50_vd(*args, **kwargs): - pretrained_model = "/mnt/liuyi22/PaddlePaddle/PaddleClas/pretrained/resnet50_vd_ssld_v2_imagenet" + pretrained_model = None return PSPNet(backbone='ResNet50_vd', pretrained_model=pretrained_model, **kwargs) @manager.MODELS.add_component def pspnet_resnet50_vd_os8(*args, **kwargs): - pretrained_model = "/mnt/liuyi22/PaddlePaddle/PaddleClas/pretrained/resnet50_vd_ssld_v2_imagenet" + pretrained_model = None return PSPNet(backbone='ResNet50_vd', output_stride=8, pretrained_model=pretrained_model, **kwargs) -- GitLab