浏览代码

fix key error

will-jl944 4 年之前
父节点
当前提交
e94698ed60
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      dygraph/paddlex/utils/checkpoint.py

+ 1 - 1
dygraph/paddlex/utils/checkpoint.py

@@ -148,7 +148,7 @@ imagenet_weights = {
     'https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileNetV3_large_x1_0_pretrained.pdparams',
     'MobileNetV3_large_x1_25_IMAGENET':
     'https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileNetV3_large_x1_25_pretrained.pdparams',
-    'MobileNetV3_large_x1_0_ssld':
+    'MobileNetV3_large_x1_0_ssld_IMAGENET':
     'https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/MobileNetV3_large_x1_0_ssld_pretrained.pdparams',
     'AlexNet_IMAGENET':
     'https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/AlexNet_pretrained.pdparams',