fix some small mistakes

deepspeech/exps/deepspeech2/model.py

@@ -331,12 +331,8 @@ class DeepSpeech2Tester(DeepSpeech2Trainer):
             exit(-1)
 
     def export(self):
-        if self.config.model.apply_online == False:
-            infer_model = DeepSpeech2InferModel.from_pretrained(
-                self.test_loader, self.config, self.args.checkpoint_path)
-        else:
-            infer_model = DeepSpeech2InferModelOnline.from_pretrained(
-                self.test_loader, self.config, self.args.checkpoint_path)
+        infer_model = DeepSpeech2InferModel.from_pretrained(
+             self.test_loader, self.config, self.args.checkpoint_path)
 
         infer_model.eval()
         feat_dim = self.test_loader.collate_fn.feature_size

deepspeech/models/ds2/__init__.py

+from .deepspeech2 import DeepSpeech2Model
+from .deepspeech2 import DeepSpeech2InferModel
+
+__all__ = ['DeepSpeech2Model', 'DeepSpeech2InferModel']

examples/aishell/s0/run.sh

@@ -2,7 +2,7 @@
 set -e
 source path.sh
 
-gpus=2,3,4,5
+gpus=0,1,2,3
 stage=0
 stop_stage=100
 conf_path=conf/deepspeech2.yaml