code: def enhancer_config():
      n_bands = 80
      latent_dim = 192
      style_depth = 4
      network_capacity = 16
      fmap_max = 192
  
      config = {
          "model": {
              "n_bands": n_bands,
              "latent_dim": latent_dim,
              "style_depth": style_depth,
              "network_capacity": network_capacity,
              "mixed_prob": 0.9,
              "fmap_max": fmap_max,
              "generator": {
                  "_target_": "nemo.collections.tts.modules.spectrogram_enhancer.Generator",
                  "n_bands": n_bands,
                  "latent_dim": latent_dim,
                  "network_capacity": network_capacity,
                  "style_depth": style_depth,
                  "fmap_max": fmap_max,
              },
              "discriminator": {
                  "_target_": "nemo.collections.tts.modules.spectrogram_enhancer.Discriminator",
                  "n_bands": n_bands,
                  "network_capacity": network_capacity,
                  "fmap_max": fmap_max,
              },
              "spectrogram_min_value": -13.18,
              "spectrogram_max_value": 4.78,
              "consistency_loss_weight": 10.0,
              "gradient_penalty_loss_weight": 10.0,
              "gradient_penalty_loss_every_n_steps": 4,
              "spectrogram_predictor_path": None,
          },
          "generator_opt": {
              "_target_": "torch.optim.Adam",
              "lr": 2e-4,
              "betas": [0.5, 0.9],
          },
          "discriminator_opt": {
              "_target_": "torch.optim.Adam",
              "lr": 2e-4,
              "betas": [0.5, 0.9],
          },
      }
  
      return DictConfig(config)
  
file_path: NeMo/tests/collections/tts/test_spectrogram_enhancer.py
imports: import pytest
  import torch
  from einops import rearrange
  from omegaconf import DictConfig
  from nemo.collections.tts.models import SpectrogramEnhancerModel
  from nemo.collections.tts.parts.utils.helpers import mask_sequence_tensor
label_desc: Use image node_id 1 for calling a global function with example usage: enhancer_config() and returns: DictConfig