flowtron.py: removing unused mask_inverse

glociks · Aug 24, 2020 · f53066d · f53066d
1 parent 8427157
commit f53066d
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/flowtron.py b/flowtron.py
@@ -46,9 +46,8 @@ def forward(self, model_output, gate_target, lengths):
         z, log_s_list, gate_pred, mean, log_var, prob = model_output
 
         # create mask for outputs computed on padded data
-        mask = get_mask_from_lengths(lengths).transpose(0, 1)[..., None]
-        mask_inverse = ~mask
-        mask, mask_inverse = mask.float(), mask_inverse.float()
+        mask = get_mask_from_lengths(lengths).transpose(0, 1)[..., None]        
+        mask = mask.float()
         n_mel_dims = z.size(2)
         n_elements = mask.sum()
         for i, log_s in enumerate(log_s_list):