Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

remove Variable from tacotron.py

This commit is contained in:
Eren Golge 2018-05-10 16:30:43 -07:00
parent c7b20d5133
commit 27f5c476da
1 changed files with 4 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
layers/tacotron.py
View File

@ -1,6 +1,5 @@
# coding: utf-8 # coding: utf-8
import torch import torch
from torch.autograd import Variable
from torch import nn from torch import nn
from .attention import AttentionRNN from .attention import AttentionRNN
@ -263,16 +262,12 @@ class Decoder(nn.Module):
self.memory_dim, self.r) self.memory_dim, self.r)
T_decoder = memory.size(1) T_decoder = memory.size(1)
# go frame - 0 frames tarting the sequence # go frame - 0 frames tarting the sequence
initial_memory = Variable( initial_memory = inputs.data.new(B, self.memory_dim * self.r).zero_()
inputs.data.new(B, self.memory_dim * self.r).zero_())
# Init decoder states # Init decoder states
attention_rnn_hidden = Variable( attention_rnn_hidden = inputs.data.new(B, 256).zero_()
inputs.data.new(B, 256).zero_()) decoder_rnn_hiddens = [inputs.data.new(B, 256).zero_()
decoder_rnn_hiddens = [Variable(
inputs.data.new(B, 256).zero_())
for _ in range(len(self.decoder_rnns))] for _ in range(len(self.decoder_rnns))]
current_context_vec = Variable( current_context_vec = inputs.data.new(B, 256).zero_()
inputs.data.new(B, 256).zero_())
# Time first (T_decoder, B, memory_dim) # Time first (T_decoder, B, memory_dim)
if memory is not None: if memory is not None:
memory = memory.transpose(0, 1) memory = memory.transpose(0, 1)