fix: update pretrained url

2022-10-18 08:48:01 +00:00 · 2022-10-18 08:48:01 +00:00 · a5607e7d09
parent defcb638c3
commit a5607e7d09
1 changed files with 2 additions and 1 deletions
--- 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()