remove abundant arguments

pull/10/head
Eren Golge 2018-05-25 03:25:01 -07:00
parent a5f66b58e0
commit 4127b66359
1 changed files with 1 additions and 3 deletions

View File

@ -198,13 +198,11 @@ class Decoder(nn.Module):
eps (float): threshold for detecting the end of a sentence. eps (float): threshold for detecting the end of a sentence.
""" """
def __init__(self, in_features, memory_dim, r, eps=0, mode='train'): def __init__(self, in_features, memory_dim, r):
super(Decoder, self).__init__() super(Decoder, self).__init__()
self.mode = mode self.mode = mode
self.max_decoder_steps = 200 self.max_decoder_steps = 200
self.memory_dim = memory_dim self.memory_dim = memory_dim
self.eps = eps
self.r = r
# memory -> |Prenet| -> processed_memory # memory -> |Prenet| -> processed_memory
self.prenet = Prenet(memory_dim * r, out_features=[256, 128]) self.prenet = Prenet(memory_dim * r, out_features=[256, 128])
# processed_inputs, processed_memory -> |Attention| -> Attention, Alignment, RNN_State # processed_inputs, processed_memory -> |Attention| -> Attention, Alignment, RNN_State