Skip to content

Commit e346b9e

Browse files
committed
fix
1 parent 9d0e700 commit e346b9e

File tree

2 files changed

+13
-8
lines changed

2 files changed

+13
-8
lines changed

egs/ami/s5b/local/rnnlm/tuning/run_lstm_tdnn_bs_1a.sh

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,9 @@ epochs=90
2323
mic=sdm1
2424
stage=-10
2525
train_stage=0
26-
alpha=0.8
27-
back_interval=1
26+
# backstitch options
27+
alpha=0.8 # backstitch training scale
28+
back_interval=1 # backstitch training interval
2829

2930
. utils/parse_options.sh
3031
train=data/$mic/train/text
@@ -95,7 +96,8 @@ if [ $stage -le 2 ]; then
9596
fi
9697

9798
if [ $stage -le 3 ]; then
98-
backstitch_opt="--rnnlm.backstitch-scale $alpha --rnnlm.backstitch-interval $back_interval --embedding.backstitch-scale $alpha --embedding.backstitch-interval $back_interval"
99+
backstitch_opt="--backstitch-training-scale $alpha \
100+
--backstitch-training-interval $back_interval"
99101
rnnlm/train_rnnlm.sh --embedding_l2 $embedding_l2 \
100102
--stage $train_stage \
101103
--num-epochs $epochs --cmd "queue.pl" $backstitch_opt $dir

scripts/rnnlm/train_rnnlm.sh

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@ final_effective_lrate=0.0001
2323
embedding_l2=0.005
2424
embedding_lrate_factor=0.1 # the embedding learning rate is the
2525
# nnet learning rate times this factor.
26-
backstitch_scale=0.0 # backstitch training scale
27-
backstitch_interval=1 # backstitch training interval
26+
backstitch_training_scale=0.0 # backstitch training scale
27+
backstitch_training_interval=1 # backstitch training interval
2828
cmd=run.pl # you might want to set this to queue.pl
2929

3030
# some options passed into rnnlm-get-egs, relating to sampling.
@@ -119,8 +119,8 @@ final_effective_lrate=$final_effective_lrate
119119
embedding_lrate_factor=$embedding_lrate_factor
120120
sample_group_size=$sample_group_size
121121
num_samples=$num_samples
122-
backstitch_scale=$backstitch_scale
123-
backstitch_interval=$backstitch_interval
122+
backstitch_training_scale=$backstitch_training_scale
123+
backstitch_training_interval=$backstitch_training_interval
124124
EOF
125125

126126

@@ -143,7 +143,10 @@ while [ $x -lt $num_iters ]; do
143143
fi
144144
if $use_gpu_for_diagnostics; then queue_gpu_opt="--gpu 1"; gpu_opt="--use-gpu=yes";
145145
else gpu_opt=''; queue_gpu_opt=''; fi
146-
backstitch_opt="--backstitch-training-scale=$backstitch_scale --backstitch-training-interval=$backstitch_interval"
146+
backstitch_opt="--rnnlm.backstitch-training-scale=$backstitch_training_scale \
147+
--rnnlm.backstitch-training-interval=$backstitch_training_interval \
148+
--embedding.backstitch-training-scale=$backstitch_training_scale \
149+
--embedding.backstitch-training-interval=$backstitch_training_interval"
147150
[ -f $dir/.error ] && rm $dir/.error
148151
$cmd $queue_gpu_opt $dir/log/compute_prob.$x.log \
149152
rnnlm-get-egs $(cat $dir/special_symbol_opts.txt) \

0 commit comments

Comments
 (0)