From 64474ab5bf03814bfd98368756d560caa884deb7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BC=A0=E6=98=A5=E4=B9=94?= <83450930+Liyulingyue@users.noreply.github.com> Date: Sun, 10 Nov 2024 07:58:27 +0800 Subject: [PATCH] Update model.py --- paddlespeech/s2t/exps/deepspeech2/model.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/paddlespeech/s2t/exps/deepspeech2/model.py b/paddlespeech/s2t/exps/deepspeech2/model.py index d007a9e39..cd68f6be7 100644 --- a/paddlespeech/s2t/exps/deepspeech2/model.py +++ b/paddlespeech/s2t/exps/deepspeech2/model.py @@ -335,7 +335,10 @@ class DeepSpeech2Tester(DeepSpeech2Trainer): self.test_loader, self.config, self.args.checkpoint_path) infer_model.eval() static_model = infer_model.export() - logger.info(f"Export code: {static_model.forward.code}") + try: + logger.info(f"Export code: {static_model.forward.code}") + except: + logger.info(f"Fail to print Export code, static_model.forward.code can not be run.") paddle.jit.save(static_model, self.args.export_path)