From 4127b6635901679d48607d48dd3355c48c657a77 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Fri, 25 May 2018 03:25:01 -0700 Subject: [PATCH] remove abundant arguments --- layers/tacotron.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/layers/tacotron.py b/layers/tacotron.py index 77291a9d..f512cd02 100644 --- a/layers/tacotron.py +++ b/layers/tacotron.py @@ -198,13 +198,11 @@ class Decoder(nn.Module): 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__() self.mode = mode self.max_decoder_steps = 200 self.memory_dim = memory_dim - self.eps = eps - self.r = r # memory -> |Prenet| -> processed_memory self.prenet = Prenet(memory_dim * r, out_features=[256, 128]) # processed_inputs, processed_memory -> |Attention| -> Attention, Alignment, RNN_State