Replaced unfold() with split_view() (#2671)

* Replaced unfold() with split_view() Signed-off-by: Boris Fomitchev <[email protected]> * fixed typo Signed-off-by: Boris Fomitchev <[email protected]> Co-authored-by: Somshubra Majumdar <[email protected]>
NVIDIA · Aug 30, 2021 · 407d955 · 407d955
1 parent 0dbd919
commit 407d955
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/nemo/collections/tts/modules/waveglow.py b/nemo/collections/tts/modules/waveglow.py
@@ -176,7 +176,6 @@ def audio_to_normal_dist(self, *, spec: torch.Tensor, audio: torch.Tensor) -> (t
         if spec.size(2) > audio.size(1):
             spec = spec[:, :, : audio.size(1)]
 
-        # logging.debug(f"spec: {spec.shape}. n_group: {self.n_group}")
         spec = split_view(spec, self.n_group, 2).permute(0, 2, 1, 3)
         spec = spec.contiguous().view(spec.size(0), spec.size(1), -1)
         spec = spec.permute(0, 2, 1)