From b4a6c4b5edccede7f7433d20b6292cc3bcab6026 Mon Sep 17 00:00:00 2001 From: haoyuying <18844182690@163.com> Date: Mon, 12 Oct 2020 11:32:23 +0800 Subject: [PATCH] revise download address --- .../image/classification/resnet50_vd_imagenet_ssld/module.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/hub_module/modules/image/classification/resnet50_vd_imagenet_ssld/module.py b/hub_module/modules/image/classification/resnet50_vd_imagenet_ssld/module.py index 976aadb5..3f9528e8 100644 --- a/hub_module/modules/image/classification/resnet50_vd_imagenet_ssld/module.py +++ b/hub_module/modules/image/classification/resnet50_vd_imagenet_ssld/module.py @@ -222,10 +222,10 @@ class ResNet50_vd(nn.Layer): print("load custom checkpoint success") else: - checkpoint = os.path.join(self.directory, 'ResNet50_vd.pdparams') + checkpoint = os.path.join(self.directory, 'resnet50_vd_ssld.pdparams') if not os.path.exists(checkpoint): os.system( - 'wget https://bj.bcebos.com/paddlehub/model/image/object_detection/yolov3_70000.pdparams -O ' + + 'wget https://paddlehub.bj.bcebos.com/dygraph/image_classification/resnet50_vd_ssld.pdparams -O ' + checkpoint) model_dict = paddle.load(checkpoint)[0] self.set_dict(model_dict) -- GitLab