Update model.py

10 months ago · 64474ab5bf
parent 0dbe89476b
commit 64474ab5bf
1 changed files with 4 additions and 1 deletions
--- 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)