Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 10 additions & 0 deletions egs/wsj/s5/local/chain/tuning/run_tdnn_1g.sh
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,14 @@ train_set=train_si284
test_sets="test_dev93 test_eval92"
gmm=tri4b # this is the source gmm-dir that we'll use for alignments; it
# should have alignments for the specified training data.

num_threads_ubm=32

nj_extractor=10
# It runs a JOB with '-pe smp N', where N=$[threads*processes]
num_threads_extractor=4
num_processes_extractor=4

nnet3_affix= # affix for exp dirs, e.g. it was _cleaned in tedlium.

# Options which are not passed through to run_ivector_common.sh
Expand Down Expand Up @@ -79,6 +86,9 @@ local/nnet3/run_ivector_common.sh \
--stage $stage --nj $nj \
--train-set $train_set --gmm $gmm \
--num-threads-ubm $num_threads_ubm \
--nj-extractor $nj_extractor \
--num-processes-extractor $num_processes_extractor \
--num-threads-extractor $num_threads_extractor \
--nnet3-affix "$nnet3_affix"


Expand Down
9 changes: 8 additions & 1 deletion egs/wsj/s5/local/nnet3/run_ivector_common.sh
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,12 @@ gmm=tri4b # This specifies a GMM-dir from the features of the typ
# it should contain alignments for 'train_set'.

num_threads_ubm=32

nj_extractor=10
# It runs a JOB with '-pe smp N', where N=$[threads*processes]
num_processes_extractor=4
num_threads_extractor=4

nnet3_affix= # affix for exp/nnet3 directory to put iVector stuff in (e.g.
# in the tedlium recip it's _cleaned).

Expand Down Expand Up @@ -110,7 +116,8 @@ if [ $stage -le 4 ]; then
# can be sensitive to the amount of data. The script defaults to an iVector dimension of
# 100.
echo "$0: training the iVector extractor"
steps/online/nnet2/train_ivector_extractor.sh --cmd "$train_cmd" --nj 10 \
steps/online/nnet2/train_ivector_extractor.sh --cmd "$train_cmd" \
--nj $nj_extractor --num-threads $num_threads_extractor --num-processes $num_processes_extractor \
data/${train_set}_sp_hires exp/nnet3${nnet3_affix}/diag_ubm exp/nnet3${nnet3_affix}/extractor || exit 1;
fi

Expand Down