Skip to content
Snippets Groups Projects
Verified Commit 8e8edf71 authored by Jean-Marc Valin's avatar Jean-Marc Valin
Browse files

Remove unneeded (I think) tanh at the end

parent 33adba02
No related branches found
No related tags found
No related merge requests found
......@@ -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):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment