mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-26 18:24:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Minor fixes.

Browse Source
This commit is contained in:
Fangjun Kuang 2021-10-14 21:55:42 +08:00
parent 4ab52eeea4
commit f7ae6e9ebc
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/run-pretrained.yml vendored
View File

@ -85,6 +85,7 @@ jobs:
export PYTHONPATH=$PWD:PYTHONPATH export PYTHONPATH=$PWD:PYTHONPATH
cd egs/librispeech/ASR cd egs/librispeech/ASR
./conformer_ctc/pretrained.py \ ./conformer_ctc/pretrained.py \
--num-classes 500 \
--checkpoint ./tmp/icefall-asr-conformer-ctc-bpe-500/exp/pretrained.pt \ --checkpoint ./tmp/icefall-asr-conformer-ctc-bpe-500/exp/pretrained.pt \
--lang-dir ./tmp/icefall-asr-conformer-ctc-bpe-500/data/lang_bpe_500 \ --lang-dir ./tmp/icefall-asr-conformer-ctc-bpe-500/data/lang_bpe_500 \
--method ctc-decoding \ --method ctc-decoding \
@ -98,6 +99,7 @@ jobs:
export PYTHONPATH=$PWD:$PYTHONPATH export PYTHONPATH=$PWD:$PYTHONPATH
cd egs/librispeech/ASR cd egs/librispeech/ASR
./conformer_ctc/pretrained.py \ ./conformer_ctc/pretrained.py \
--num-classes 500 \
--checkpoint ./tmp/icefall-asr-conformer-ctc-bpe-500/exp/pretrained.pt \ --checkpoint ./tmp/icefall-asr-conformer-ctc-bpe-500/exp/pretrained.pt \
--lang-dir ./tmp/icefall-asr-conformer-ctc-bpe-500/data/lang_bpe_500 \ --lang-dir ./tmp/icefall-asr-conformer-ctc-bpe-500/data/lang_bpe_500 \
./tmp/icefall-asr-conformer-ctc-bpe-500/test_wavs/1089-134686-0001.flac \ ./tmp/icefall-asr-conformer-ctc-bpe-500/test_wavs/1089-134686-0001.flac \

10
egs/librispeech/ASR/conformer_ctc/pretrained.py
View File

@ -150,6 +150,15 @@ def get_parser():
""", """,
) )
parser.add_argument(
"--num-classes",
type=int,
default=5000,
help="""
Vocab size in the BPE model.
""",
)
parser.add_argument( parser.add_argument(
"--eos-id", "--eos-id",
type=int, type=int,
@ -183,7 +192,6 @@ def get_params() -> AttributeDict:
"use_feat_batchnorm": True, "use_feat_batchnorm": True,
"feature_dim": 80, "feature_dim": 80,
"nhead": 8, "nhead": 8,
"num_classes": 5000,
"attention_dim": 512, "attention_dim": 512,
"num_decoder_layers": 6, "num_decoder_layers": 6,
# parameters for decoding # parameters for decoding