Merge pull request #53 from lfchener/reborn

move model.eval() to manually
2020-12-12 18:36:13 +08:00 · 2020-12-12 18:36:13 +08:00 · ab56eac676
parent 814d047129 3a19150344
commit ab56eac676
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(