diff options
| author | Jean-Marc Valin <jmvalin@amazon.com> | 2023-10-02 04:34:58 +0300 |
|---|---|---|
| committer | Jean-Marc Valin <jmvalin@amazon.com> | 2023-10-02 04:34:58 +0300 |
| commit | 8e8edf71bde743c21960a97815f14dd48c86d6ad (patch) | |
| tree | 58b91f1ee9573c618d1f61020a83095f57113880 | |
| parent | 33adba02c7ac5fe1d1f3bd4027f42b87cddc933c (diff) | |
Remove unneeded (I think) tanh at the end
| -rw-r--r-- | dnn/torch/neural-pitch/models.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/dnn/torch/neural-pitch/models.py b/dnn/torch/neural-pitch/models.py index 2d24f9a5..34b418e8 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): |