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

Release/2.5.1 (#2990) 

* add hgnetv2 (#2987)

* support load ssld state1 pretrain (#2988)
pull/2994/head
cuicheng01 2023-09-27 00:03:01 +08:00 committed by GitHub
parent 43e6382aa3
commit bfea8e83a6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
ppcls/utils/save_load.py
View File

@ -74,9 +74,13 @@ def load_dygraph_pretrain_from_url(model,
use_ssld=False,
use_imagenet22k_pretrained=False,
use_imagenet22kto1k_pretrained=False):
if use_ssld:
pretrained_url = pretrained_url.replace("_pretrained",
"_ssld_pretrained")
if "ssld" not in pretrained_url:
if use_ssld:
pretrained_url = pretrained_url.replace("_pretrained",
"_ssld_pretrained")
else:
pretrained_url = pretrained_url.replace("ssld_pretrained",
"ssld_stage1_pretrained")
if use_imagenet22k_pretrained:
pretrained_url = pretrained_url.replace("_pretrained",
"_22k_pretrained")