fixed backbone bug (#684)

PaddlePaddle · Oct 14, 2021 · 9ca4f40 · 9ca4f40
1 parent 7220d93
commit 9ca4f40
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/x2paddle/project_convertor/pytorch/models/resnet.py b/x2paddle/project_convertor/pytorch/models/resnet.py
@@ -302,7 +302,7 @@ def _resnet(arch: str,
             **kwargs: Any) -> ResNet:
     model = ResNet(block, layers, **kwargs)
     if pretrained:
-        state_dict = get_weights_path_from_url(model_urls[arch])
+        state_dict = paddle.load(get_weights_path_from_url(model_urls[arch]))
         model.load_dict(state_dict)
     return model
 

diff --git a/x2paddle/project_convertor/pytorch/models/vgg.py b/x2paddle/project_convertor/pytorch/models/vgg.py
@@ -109,7 +109,7 @@ def _vgg(arch: str, cfg: str, batch_norm: bool, pretrained: bool,
         kwargs['init_weights'] = False
     model = VGG(make_layers(cfgs[cfg], batch_norm=batch_norm), **kwargs)
     if pretrained:
-        state_dict = get_weights_path_from_url(model_urls[arch])
+        state_dict = paddle.load(get_weights_path_from_url(model_urls[arch]))
         model.load_dict(state_dict)
     return model