maxmustermann/DeepLearningExamples
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

fixed device in tacotron2 inference

Browse source
This commit is contained in:
gkarch 2020-04-21 22:07:11 +02:00
parent 362dfe6b3b
commit 46758dadf8
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
PyTorch/SpeechSynthesis/Tacotron2/tacotron2/model.py
View file

@ -530,8 +530,8 @@ class Decoder(nn.Module):
self.initialize_decoder_states(memory, mask=None)
mel_lengths = torch.zeros([memory.size(0)], dtype=torch.int32).cuda()
not_finished = torch.ones([memory.size(0)], dtype=torch.int32).cuda()
mel_lengths = torch.zeros([memory.size(0)], dtype=torch.int32).to(memory.device)
not_finished = torch.ones([memory.size(0)], dtype=torch.int32).to(memory.device)
mel_outputs, gate_outputs, alignments = [], [], []
while True:
decoder_input = self.prenet(decoder_input)