From 5f0f76f249714f8d845ecd89926d9c83c2309a33 Mon Sep 17 00:00:00 2001
From: TianYuan <white-sky@qq.com>
Date: Tue, 14 Dec 2021 14:07:47 +0800
Subject: [PATCH] add eval() for inference model (#1114)

---
 paddlespeech/cli/tts/infer.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/paddlespeech/cli/tts/infer.py b/paddlespeech/cli/tts/infer.py
index fdf93e2c..b3733e05 100644
--- a/paddlespeech/cli/tts/infer.py
+++ b/paddlespeech/cli/tts/infer.py
@@ -461,6 +461,7 @@ class TTSExecutor(BaseExecutor):
         am_std = paddle.to_tensor(am_std)
         am_normalizer = ZScore(am_mu, am_std)
         self.am_inference = am_inference_class(am_normalizer, am)
+        self.am_inference.eval()
         print("acoustic model done!")
 
         # vocoder
@@ -478,6 +479,7 @@ class TTSExecutor(BaseExecutor):
         voc_std = paddle.to_tensor(voc_std)
         voc_normalizer = ZScore(voc_mu, voc_std)
         self.voc_inference = voc_inference_class(voc_normalizer, voc)
+        self.voc_inference.eval()
         print("voc done!")
 
     def preprocess(self, input: Any, *args, **kwargs):
-- 
GitLab