提交 da4bb135 编写于 作者: W wangguanzhong 提交者: GitHub

minor fix for function name (#3471)

上级 2f55a940
......@@ -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))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册