csukuangfj
/

icefall-asr-librispeech-pruned-transducer-stateless5-L-2022-05-23

Model card Files Files and versions Metrics Training metrics Community

icefall-asr-librispeech-pruned-transducer-stateless5-L-2022-05-23 / exp /decode-pruned-rnnt-L.sh

csukuangfj's picture

Add pretrained model.

a9b946e about 2 years ago

raw history blame contribute delete

No virus

1.22 kB

	#!/usr/bin/env bash

	export CUDA_VISIBLE_DEVICES="7"

	. path.sh
	set -ex

	# epoch=1
	# iter=2
	mode=1
	use_averaged_model=1

	epoch=39
	iter=116000
	avg_list="11 12 13 14 15"

	if [ $mode -eq 1 ]; then
	for avg in $avg_list; do
	./pruned_transducer_stateless5/decode.py \
	--epoch $epoch \
	--avg $avg \
	--exp-dir ./pruned_transducer_stateless5/exp-L \
	--max-duration 600 \
	--decoding-method fast_beam_search \
	--max-sym-per-frame 1 \
	--num-encoder-layers 18 \
	--dim-feedforward 2048 \
	--nhead 8 \
	--encoder-dim 512 \
	--decoder-dim 512 \
	--joiner-dim 512 \
	--use-averaged-model $use_averaged_model
	done
	elif [ $mode -eq 2 ]; then
	for avg in $avg_list; do
	./pruned_transducer_stateless5/decode.py \
	--iter $iter \
	--avg $avg \
	--exp-dir ./pruned_transducer_stateless5/exp-L \
	--max-duration 600 \
	--decoding-method greedy_search \
	--max-sym-per-frame 1 \
	--num-encoder-layers 18 \
	--dim-feedforward 2048 \
	--nhead 8 \
	--encoder-dim 512 \
	--decoder-dim 512 \
	--joiner-dim 512 \
	--use-averaged-model $use_averaged_model
	done
	else
	echo "Unsupported mode $mode"
	fi