fix some small mistakes

pull/735/head
huangyuxin 4 years ago
parent 5dd9e2f8ec
commit 2c8d28111a

@ -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

@ -0,0 +1,4 @@
from .deepspeech2 import DeepSpeech2Model
from .deepspeech2 import DeepSpeech2InferModel
__all__ = ['DeepSpeech2Model', 'DeepSpeech2InferModel']

@ -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

Loading…
Cancel
Save