Remove unneeded (I think) tanh at the end

This commit is contained in:
Jean-Marc Valin 2023-10-01 21:34:58 -04:00
parent 33adba02c7
commit 8e8edf71bd
No known key found for this signature in database
GPG key ID: 531A52533318F00A

View file

@ -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):