Explorar o código

fix key error

will-jl944 %!s(int64=4) %!d(string=hai) anos
pai
achega
e94698ed60
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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',