diff --git a/layers/tacotron.py b/layers/tacotron.py index 034f682b..1f0d0be6 100644 --- a/layers/tacotron.py +++ b/layers/tacotron.py @@ -67,6 +67,7 @@ class BatchNormConv1d(nn.Module): class Highway(nn.Module): + # TODO: Try GLU layer def __init__(self, in_size, out_size): super(Highway, self).__init__() self.H = nn.Linear(in_size, out_size)