提交 70e43c18 编写于 作者: Y Yibing Liu

add model path check in tuning script

上级 7d8402ab
...@@ -4,6 +4,7 @@ from __future__ import division ...@@ -4,6 +4,7 @@ from __future__ import division
from __future__ import print_function from __future__ import print_function
import sys import sys
import os
import numpy as np import numpy as np
import argparse import argparse
import functools import functools
...@@ -111,6 +112,8 @@ def tune(): ...@@ -111,6 +112,8 @@ def tune():
shuffle_method=None) shuffle_method=None)
# load parameters # load parameters
if not os.path.isfile(args.model_path):
raise IOError("Invaid model path: %s" % args.model_path)
parameters = paddle.parameters.Parameters.from_tar( parameters = paddle.parameters.Parameters.from_tar(
gzip.open(args.model_path)) gzip.open(args.model_path))
...@@ -124,6 +127,8 @@ def tune(): ...@@ -124,6 +127,8 @@ def tune():
logger.setLevel(level=logging.INFO) logger.setLevel(level=logging.INFO)
# init external scorer # init external scorer
logger.info("begin to initialize the external scorer for tuning") logger.info("begin to initialize the external scorer for tuning")
if not os.path.isfile(args.lang_model_path):
raise IOError("Invaid language model path: %s" % args.lang_model_path)
ext_scorer = Scorer( ext_scorer = Scorer(
alpha=args.alpha_from, alpha=args.alpha_from,
beta=args.beta_from, beta=args.beta_from,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册