diff --git a/ppcls/arch/backbone/model_zoo/peleenet.py b/ppcls/arch/backbone/model_zoo/peleenet.py index eef8139e6f7d981476109ea6b27f9c06048f8512..95153e2e12a597c4c35cf6fc2f074af07d676fc1 100644 --- 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()