mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-09 01:52:41 +00:00
Fix typo (#1114)
* Fix typo for zipformer * Fix typo for pruned_transducer_stateless7 * Fix typo for pruned_transducer_stateless7_ctc * Fix typo for pruned_transducer_stateless7_ctc_bs * Fix typo for pruned_transducer_stateless7_streaming * Fix typo for pruned_transducer_stateless7_streaming_multi * Fix file permissions for pruned_transducer_stateless7_streaming_multi * Fix typo for pruned_transducer_stateless8 * Fix typo for pruned_transducer_stateless6 * Fix typo for pruned_transducer_stateless5 * Fix typo for pruned_transducer_stateless4 * Fix typo for pruned_transducer_stateless3
This commit is contained in:
parent
82f34a2388
commit
ca60ced213
@ -99,7 +99,7 @@ def get_parser():
|
|||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--exp-dir",
|
"--exp-dir",
|
||||||
type=str,
|
type=str,
|
||||||
default="pruned_transducer_stateless2/exp",
|
default="pruned_transducer_stateless3/exp",
|
||||||
help="The experiment dir",
|
help="The experiment dir",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -78,7 +78,7 @@ def get_parser():
|
|||||||
type=int,
|
type=int,
|
||||||
default=28,
|
default=28,
|
||||||
help="""It specifies the checkpoint to use for decoding.
|
help="""It specifies the checkpoint to use for decoding.
|
||||||
Note: Epoch counts from 0.
|
Note: Epoch counts from 1.
|
||||||
You can specify --avg to use more checkpoints for model averaging.""",
|
You can specify --avg to use more checkpoints for model averaging.""",
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -115,7 +115,7 @@ def get_parser():
|
|||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--exp-dir",
|
"--exp-dir",
|
||||||
type=str,
|
type=str,
|
||||||
default="pruned_transducer_stateless2/exp",
|
default="pruned_transducer_stateless4/exp",
|
||||||
help="The experiment dir",
|
help="The experiment dir",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -26,7 +26,7 @@ export CUDA_VISIBLE_DEVICES="0,1,2,3"
|
|||||||
--world-size 4 \
|
--world-size 4 \
|
||||||
--num-epochs 30 \
|
--num-epochs 30 \
|
||||||
--start-epoch 1 \
|
--start-epoch 1 \
|
||||||
--exp-dir pruned_transducer_stateless2/exp \
|
--exp-dir pruned_transducer_stateless4/exp \
|
||||||
--full-libri 1 \
|
--full-libri 1 \
|
||||||
--max-duration 300
|
--max-duration 300
|
||||||
|
|
||||||
@ -37,7 +37,7 @@ export CUDA_VISIBLE_DEVICES="0,1,2,3"
|
|||||||
--num-epochs 30 \
|
--num-epochs 30 \
|
||||||
--start-epoch 1 \
|
--start-epoch 1 \
|
||||||
--use-fp16 1 \
|
--use-fp16 1 \
|
||||||
--exp-dir pruned_transducer_stateless2/exp \
|
--exp-dir pruned_transducer_stateless4/exp \
|
||||||
--full-libri 1 \
|
--full-libri 1 \
|
||||||
--max-duration 550
|
--max-duration 550
|
||||||
|
|
||||||
@ -195,7 +195,7 @@ def get_parser():
|
|||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--exp-dir",
|
"--exp-dir",
|
||||||
type=str,
|
type=str,
|
||||||
default="pruned_transducer_stateless2/exp",
|
default="pruned_transducer_stateless4/exp",
|
||||||
help="""The experiment dir.
|
help="""The experiment dir.
|
||||||
It specifies the directory where all training related
|
It specifies the directory where all training related
|
||||||
files, e.g., checkpoints, log, etc, are saved
|
files, e.g., checkpoints, log, etc, are saved
|
||||||
@ -296,7 +296,7 @@ def get_parser():
|
|||||||
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
||||||
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
||||||
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
||||||
end of each epoch where `xxx` is the epoch number counting from 0.
|
end of each epoch where `xxx` is the epoch number counting from 1.
|
||||||
""",
|
""",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -87,7 +87,7 @@ def get_parser():
|
|||||||
type=int,
|
type=int,
|
||||||
default=28,
|
default=28,
|
||||||
help="""It specifies the checkpoint to use for averaging.
|
help="""It specifies the checkpoint to use for averaging.
|
||||||
Note: Epoch counts from 0.
|
Note: Epoch counts from 1.
|
||||||
You can specify --avg to use more checkpoints for model averaging.""",
|
You can specify --avg to use more checkpoints for model averaging.""",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -84,7 +84,7 @@ def get_parser():
|
|||||||
type=int,
|
type=int,
|
||||||
default=28,
|
default=28,
|
||||||
help="""It specifies the checkpoint to use for averaging.
|
help="""It specifies the checkpoint to use for averaging.
|
||||||
Note: Epoch counts from 0.
|
Note: Epoch counts from 1.
|
||||||
You can specify --avg to use more checkpoints for model averaging.""",
|
You can specify --avg to use more checkpoints for model averaging.""",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -78,7 +78,7 @@ def get_parser():
|
|||||||
type=int,
|
type=int,
|
||||||
default=28,
|
default=28,
|
||||||
help="""It specifies the checkpoint to use for decoding.
|
help="""It specifies the checkpoint to use for decoding.
|
||||||
Note: Epoch counts from 0.
|
Note: Epoch counts from 1.
|
||||||
You can specify --avg to use more checkpoints for model averaging.""",
|
You can specify --avg to use more checkpoints for model averaging.""",
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -115,7 +115,7 @@ def get_parser():
|
|||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--exp-dir",
|
"--exp-dir",
|
||||||
type=str,
|
type=str,
|
||||||
default="pruned_transducer_stateless2/exp",
|
default="pruned_transducer_stateless5/exp",
|
||||||
help="The experiment dir",
|
help="The experiment dir",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -20,7 +20,7 @@
|
|||||||
To run this file, do:
|
To run this file, do:
|
||||||
|
|
||||||
cd icefall/egs/librispeech/ASR
|
cd icefall/egs/librispeech/ASR
|
||||||
python ./pruned_transducer_stateless4/test_model.py
|
python ./pruned_transducer_stateless5/test_model.py
|
||||||
"""
|
"""
|
||||||
|
|
||||||
from train import get_params, get_transducer_model
|
from train import get_params, get_transducer_model
|
||||||
|
@ -328,7 +328,7 @@ def get_parser():
|
|||||||
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
||||||
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
||||||
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
||||||
end of each epoch where `xxx` is the epoch number counting from 0.
|
end of each epoch where `xxx` is the epoch number counting from 1.
|
||||||
""",
|
""",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -20,23 +20,23 @@
|
|||||||
# to a single one using model averaging.
|
# to a single one using model averaging.
|
||||||
"""
|
"""
|
||||||
Usage:
|
Usage:
|
||||||
./pruned_transducer_stateless2/export.py \
|
./pruned_transducer_stateless6/export.py \
|
||||||
--exp-dir ./pruned_transducer_stateless2/exp \
|
--exp-dir ./pruned_transducer_stateless6/exp \
|
||||||
--bpe-model data/lang_bpe_500/bpe.model \
|
--bpe-model data/lang_bpe_500/bpe.model \
|
||||||
--epoch 20 \
|
--epoch 20 \
|
||||||
--avg 10
|
--avg 10
|
||||||
|
|
||||||
It will generate a file exp_dir/pretrained.pt
|
It will generate a file exp_dir/pretrained.pt
|
||||||
|
|
||||||
To use the generated file with `pruned_transducer_stateless2/decode.py`,
|
To use the generated file with `pruned_transducer_stateless6/decode.py`,
|
||||||
you can do:
|
you can do:
|
||||||
|
|
||||||
cd /path/to/exp_dir
|
cd /path/to/exp_dir
|
||||||
ln -s pretrained.pt epoch-9999.pt
|
ln -s pretrained.pt epoch-9999.pt
|
||||||
|
|
||||||
cd /path/to/egs/librispeech/ASR
|
cd /path/to/egs/librispeech/ASR
|
||||||
./pruned_transducer_stateless2/decode.py \
|
./pruned_transducer_stateless6/decode.py \
|
||||||
--exp-dir ./pruned_transducer_stateless2/exp \
|
--exp-dir ./pruned_transducer_stateless6/exp \
|
||||||
--epoch 9999 \
|
--epoch 9999 \
|
||||||
--avg 1 \
|
--avg 1 \
|
||||||
--max-duration 100 \
|
--max-duration 100 \
|
||||||
@ -65,7 +65,7 @@ def get_parser():
|
|||||||
type=int,
|
type=int,
|
||||||
default=28,
|
default=28,
|
||||||
help="""It specifies the checkpoint to use for averaging.
|
help="""It specifies the checkpoint to use for averaging.
|
||||||
Note: Epoch counts from 0.
|
Note: Epoch counts from 1.
|
||||||
You can specify --avg to use more checkpoints for model averaging.""",
|
You can specify --avg to use more checkpoints for model averaging.""",
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -91,7 +91,7 @@ def get_parser():
|
|||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--exp-dir",
|
"--exp-dir",
|
||||||
type=str,
|
type=str,
|
||||||
default="pruned_transducer_stateless2/exp",
|
default="pruned_transducer_stateless6/exp",
|
||||||
help="""It specifies the directory where all training related
|
help="""It specifies the directory where all training related
|
||||||
files, e.g., checkpoints, log, etc, are saved
|
files, e.g., checkpoints, log, etc, are saved
|
||||||
""",
|
""",
|
||||||
|
@ -267,7 +267,7 @@ def get_parser():
|
|||||||
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
||||||
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
||||||
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
||||||
end of each epoch where `xxx` is the epoch number counting from 0.
|
end of each epoch where `xxx` is the epoch number counting from 1.
|
||||||
""",
|
""",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -94,10 +94,10 @@ Usage:
|
|||||||
--max-states 64
|
--max-states 64
|
||||||
|
|
||||||
(8) modified beam search with RNNLM shallow fusion
|
(8) modified beam search with RNNLM shallow fusion
|
||||||
./pruned_transducer_stateless5/decode.py \
|
./pruned_transducer_stateless7/decode.py \
|
||||||
--epoch 35 \
|
--epoch 35 \
|
||||||
--avg 15 \
|
--avg 15 \
|
||||||
--exp-dir ./pruned_transducer_stateless5/exp \
|
--exp-dir ./pruned_transducer_stateless7/exp \
|
||||||
--max-duration 600 \
|
--max-duration 600 \
|
||||||
--decoding-method modified_beam_search_lm_shallow_fusion \
|
--decoding-method modified_beam_search_lm_shallow_fusion \
|
||||||
--beam-size 4 \
|
--beam-size 4 \
|
||||||
@ -110,11 +110,11 @@ Usage:
|
|||||||
--rnn-lm-tie-weights 1
|
--rnn-lm-tie-weights 1
|
||||||
|
|
||||||
(9) modified beam search with LM shallow fusion + LODR
|
(9) modified beam search with LM shallow fusion + LODR
|
||||||
./pruned_transducer_stateless5/decode.py \
|
./pruned_transducer_stateless7/decode.py \
|
||||||
--epoch 28 \
|
--epoch 28 \
|
||||||
--avg 15 \
|
--avg 15 \
|
||||||
--max-duration 600 \
|
--max-duration 600 \
|
||||||
--exp-dir ./pruned_transducer_stateless5/exp \
|
--exp-dir ./pruned_transducer_stateless7/exp \
|
||||||
--decoding-method modified_beam_search_LODR \
|
--decoding-method modified_beam_search_LODR \
|
||||||
--beam-size 4 \
|
--beam-size 4 \
|
||||||
--lm-type rnn \
|
--lm-type rnn \
|
||||||
|
@ -79,7 +79,7 @@ def get_parser():
|
|||||||
type=int,
|
type=int,
|
||||||
default=28,
|
default=28,
|
||||||
help="""It specifies the checkpoint to use for averaging.
|
help="""It specifies the checkpoint to use for averaging.
|
||||||
Note: Epoch counts from 0.
|
Note: Epoch counts from 1.
|
||||||
You can specify --avg to use more checkpoints for model averaging.""",
|
You can specify --avg to use more checkpoints for model averaging.""",
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -116,7 +116,7 @@ def get_parser():
|
|||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--exp-dir",
|
"--exp-dir",
|
||||||
type=str,
|
type=str,
|
||||||
default="pruned_transducer_stateless5/exp",
|
default="pruned_transducer_stateless7/exp",
|
||||||
help="""It specifies the directory where all training related
|
help="""It specifies the directory where all training related
|
||||||
files, e.g., checkpoints, log, etc, are saved
|
files, e.g., checkpoints, log, etc, are saved
|
||||||
""",
|
""",
|
||||||
|
@ -389,7 +389,7 @@ def get_parser():
|
|||||||
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
||||||
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
||||||
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
||||||
end of each epoch where `xxx` is the epoch number counting from 0.
|
end of each epoch where `xxx` is the epoch number counting from 1.
|
||||||
""",
|
""",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -340,7 +340,7 @@ def get_parser():
|
|||||||
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
||||||
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
||||||
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
||||||
end of each epoch where `xxx` is the epoch number counting from 0.
|
end of each epoch where `xxx` is the epoch number counting from 1.
|
||||||
""",
|
""",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -346,7 +346,7 @@ def get_parser():
|
|||||||
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
||||||
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
||||||
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
||||||
end of each epoch where `xxx` is the epoch number counting from 0.
|
end of each epoch where `xxx` is the epoch number counting from 1.
|
||||||
""",
|
""",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -342,7 +342,7 @@ def get_parser():
|
|||||||
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
||||||
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
||||||
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
||||||
end of each epoch where `xxx` is the epoch number counting from 0.
|
end of each epoch where `xxx` is the epoch number counting from 1.
|
||||||
""",
|
""",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -90,7 +90,7 @@ def get_parser():
|
|||||||
type=int,
|
type=int,
|
||||||
default=28,
|
default=28,
|
||||||
help="""It specifies the checkpoint to use for averaging.
|
help="""It specifies the checkpoint to use for averaging.
|
||||||
Note: Epoch counts from 0.
|
Note: Epoch counts from 1.
|
||||||
You can specify --avg to use more checkpoints for model averaging.""",
|
You can specify --avg to use more checkpoints for model averaging.""",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -88,7 +88,7 @@ def get_parser():
|
|||||||
type=int,
|
type=int,
|
||||||
default=28,
|
default=28,
|
||||||
help="""It specifies the checkpoint to use for averaging.
|
help="""It specifies the checkpoint to use for averaging.
|
||||||
Note: Epoch counts from 0.
|
Note: Epoch counts from 1.
|
||||||
You can specify --avg to use more checkpoints for model averaging.""",
|
You can specify --avg to use more checkpoints for model averaging.""",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -39,7 +39,7 @@ def get_parser():
|
|||||||
type=int,
|
type=int,
|
||||||
default=28,
|
default=28,
|
||||||
help="""It specifies the checkpoint to use for averaging.
|
help="""It specifies the checkpoint to use for averaging.
|
||||||
Note: Epoch counts from 0.
|
Note: Epoch counts from 1.
|
||||||
You can specify --avg to use more checkpoints for model averaging.""",
|
You can specify --avg to use more checkpoints for model averaging.""",
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -65,7 +65,7 @@ def get_parser():
|
|||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--exp-dir",
|
"--exp-dir",
|
||||||
type=str,
|
type=str,
|
||||||
default="pruned_transducer_stateless2/exp",
|
default="pruned_transducer_stateless7_streaming/exp",
|
||||||
help="""It specifies the directory where all training related
|
help="""It specifies the directory where all training related
|
||||||
files, e.g., checkpoints, log, etc, are saved
|
files, e.g., checkpoints, log, etc, are saved
|
||||||
""",
|
""",
|
||||||
|
@ -77,7 +77,7 @@ def get_parser():
|
|||||||
type=int,
|
type=int,
|
||||||
default=28,
|
default=28,
|
||||||
help="""It specifies the checkpoint to use for decoding.
|
help="""It specifies the checkpoint to use for decoding.
|
||||||
Note: Epoch counts from 0.
|
Note: Epoch counts from 1.
|
||||||
You can specify --avg to use more checkpoints for model averaging.""",
|
You can specify --avg to use more checkpoints for model averaging.""",
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -114,7 +114,7 @@ def get_parser():
|
|||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--exp-dir",
|
"--exp-dir",
|
||||||
type=str,
|
type=str,
|
||||||
default="pruned_transducer_stateless2/exp",
|
default="pruned_transducer_stateless7_streaming/exp",
|
||||||
help="The experiment dir",
|
help="The experiment dir",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -355,7 +355,7 @@ def get_parser():
|
|||||||
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
||||||
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
||||||
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
||||||
end of each epoch where `xxx` is the epoch number counting from 0.
|
end of each epoch where `xxx` is the epoch number counting from 1.
|
||||||
""",
|
""",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -355,7 +355,7 @@ def get_parser():
|
|||||||
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
||||||
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
||||||
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
||||||
end of each epoch where `xxx` is the epoch number counting from 0.
|
end of each epoch where `xxx` is the epoch number counting from 1.
|
||||||
""",
|
""",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
0
egs/librispeech/ASR/pruned_transducer_stateless7_streaming_multi/decode_gigaspeech.py
Normal file → Executable file
0
egs/librispeech/ASR/pruned_transducer_stateless7_streaming_multi/decode_gigaspeech.py
Normal file → Executable file
@ -88,7 +88,7 @@ def get_parser():
|
|||||||
type=int,
|
type=int,
|
||||||
default=28,
|
default=28,
|
||||||
help="""It specifies the checkpoint to use for averaging.
|
help="""It specifies the checkpoint to use for averaging.
|
||||||
Note: Epoch counts from 0.
|
Note: Epoch counts from 1.
|
||||||
You can specify --avg to use more checkpoints for model averaging.""",
|
You can specify --avg to use more checkpoints for model averaging.""",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
4
egs/librispeech/ASR/pruned_transducer_stateless7_streaming_multi/streaming_decode.py
Normal file → Executable file
4
egs/librispeech/ASR/pruned_transducer_stateless7_streaming_multi/streaming_decode.py
Normal file → Executable file
@ -78,7 +78,7 @@ def get_parser():
|
|||||||
type=int,
|
type=int,
|
||||||
default=28,
|
default=28,
|
||||||
help="""It specifies the checkpoint to use for decoding.
|
help="""It specifies the checkpoint to use for decoding.
|
||||||
Note: Epoch counts from 0.
|
Note: Epoch counts from 1.
|
||||||
You can specify --avg to use more checkpoints for model averaging.""",
|
You can specify --avg to use more checkpoints for model averaging.""",
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -115,7 +115,7 @@ def get_parser():
|
|||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--exp-dir",
|
"--exp-dir",
|
||||||
type=str,
|
type=str,
|
||||||
default="pruned_transducer_stateless2/exp",
|
default="pruned_transducer_stateless7_streaming_multi/exp",
|
||||||
help="The experiment dir",
|
help="The experiment dir",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -366,7 +366,7 @@ def get_parser():
|
|||||||
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
||||||
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
||||||
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
||||||
end of each epoch where `xxx` is the epoch number counting from 0.
|
end of each epoch where `xxx` is the epoch number counting from 1.
|
||||||
""",
|
""",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -348,7 +348,7 @@ def get_parser():
|
|||||||
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
||||||
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
||||||
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
||||||
end of each epoch where `xxx` is the epoch number counting from 0.
|
end of each epoch where `xxx` is the epoch number counting from 1.
|
||||||
""",
|
""",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -81,7 +81,7 @@ def get_parser():
|
|||||||
type=int,
|
type=int,
|
||||||
default=28,
|
default=28,
|
||||||
help="""It specifies the checkpoint to use for decoding.
|
help="""It specifies the checkpoint to use for decoding.
|
||||||
Note: Epoch counts from 0.
|
Note: Epoch counts from 1.
|
||||||
You can specify --avg to use more checkpoints for model averaging.""",
|
You can specify --avg to use more checkpoints for model averaging.""",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -408,7 +408,7 @@ def get_parser():
|
|||||||
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
params.batch_idx_train % save_every_n == 0. The checkpoint filename
|
||||||
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
has the form: f'exp-dir/checkpoint-{params.batch_idx_train}.pt'
|
||||||
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
Note: It also saves checkpoint to `exp-dir/epoch-xxx.pt` at the
|
||||||
end of each epoch where `xxx` is the epoch number counting from 0.
|
end of each epoch where `xxx` is the epoch number counting from 1.
|
||||||
""",
|
""",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user