From 592d981715e22cc8395471d5e837644e973c9166 Mon Sep 17 00:00:00 2001 From: AmirHussein96 Date: Mon, 15 Sep 2025 23:36:29 -0400 Subject: [PATCH] add HF --- egs/iwslt22_ta/ASR/RESULTS.md | 15 ++++++++------- egs/iwslt22_ta/ST/RESULTS.md | 5 ++--- 2 files changed, 10 insertions(+), 10 deletions(-) diff --git a/egs/iwslt22_ta/ASR/RESULTS.md b/egs/iwslt22_ta/ASR/RESULTS.md index 6b335f676..97e2e003a 100644 --- a/egs/iwslt22_ta/ASR/RESULTS.md +++ b/egs/iwslt22_ta/ASR/RESULTS.md @@ -1,6 +1,7 @@ # Results + ### IWSLT Tunisian training results (Stateless Pruned Transducer) #### 2023-06-01 @@ -8,7 +9,7 @@ | Decoding method | dev WER | test WER | comment | |------------------------------------|------------|------------|------------------------------------------| -| modified beam search | 47.6 | 51.2 | --epoch 20, --avg 10 | +| modified beam search | 47.6 | 51.2 | --epoch 20, --avg 13 | The training command for reproducing is given below: @@ -54,14 +55,14 @@ done #### 2023-06-01 -You can find a pretrained model, training logs, decoding logs, and decoding results at: -https://tensorboard.dev/experiment/yLE399ZPTzePG8B39jRyOw/ +You can find a pretrained model, training logs, decoding logs, and decoding results at: + | Decoding method | dev WER | test WER | comment | |------------------------------------|------------|------------|------------------------------------------| -| modified beam search | 40.8 | 44.4 | --epoch 20, --avg 10 | +| modified beam search | 40.8 | 44.1 | --epoch 20, --avg 13 | To reproduce the above result, use the following commands for training: @@ -70,7 +71,7 @@ To reproduce the above result, use the following commands for training: ``` export CUDA_VISIBLE_DEVICES="0,1" ./zipformer/train.py \ - --world-size 2 \ + --world-size 4 \ --num-epochs 20 \ --start-epoch 1 \ --use-fp16 1 \ @@ -82,7 +83,7 @@ export CUDA_VISIBLE_DEVICES="0,1" --encoder-unmasked-dim 192,192,256,256,256,192 \ --max-duration 800 \ --prune-range 10 - + ``` The decoding command is: @@ -99,7 +100,7 @@ for method in modified_beam_search; do --num-encoder-layers 2,2,2,2,2,2 \ --feedforward-dim 512,768,1024,1536,1024,768 \ --encoder-dim 192,256,384,512,384,256 \ - --encoder-unmasked-dim 192,192,256,256,256,192 + --encoder-unmasked-dim 192,192,256,256,256,192 \ --use-averaged-model true done ``` diff --git a/egs/iwslt22_ta/ST/RESULTS.md b/egs/iwslt22_ta/ST/RESULTS.md index bc59f1503..2a424dd05 100644 --- a/egs/iwslt22_ta/ST/RESULTS.md +++ b/egs/iwslt22_ta/ST/RESULTS.md @@ -57,7 +57,8 @@ done #### 2023-06-01 -You can find a pretrained model, training logs, decoding logs, and decoding results at: +You can find a pretrained model, training logs, decoding logs, and decoding results at: + @@ -95,8 +96,6 @@ To reproduce the above result, use the following commands for training: ``` -The tensorboard training log can be found at -https://tensorboard.dev/experiment/4sa4M1mRQyKjOE4o95mWUw/ The decoding command is: