diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/decode.py b/egs/librispeech/ASR/pruned_transducer_stateless2/decode.py index 8e924bf96..38aff8834 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless2/decode.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless2/decode.py @@ -18,36 +18,36 @@ """ Usage: (1) greedy search -./pruned_transducer_stateless/decode.py \ +./pruned_transducer_stateless2/decode.py \ --epoch 28 \ --avg 15 \ - --exp-dir ./pruned_transducer_stateless/exp \ + --exp-dir ./pruned_transducer_stateless2/exp \ --max-duration 100 \ --decoding-method greedy_search (2) beam search -./pruned_transducer_stateless/decode.py \ +./pruned_transducer_stateless2/decode.py \ --epoch 28 \ --avg 15 \ - --exp-dir ./pruned_transducer_stateless/exp \ + --exp-dir ./pruned_transducer_stateless2/exp \ --max-duration 100 \ --decoding-method beam_search \ --beam-size 4 (3) modified beam search -./pruned_transducer_stateless/decode.py \ +./pruned_transducer_stateless2/decode.py \ --epoch 28 \ --avg 15 \ - --exp-dir ./pruned_transducer_stateless/exp \ + --exp-dir ./pruned_transducer_stateless2/exp \ --max-duration 100 \ --decoding-method modified_beam_search \ --beam-size 4 (4) fast beam search -./pruned_transducer_stateless/decode.py \ +./pruned_transducer_stateless2/decode.py \ --epoch 28 \ --avg 15 \ - --exp-dir ./pruned_transducer_stateless/exp \ + --exp-dir ./pruned_transducer_stateless2/exp \ --max-duration 1500 \ --decoding-method fast_beam_search \ --beam 4 \ @@ -124,7 +124,7 @@ def get_parser(): parser.add_argument( "--exp-dir", type=str, - default="pruned_transducer_stateless/exp", + default="pruned_transducer_stateless2/exp", help="The experiment dir", ) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/export.py b/egs/librispeech/ASR/pruned_transducer_stateless2/export.py index 7d2a07817..b5757ee8c 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless2/export.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless2/export.py @@ -20,23 +20,23 @@ # to a single one using model averaging. """ Usage: -./pruned_transducer_stateless/export.py \ - --exp-dir ./pruned_transducer_stateless/exp \ +./pruned_transducer_stateless2/export.py \ + --exp-dir ./pruned_transducer_stateless2/exp \ --bpe-model data/lang_bpe_500/bpe.model \ --epoch 20 \ --avg 10 It will generate a file exp_dir/pretrained.pt -To use the generated file with `pruned_transducer_stateless/decode.py`, +To use the generated file with `pruned_transducer_stateless2/decode.py`, you can do: cd /path/to/exp_dir ln -s pretrained.pt epoch-9999.pt cd /path/to/egs/librispeech/ASR - ./pruned_transducer_stateless/decode.py \ - --exp-dir ./pruned_transducer_stateless/exp \ + ./pruned_transducer_stateless2/decode.py \ + --exp-dir ./pruned_transducer_stateless2/exp \ --epoch 9999 \ --avg 1 \ --max-duration 100 \ @@ -80,7 +80,7 @@ def get_parser(): parser.add_argument( "--exp-dir", type=str, - default="pruned_transducer_stateless/exp", + default="pruned_transducer_stateless2/exp", help="""It specifies the directory where all training related files, e.g., checkpoints, log, etc, are saved """, diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py index 62dc825b6..c24fbe9a1 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless2/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless2/train.py @@ -116,7 +116,7 @@ def get_parser(): default=0, help="""Resume training from from this epoch. If it is positive, it will load checkpoint from - transducer_stateless/exp/epoch-{start_epoch-1}.pt + transducer_stateless2/exp/epoch-{start_epoch-1}.pt """, )