From 5a3a9e1f5055260f966d24680d5bb2e83f1d5b54 Mon Sep 17 00:00:00 2001
From: Hui Zhang <zhtclz@foxmail.com>
Date: Tue, 22 Jun 2021 02:58:58 +0000
Subject: [PATCH] fix chunk default config; tarball ckpt prfix dir;

---
 examples/aishell/s1/README.md                 | 10 ++++++++++
 examples/aishell/s1/conf/chunk_conformer.yaml |  2 +-
 utils/tarball.sh                              |  3 ++-
 3 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/examples/aishell/s1/README.md b/examples/aishell/s1/README.md
index 2048c4d5..c306f8aa 100644
--- a/examples/aishell/s1/README.md
+++ b/examples/aishell/s1/README.md
@@ -9,6 +9,16 @@
 | conformer | conf/conformer.yaml | spec_aug + shift | test | ctc_prefix_beam_search | - | 0.062196 |  
 | conformer | conf/conformer.yaml | spec_aug + shift | test | attention_rescoring | - | 0.054694 |  
 
+## Chunk Conformer
+
+| Model | Config | Augmentation| Test set | Decode method | Chunk | Loss | WER |  
+| --- | --- | --- | --- | --- | --- | --- | --- |   
+| conformer | conf/chunk_conformer.yaml | spec_aug + shift | test | attention | 16 | - | 0.061939 |  
+| conformer | conf/chunk_conformer.yaml | spec_aug + shift | test | ctc_greedy_search | 16 | - | 0.070806 |  
+| conformer | conf/chunk_conformer.yaml | spec_aug + shift | test | ctc_prefix_beam_search | 16 | - | 0.070739 |  
+| conformer | conf/chunk_conformer.yaml | spec_aug + shift | test | attention_rescoring | 16 |  - | 0.059400 |  
+
+
 ## Transformer
 
 | Model | Config | Augmentation| Test set | Decode method | Loss | WER |  
diff --git a/examples/aishell/s1/conf/chunk_conformer.yaml b/examples/aishell/s1/conf/chunk_conformer.yaml
index 904624c3..e626e106 100644
--- a/examples/aishell/s1/conf/chunk_conformer.yaml
+++ b/examples/aishell/s1/conf/chunk_conformer.yaml
@@ -78,7 +78,7 @@ model:
 
 
 training:
-  n_epoch: 180
+  n_epoch: 240
   accum_grad: 4
   global_grad_clip: 5.0
   optim: adam
diff --git a/utils/tarball.sh b/utils/tarball.sh
index 100b4719..224b740c 100755
--- a/utils/tarball.sh
+++ b/utils/tarball.sh
@@ -18,7 +18,8 @@ function clean() {
 }
 trap clean EXIT
 
-cp ${ckpt_prefix}.* ${output}
+# ckpt_prfix.{json,...} and ckpt_prfix dir
+cp -r ${ckpt_prefix}* ${output}
 cp ${model_config} ${mean_std} ${vocab} ${output}
 
 tar zcvf release.tar.gz ${output}