fix bugs, add coverage, add scripts

.travis/unittest.sh

@@ -1,5 +1,14 @@
 #!/bin/bash
 
+
+
+abort(){
+    echo "Run unittest failed" 1>&2
+    echo "Please check your code" 1>&2
+    exit 1
+}
+
+
 unittest(){
     cd $1 > /dev/null
     find . -path ./tools/venv -prune -false -o -name 'tests' -type d -print0 | \
@@ -8,17 +17,22 @@ unittest(){
     cd - > /dev/null
 }
 
-abort(){
-    echo "Run unittest failed" 1>&2
-    echo "Please check your code" 1>&2
-    exit 1
+coverage(){
+    cd $1 > /dev/null
+    find . -path ./tools/venv -prune -false -o -name 'tests' -type d -print0 | \
+        xargs -0 -I{} -n1 bash -c \
+        'python3 -m coverage run --branch {}'
+    python3 -m coverage report -m
+    python3 -m coverage html
+    cd - > /dev/null
 }
 
 trap 'abort' 0
 set -e
 
 source tools/venv/bin/activate
-pip3 install pytest
-unittest .
+#pip3 install pytest
+#unittest .
+coverage .
 
 trap : 0

deepspeech/exps/u2/model.py

@@ -74,7 +74,7 @@ class U2Trainer(Trainer):
     def __init__(self, config, args):
         super().__init__(config, args)
 
-    def train_batch(self, batch_data):
+    def train_batch(self, batch_data, msg):
         train_conf = self.config.training
         self.model.train()
 
@@ -93,12 +93,9 @@ class U2Trainer(Trainer):
             'train_att_loss': float(attention_loss),
             'train_ctc_loss': float(ctc_loss),
         }
-        msg = "Train: Rank: {}, ".format(dist.get_rank())
-        msg += "epoch: {}, ".format(self.epoch)
-        msg += "step: {}, ".format(self.iteration)
         msg += "time: {:>.3f}s, ".format(iteration_time)
-        msg += f"batch size: {self.config.data.batch_size}, "
-        msg += f"accum: {train_config.accum_grad}, "
+        msg += "batch size: {}, ".format(self.config.data.batch_size)
+        msg += "accum: {}, ".format(train_conf.accum_grad)
         msg += ', '.join('{}: {:>.6f}'.format(k, v)
                          for k, v in losses_np.items())
         if self.iteration % train_conf.log_interval == 0:

deepspeech/models/deepspeech2.py

@@ -16,6 +16,7 @@ import logging
 from typing import Optional
 from yacs.config import CfgNode
 
+import paddle
 from paddle import nn
 
 from deepspeech.modules.conv import ConvStack

deepspeech/models/u2.py

@@ -761,8 +761,10 @@ class U2Model(U2BaseModel):
         Returns:
             DeepSpeech2Model: The model built from pretrained result.
         """
-        config.input_dim = self.dataset.feature_size
-        config.output_dim = self.dataset.vocab_size
+        config.defrost()
+        config.input_dim = dataset.feature_size
+        config.output_dim = dataset.vocab_size
+        config.freeze()
         model = cls.from_config(config)
 
         if checkpoint_path:

deepspeech/training/trainer.py

@@ -181,13 +181,12 @@ class Trainer():
                     msg += "epoch: {}, ".format(self.epoch)
                     msg += "step: {}, ".format(self.iteration)
                     msg += "dataloader time: {:>.3f}s, ".format(dataload_time)
-                    self.logger.info(msg)
                     self.iteration += 1
-                    self.train_batch(batch)
+                    self.train_batch(batch, msg)
                     data_start_time = time.time()
             except Exception as e:
                 self.logger.error(e)
-                pass
+                raise e
 
             self.valid()
             self.save()

examples/tiny/s1/local/export.sh

@@ -0,0 +1,20 @@
+#! /usr/bin/env bash
+
+if [ $# != 2 ];then
+    echo "usage: export ckpt_path jit_model_path"
+    exit -1
+fi
+
+python3 -u ${BIN_DIR}/export.py \
+--config conf/conformer.yaml \
+--checkpoint_path ${1} \
+--export_path ${2} 
+
+
+if [ $? -ne 0 ]; then
+    echo "Failed in evaluation!"
+    exit 1
+fi
+
+
+exit 0

examples/tiny/s1/local/test.sh

@@ -0,0 +1,22 @@
+#! /usr/bin/env bash
+
+# download language model
+bash local/download_lm_en.sh
+if [ $? -ne 0 ]; then
+    exit 1
+fi
+
+CUDA_VISIBLE_DEVICES=0 \
+python3 -u ${BIN_DIR}/test.py \
+--device 'gpu' \
+--nproc 1 \
+--config conf/conformer.yaml \
+--output ckpt
+
+if [ $? -ne 0 ]; then
+    echo "Failed in evaluation!"
+    exit 1
+fi
+
+
+exit 0

examples/tiny/s1/local/train.sh

@@ -1,7 +1,5 @@
 #! /usr/bin/env bash
 
-export FLAGS_sync_nccl_allreduce=0
-
 CUDA_VISIBLE_DEVICES=0 \
 python3 -u ${BIN_DIR}/train.py \
 --device 'gpu' \

requirements.txt

@@ -1,3 +1,4 @@
+coverage
 pre-commit
 python_speech_features
 resampy==0.2.2