class_docstr: WaveGlow model (https://arxiv.org/abs/1811.00002) that is used to generate audio from mel spectrogram.
class_docstr_tok: ["WaveGlow","model","(","https",":","\/\/arxiv.org\/abs\/1811.00002",")","that","is","used","to","generate","audio","from","mel","spectrogram","."]
code: def output_types(self):
      if (
          self.mode == OperationMode.training
          or self.mode == OperationMode.validation
      ):
          output_dict = {
              "pred_normal_dist": NeuralType(
                  ("B", "flowgroup", "T"),
                  NormalDistributionSamplesType(),
              ),
              "log_s_list": [
                  NeuralType(("B", "flowgroup", "T"), VoidType())
              ],  # TODO: Figure out a good typing
              "log_det_W_list": [
                  NeuralType(elements_type=LogDeterminantType())
              ],
          }
          if self.mode == OperationMode.validation:
              output_dict["audio_pred"] = NeuralType(
                  ("B", "T"), AudioSignal()
              )
              output_dict["spec"] = NeuralType(
                  ("B", "T", "D"), MelSpectrogramType()
              )
              output_dict["spec_len"] = NeuralType(("B"), LengthsType())
          return output_dict
      return {
          "audio_pred": NeuralType(("B", "T"), AudioSignal()),
      }
  
imports: import torch
  from hydra.utils import instantiate
  from omegaconf import DictConfig, open_dict
  from pytorch_lightning.loggers import TensorBoardLogger
  from nemo.collections.tts.losses.waveglowloss import WaveGlowLoss
  from nemo.collections.tts.models.base import GlowVocoder
  from nemo.collections.tts.parts.utils.helpers import OperationMode, waveglow_log_to_tb_func
  from nemo.core.classes import Exportable
  from nemo.core.classes.common import PretrainedModelInfo, typecheck
  from nemo.core.neural_types.elements import AudioSignal, LengthsType, LogDeterminantType, MelSpectrogramType, NormalDistributionSamplesType, VoidType
  from nemo.core.neural_types.neural_type import NeuralType
  from nemo.utils import logging, model_utils
label_desc: Use image node_id 17 for calling the WaveGlowModel obj's underlying member method code with example usage: obj.output_types() and returns: dict, output_dict