From f7ae6e9ebc5982f0b9e07194112fa3d51a4dec11 Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Thu, 14 Oct 2021 21:55:42 +0800 Subject: [PATCH] Minor fixes. --- .github/workflows/run-pretrained.yml | 2 ++ egs/librispeech/ASR/conformer_ctc/pretrained.py | 10 +++++++++- 2 files changed, 11 insertions(+), 1 deletion(-) diff --git a/.github/workflows/run-pretrained.yml b/.github/workflows/run-pretrained.yml index eb6b65b1d..9584d1f27 100644 --- a/.github/workflows/run-pretrained.yml +++ b/.github/workflows/run-pretrained.yml @@ -85,6 +85,7 @@ jobs: export PYTHONPATH=$PWD:PYTHONPATH cd egs/librispeech/ASR ./conformer_ctc/pretrained.py \ + --num-classes 500 \ --checkpoint ./tmp/icefall-asr-conformer-ctc-bpe-500/exp/pretrained.pt \ --lang-dir ./tmp/icefall-asr-conformer-ctc-bpe-500/data/lang_bpe_500 \ --method ctc-decoding \ @@ -98,6 +99,7 @@ jobs: export PYTHONPATH=$PWD:$PYTHONPATH cd egs/librispeech/ASR ./conformer_ctc/pretrained.py \ + --num-classes 500 \ --checkpoint ./tmp/icefall-asr-conformer-ctc-bpe-500/exp/pretrained.pt \ --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 \ diff --git a/egs/librispeech/ASR/conformer_ctc/pretrained.py b/egs/librispeech/ASR/conformer_ctc/pretrained.py index c6de743ea..ef18d13d3 100755 --- a/egs/librispeech/ASR/conformer_ctc/pretrained.py +++ b/egs/librispeech/ASR/conformer_ctc/pretrained.py @@ -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( "--eos-id", type=int, @@ -183,7 +192,6 @@ def get_params() -> AttributeDict: "use_feat_batchnorm": True, "feature_dim": 80, "nhead": 8, - "num_classes": 5000, "attention_dim": 512, "num_decoder_layers": 6, # parameters for decoding