From 96f4658b3901d79fe6ffab464af24ebf9f21c6d7 Mon Sep 17 00:00:00 2001 From: Jan Trmal Date: Mon, 5 Aug 2019 18:33:57 +0000 Subject: [PATCH] prevent crash when adapting on much smaller data than the tree was created0 --- egs/wsj/s5/steps/nnet3/chain/train.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/egs/wsj/s5/steps/nnet3/chain/train.py b/egs/wsj/s5/steps/nnet3/chain/train.py index 91b7df4e8df..1af33590781 100755 --- a/egs/wsj/s5/steps/nnet3/chain/train.py +++ b/egs/wsj/s5/steps/nnet3/chain/train.py @@ -60,6 +60,13 @@ def get_args(): should halve --trainer.samples-per-iter. May be a comma-separated list of alternatives: first width is the 'principal' chunk-width, used preferentially""") + parser.add_argument("--egs.nj", type=int, required=False, + default=0, dest="egs_nj", + help="""Number of jobs to use when generating egs. + Default: the same number as used for tree generation. + You probably do not need to tweak this, unless you + want to adapt a neural network on some different, + smaller-size data.""") # chain options parser.add_argument("--chain.lm-opts", type=str, dest='lm_opts', @@ -283,7 +290,10 @@ def train(args, run_opts): shutil.copy('{0}/phones.txt'.format(args.tree_dir), args.dir) # Set some variables. - num_jobs = common_lib.get_number_of_jobs(args.tree_dir) + if args.egs_nj <= 0: + num_jobs = common_lib.get_number_of_jobs(args.tree_dir) + else: + num_jobs = args.egs_nj feat_dim = common_lib.get_feat_dim(args.feat_dir) ivector_dim = common_lib.get_ivector_dim(args.online_ivector_dir) ivector_id = common_lib.get_ivector_extractor_id(args.online_ivector_dir)