class_docstr: STFT loss module.
class_docstr_tok: ["STFT","loss","module","."]
docstr: Initialize STFT loss module.
docstr_tok: ["Initialize","STFT","loss","module","."]
code: def __init__(
      self,
      fft_size=1024,
      shift_size=120,
      win_length=600,
      window="hann_window",
  ):
      """Initialize STFT loss module."""
      super(STFTLoss, self).__init__()
      self.fft_size = fft_size
      self.shift_size = shift_size
      self.win_length = win_length
      self.window = getattr(torch, window)(win_length)
      self.spectral_convergence_loss = SpectralConvergenceLoss()
      self.log_stft_magnitude_loss = LogSTFTMagnitudeLoss()
  
imports: import torch
  import torch.nn.functional
  from nemo.core.classes import Loss, typecheck
  from nemo.core.neural_types.elements import AudioSignal, LengthsType, LossType, SpectrogramType
  from nemo.core.neural_types.neural_type import NeuralType
label_desc: Use image node_id 1 to create a new STFTLoss object from inherited base classes: Loss with example: obj = STFTLoss(fft_size, shift_size, win_length, window)