From 44ec59fc1fa71217d05f2689f783b98705b40a68 Mon Sep 17 00:00:00 2001
From: luomingshuang <739314837@qq.com>
Date: Thu, 28 Jul 2022 10:22:26 +0800
Subject: [PATCH] do some changes for export.py

---
 .../pruned_transducer_stateless5/export.py    | 50 +++++++++++++++----
 1 file changed, 39 insertions(+), 11 deletions(-)

diff --git a/egs/wenetspeech/ASR/pruned_transducer_stateless5/export.py b/egs/wenetspeech/ASR/pruned_transducer_stateless5/export.py
index 3630cebeb..d0a7fd69f 100644
--- a/egs/wenetspeech/ASR/pruned_transducer_stateless5/export.py
+++ b/egs/wenetspeech/ASR/pruned_transducer_stateless5/export.py
@@ -17,26 +17,54 @@
 # This script converts several saved checkpoints
 # to a single one using model averaging.
 """
-Usage:
-./pruned_transducer_stateless2/export.py \
-  --exp-dir ./pruned_transducer_stateless2/exp \
+Usage for offline:
+./pruned_transducer_stateless5/export.py \
+  --exp-dir ./pruned_transducer_stateless5/exp_L_offline \
   --lang-dir data/lang_char \
-  --epoch 10 \
-  --avg 2
+  --epoch 4 \
+  --avg 1
 
-It will generate a file exp_dir/pretrained.pt
+It will generate a file exp_dir/pretrained.pt for offline ASR.
 
-To use the generated file with `pruned_transducer_stateless2/decode.py`,
+./pruned_transducer_stateless5/export.py \
+  --exp-dir ./pruned_transducer_stateless5/exp_L_offline \
+  --lang-dir data/lang_char \
+  --epoch 4 \
+  --avg 1 \
+  --jit True
+
+It will generate a file exp_dir/cpu_jit.pt for offline ASR.
+
+Usage for streaming:
+./pruned_transducer_stateless5/export.py \
+  --exp-dir ./pruned_transducer_stateless5/exp_L_streaming \
+  --lang-dir data/lang_char \
+  --epoch 7 \
+  --avg 1
+
+It will generate a file exp_dir/pretrained.pt for streaming ASR.
+
+./pruned_transducer_stateless5/export.py \
+  --exp-dir ./pruned_transducer_stateless5/exp_L_streaming \
+  --lang-dir data/lang_char \
+  --epoch 7 \
+  --avg 1 \
+  --jit True
+
+It will generate a file exp_dir/cpu_jit.pt for streaming ASR.
+
+To use the generated file with `pruned_transducer_stateless5/decode.py`,
 you can do:
 
     cd /path/to/exp_dir
     ln -s pretrained.pt epoch-9999.pt
 
     cd /path/to/egs/wenetspeech/ASR
-    ./pruned_transducer_stateless2/decode.py \
-        --exp-dir ./pruned_transducer_stateless2/exp \
-        --epoch 10 \
-        --avg 2 \
+    ./pruned_transducer_stateless5/decode.py \
+        --exp-dir ./pruned_transducer_stateless5/exp \
+        --epoch 4 \
+        --avg 1 \
+        --decoding-method greedy_search \
         --max-duration 100 \
         --lang-dir data/lang_char
 """