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

support web link for 'Global.pretrained_model' when export models

pull/1820/merge
HydrogenSulfate 2022-12-13 07:43:41 +00:00
parent 0288285c91
commit 51d71016ac
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ppcls/engine/engine.py
View File

@ -548,7 +548,10 @@ class Engine(object):
"use_multilabel",
False) or "ATTRMetric" in self.config["Metric"]["Eval"][0]
model = ExportModel(self.config["Arch"], self.model, use_multilabel)
if self.config["Global"]["pretrained_model"] is not None:
if self.config["Global"]["pretrained_model"].startswith("http"):
load_dygraph_pretrain_from_url(
model.base_model, self.config["Global"]["pretrained_model"])
else:
load_dygraph_pretrain(model.base_model,
self.config["Global"]["pretrained_model"])