fix: update pretrained url

e21c060f · gaotingquan · Tingquan Gao · 8b86c8b0 · e21c060f
显示空白变更内容
内联并排

Showing with 2 addition and 1 deletion

ppcls/arch/backbone/model_zoo/peleenet.py ppcls/arch/backbone/model_zoo/peleenet.py +2 -1

未找到文件。
--- a/ppcls/arch/backbone/model_zoo/peleenet.py
+++ b/ppcls/arch/backbone/model_zoo/peleenet.py
@@ -25,7 +25,8 @@ from paddle.nn.initializer import Normal, Constant
 from ....utils.save_load import load_dygraph_pretrain, load_dygraph_pretrain_from_url

 MODEL_URLS = {
-    "PeleeNet": ""  # TODO
+    "PeleeNet":
+    "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/PeleeNet_pretrained.pdparams"
 }

 __all__ = MODEL_URLS.keys()