Update model.py

4 years ago · e4af876d31
parent 33c79804a4
commit e4af876d31
1 changed files with 1 additions and 1 deletions
--- a/deepspeech/exps/deepspeech2/model.py
+++ b/deepspeech/exps/deepspeech2/model.py
@ -332,7 +332,7 @@ class DeepSpeech2Tester(DeepSpeech2Trainer):

    def export(self):
        infer_model = DeepSpeech2InferModel.from_pretrained(
-             self.test_loader, self.config, self.args.checkpoint_path)
+            self.test_loader, self.config, self.args.checkpoint_path)

        infer_model.eval()
        feat_dim = self.test_loader.collate_fn.feature_size