add model path check in tuning script

pull/2/head
Yibing Liu 8 years ago
parent 7d8402ab14
commit 70e43c184c

@ -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():
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():
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,

Loading…
Cancel
Save