Removed relu.

geneing · Jul 6, 2020 · 6e28829 · 6e28829
1 parent 0e4f6a3
commit 6e28829
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/models/forward_tacotron.py b/models/forward_tacotron.py
@@ -21,7 +21,8 @@ def forward(self, x, dur):
     @staticmethod
     def build_index(duration, x):
         duration[duration < 0] = 0
-        tot_duration = duration.cumsum(1).detach().cpu().numpy().astype('int')
+        tot_duration = duration.cumsum(1) + 0.5
+        tot_duration = tot_duration.detach().cpu().numpy().astype('int')
         max_duration = int(tot_duration.max().item())
         index = np.zeros([x.shape[0], max_duration, x.shape[2]], dtype='long')
 
@@ -61,7 +62,6 @@ def forward(self, x, alpha=1.0):
         x = x.transpose(1, 2)
         x, _ = self.rnn(x)
         x = self.lin(x)
-        x = torch.relu(x)
         return x / alpha