diff --git a/hub_module/modules/image/classification/mobilenet_v2_imagenet_ssld/module.py b/hub_module/modules/image/classification/mobilenet_v2_imagenet_ssld/module.py index 45393492d24aaf17e8a7ae6b4b7f62edfc8b1818..ce69252f6f6332507d0548d7029fca56b202f70c 100644 --- a/hub_module/modules/image/classification/mobilenet_v2_imagenet_ssld/module.py +++ b/hub_module/modules/image/classification/mobilenet_v2_imagenet_ssld/module.py @@ -189,7 +189,7 @@ class MobileNet(nn.Layer): print("load custom checkpoint success") else: - checkpoint = os.path.join(self.directory, 'MobileNetV2_ssld_pretrained.pdparams') + checkpoint = os.path.join(self.directory, 'mobilenet_v2_ssld.pdparams.pdparams') if not os.path.exists(checkpoint): os.system( 'wget https://paddlehub.bj.bcebos.com/dygraph/image_classification/mobilenet_v2_ssld.pdparams -O ' +