未验证 提交 87163864 编写于 作者: J Jackwaterveg 提交者: GitHub

Update model.py

上级 2c8d2811
...@@ -332,7 +332,7 @@ class DeepSpeech2Tester(DeepSpeech2Trainer): ...@@ -332,7 +332,7 @@ class DeepSpeech2Tester(DeepSpeech2Trainer):
def export(self): def export(self):
infer_model = DeepSpeech2InferModel.from_pretrained( 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() infer_model.eval()
feat_dim = self.test_loader.collate_fn.feature_size feat_dim = self.test_loader.collate_fn.feature_size
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册