Update model.py

3 years ago · 8716386464
parent 2c8d28111a
commit 8716386464
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