testadmin
/
Parakeet
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

move model.eval() to manually

This commit is contained in:
lfchener 2020-12-12 10:34:48 +00:00
parent 814d047129
commit 3a19150344
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
parakeet/models/tacotron2.py
View File

@ -389,7 +389,6 @@ class Tacotron2(nn.Layer):
@paddle.no_grad() @paddle.no_grad()
def infer(self, text_inputs, stop_threshold=0.5, max_decoder_steps=1000): def infer(self, text_inputs, stop_threshold=0.5, max_decoder_steps=1000):
self.eval()
embedded_inputs = self.embedding(text_inputs) embedded_inputs = self.embedding(text_inputs)
encoder_outputs = self.encoder(embedded_inputs) encoder_outputs = self.encoder(embedded_inputs)
mel_outputs, stop_logits, alignments = self.decoder.infer( mel_outputs, stop_logits, alignments = self.decoder.infer(