mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge pull request #1012 from RainFrost1/resnet50_variant 

fix resnet50_variant pretrained bugs
pull/1014/head
cuicheng01 2021-07-05 20:34:16 +08:00 committed by GitHub
parent 77557082d4 1b004933e1
commit 1c0b75e621
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ppcls/arch/backbone/variant_models/resnet_variant.py
View File

@ -1,5 +1,5 @@
from paddle.nn import Conv2D
from ppcls.arch.backbone.legendary_models.resnet import ResNet50
from ppcls.arch.backbone.legendary_models.resnet import ResNet50, MODEL_URLS, _load_pretrained
__all__ = ["ResNet50_last_stage_stride1"]
@ -17,6 +17,7 @@ def ResNet50_last_stage_stride1(pretrained=False, use_ssld=False, **kwargs):
return new_conv
match_re = "conv2d_4[4|6]"
model = ResNet50(pretrained=pretrained, use_ssld=use_ssld, **kwargs)
model = ResNet50(pretrained=False, use_ssld=use_ssld, **kwargs)
model.replace_sub(match_re, replace_function, True)
_load_pretrained(pretrained, model, MODEL_URLS["ResNet50"], use_ssld)
return model