Skip to content

Commit c1d6719

Browse files
david-ryan-snyderdanpovey
authored andcommitted
[scripts] fixing bug present since #2515 in steps/nnet3/train_raw_{dnn,rnn}.py (#2527)
1 parent 7605fe5 commit c1d6719

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

egs/wsj/s5/steps/nnet3/train_raw_dnn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -305,7 +305,7 @@ def train(args, run_opts):
305305

306306
if args.stage <= -1:
307307
logger.info("Preparing the initial network.")
308-
common_train_lib.prepare_initial_network(args.dir, run_opts, args.input_model)
308+
common_train_lib.prepare_initial_network(args.dir, run_opts, args.srand, args.input_model)
309309

310310
# set num_iters so that as close as possible, we process the data
311311
# $num_epochs times, i.e. $num_iters*$avg_num_jobs) ==

egs/wsj/s5/steps/nnet3/train_raw_rnn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -360,7 +360,7 @@ def train(args, run_opts):
360360

361361
if args.stage <= -1:
362362
logger.info("Preparing the initial network.")
363-
common_train_lib.prepare_initial_network(args.dir, run_opts, args.input_model)
363+
common_train_lib.prepare_initial_network(args.dir, run_opts, args.srand, args.input_model)
364364

365365
# set num_iters so that as close as possible, we process the data
366366
# $num_epochs times, i.e. $num_iters*$avg_num_jobs) ==

0 commit comments

Comments
 (0)