move model.eval() to manually

2020-12-12 10:34:48 +00:00 · 2020-12-12 10:34:48 +00:00 · 3a19150344
parent 814d047129
commit 3a19150344
1 changed files with 0 additions and 1 deletions
--- a/parakeet/models/tacotron2.py
+++ b/parakeet/models/tacotron2.py
@ -389,7 +389,6 @@ class Tacotron2(nn.Layer):

    @paddle.no_grad()
    def infer(self, text_inputs, stop_threshold=0.5, max_decoder_steps=1000):
-        self.eval()
        embedded_inputs = self.embedding(text_inputs)
        encoder_outputs = self.encoder(embedded_inputs)
        mel_outputs, stop_logits, alignments = self.decoder.infer(