From 3a191503446e1bb06158adee2f38f971d2a0342d Mon Sep 17 00:00:00 2001 From: lfchener Date: Sat, 12 Dec 2020 10:34:48 +0000 Subject: [PATCH] move model.eval() to manually --- parakeet/models/tacotron2.py | 1 - 1 file changed, 1 deletion(-) diff --git a/parakeet/models/tacotron2.py b/parakeet/models/tacotron2.py index 334eaf2..599e6a2 100644 --- 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(