Skip to content

Commit

Permalink
remove pretrained_ctc.py
Browse files Browse the repository at this point in the history
  • Loading branch information
AmirHussein96 committed Apr 5, 2024
1 parent bd503f9 commit c7f74e4
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 9 deletions.
20 changes: 12 additions & 8 deletions egs/librispeech/ASR/zipformer/export.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,13 +29,17 @@
(1) Export to torchscript model using torch.jit.script()
- For non-streaming model:
./zipformer/export.py \
--exp-dir ./zipformer/exp \
--tokens data/lang_bpe_500/tokens.txt \
--epoch 30 \
--avg 9 \
- For non-streaming model:
./zipformer_hat_seame/export.py \
--exp-dir ./zipformer_hat/exp \
--tokens data_seame/lang_bpe_4000/tokens.txt \
--epoch 20 \
--avg 5 \
--num-encoder-layers 2,2,2,2,2,2 \
--feedforward-dim 512,768,1024,1024,1024,768 \
--encoder-dim 192,256,256,256,256,256 \
--encoder-unmasked-dim 192,192,192,192,192,192 \
--jit 1
It will generate a file `jit_script.pt` in the given `exp_dir`. You can later
Expand Down Expand Up @@ -234,7 +238,7 @@ def get_parser():
parser.add_argument(
"--tokens",
type=str,
default="data/lang_bpe_500/tokens.txt",
default="data_libri/lang_bpe_500/tokens.txt",
help="Path to the tokens.txt",
)

Expand Down
1 change: 0 additions & 1 deletion egs/seame/ASR/zipformer_hat/pretrained_ctc.py

This file was deleted.

0 comments on commit c7f74e4

Please sign in to comment.