提交 b4a6c4b5 编写于 作者: H haoyuying

revise download address

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