diff --git a/dnn/torch/neural-pitch/models.py b/dnn/torch/neural-pitch/models.py index 2d24f9a514947cad2a1e39b252b45dc18fc03397..34b418e8a3bb6581be9b135b7e313fd5fdbd0df4 100644 --- a/dnn/torch/neural-pitch/models.py +++ b/dnn/torch/neural-pitch/models.py @@ -102,8 +102,7 @@ class PitchDNN(torch.nn.Module): ) self.GRU = torch.nn.GRU(input_size=gru_dim, hidden_size=gru_dim, num_layers=1, batch_first=True) self.upsample = torch.nn.Sequential( - torch.nn.Linear(gru_dim, output_dim), - self.activation + torch.nn.Linear(gru_dim, output_dim) ) def forward(self, x):