diff --git a/ppdet/utils/checkpoint.py b/ppdet/utils/checkpoint.py index 37eddac1c44575d7351075abfc4ced14eb63631e..d19653494732685fe8eb45aa3d3109e0342df8d0 100644 --- a/ppdet/utils/checkpoint.py +++ b/ppdet/utils/checkpoint.py @@ -89,7 +89,7 @@ def load_params(exe, prog, path, ignore_params=[]): """ if is_url(path): - path = _get_weights_path(path) + path = _get_weight_path(path) if not os.path.exists(path): raise ValueError("Model pretrain path {} does not " @@ -125,7 +125,7 @@ def load_checkpoint(exe, prog, path): path (string): URL string or loca model path. """ if is_url(path): - path = _get_weights_path(path) + path = _get_weight_path(path) if not os.path.exists(path): raise ValueError("Model checkpoint path {} does not " @@ -178,7 +178,7 @@ def load_and_fusebn(exe, prog, path): logger.info('Load model and fuse batch norm from {}...'.format(path)) if is_url(path): - path = _get_weights_path(path) + path = _get_weight_path(path) if not os.path.exists(path): raise ValueError("Model path {} does not exists.".format(path))