diff --git a/egs/librispeech/ASR/zipformer_adapter/decode.py b/egs/librispeech/ASR/zipformer_adapter/decode.py index e8798aed69..4427822a82 100755 --- a/egs/librispeech/ASR/zipformer_adapter/decode.py +++ b/egs/librispeech/ASR/zipformer_adapter/decode.py @@ -20,39 +20,45 @@ 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 \ @@ -60,10 +66,12 @@ --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 \ @@ -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 \ @@ -86,10 +96,12 @@ --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 \ @@ -97,7 +109,9 @@ --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 """ @@ -377,6 +391,20 @@ def get_parser(): modified_beam_search_LODR. """, ) + + parser.add_argument( + "--use-adapters", + type=int, + default=1, + help="Whether to use adapters", + ) + + parser.add_argument( + "--adapter-dim", + type=int, + default=16, + help="Adapter dimension", + ) add_model_arguments(parser) return parser