Skip to content
Open
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
60 changes: 44 additions & 16 deletions egs/librispeech/ASR/zipformer_adapter/decode.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,50 +20,58 @@
Usage:

- To activate the adapter (test on the target domain)
set --use-adapter True
set --use-adapters 1

- To deactivate the adapter (test on the original domain)
set --use-adapter False
set --use-adapters 0

(1) greedy search
./zipformer/decode.py \
./zipformer_adapter/decode.py \
--epoch 28 \
--avg 15 \
--exp-dir ./zipformer/exp \
--max-duration 600 \
--decoding-method greedy_search
--decoding-method greedy_search \
--use-adapters 1 \
--adapter-dim 16

(2) beam search (not recommended)
./zipformer/decode.py \
./zipformer_adapter/decode.py \
--epoch 28 \
--avg 15 \
--exp-dir ./zipformer/exp \
--max-duration 600 \
--decoding-method beam_search \
--beam-size 4
--beam-size 4 \
--use-adapters 1 \
--adapter-dim 16

(3) modified beam search
./zipformer/decode.py \
./zipformer_adapter/decode.py \
--epoch 28 \
--avg 15 \
--exp-dir ./zipformer/exp \
--max-duration 600 \
--decoding-method modified_beam_search \
--beam-size 4
--beam-size 4 \
--use-adapters 1 \
--adapter-dim 16

(4) fast beam search (one best)
./zipformer/decode.py \
./zipformer_adapter/decode.py \
--epoch 28 \
--avg 15 \
--exp-dir ./zipformer/exp \
--max-duration 600 \
--decoding-method fast_beam_search \
--beam 20.0 \
--max-contexts 8 \
--max-states 64
--max-states 64 \
--use-adapters 1 \
--adapter-dim 16

(5) fast beam search (nbest)
./zipformer/decode.py \
./zipformer_adapter/decode.py \
--epoch 28 \
--avg 15 \
--exp-dir ./zipformer/exp \
Expand All @@ -73,10 +81,12 @@
--max-contexts 8 \
--max-states 64 \
--num-paths 200 \
--nbest-scale 0.5
--nbest-scale 0.5 \
--use-adapters 1 \
--adapter-dim 16

(6) fast beam search (nbest oracle WER)
./zipformer/decode.py \
./zipformer_adapter/decode.py \
--epoch 28 \
--avg 15 \
--exp-dir ./zipformer/exp \
Expand All @@ -86,18 +96,22 @@
--max-contexts 8 \
--max-states 64 \
--num-paths 200 \
--nbest-scale 0.5
--nbest-scale 0.5 \
--use-adapters 1 \
--adapter-dim 16

(7) fast beam search (with LG)
./zipformer/decode.py \
./zipformer_adapter/decode.py \
--epoch 28 \
--avg 15 \
--exp-dir ./zipformer/exp \
--max-duration 600 \
--decoding-method fast_beam_search_nbest_LG \
--beam 20.0 \
--max-contexts 8 \
--max-states 64
--max-states 64 \
--use-adapters 1 \
--adapter-dim 16
"""


Expand Down Expand Up @@ -377,6 +391,20 @@ def get_parser():
modified_beam_search_LODR.
""",
)

parser.add_argument(
"--use-adapters",
type=int,
Comment thread
Elquintas marked this conversation as resolved.
default=0,
help="Whether to use adapters",
)
Comment thread
Elquintas marked this conversation as resolved.

parser.add_argument(
"--adapter-dim",
type=int,
default=0,
help="Adapter dimension",
)
Comment thread
Elquintas marked this conversation as resolved.
Comment thread
Elquintas marked this conversation as resolved.
add_model_arguments(parser)

return parser
Expand Down
Loading