espnet2.gan_tts.hifigan.hifigan.HiFiGANMultiPeriodDiscriminator

source

class espnet2.gan_tts.hifigan.hifigan.HiFiGANMultiPeriodDiscriminator(periods: List[int] = [2, 3, 5, 7, 11], discriminator_params: Dict[str, Any] = {'bias': True, 'channels': 32, 'downsample_scales': [3, 3, 3, 3, 1], 'in_channels': 1, 'kernel_sizes': [5, 3], 'max_downsample_channels': 1024, 'nonlinear_activation': 'LeakyReLU', 'nonlinear_activation_params': {'negative_slope': 0.1}, 'out_channels': 1, 'use_spectral_norm': False, 'use_weight_norm': True})

Bases: Module

HiFiGAN multi-period discriminator module.

Initialize HiFiGANMultiPeriodDiscriminator module.

• Parameters:
  • periods (List *[*int ]) – List of periods.
  • discriminator_params (Dict *[*str , Any ]) – Parameters for hifi-gan period discriminator module. The period parameter will be overwritten.

forward(x: Tensor) → Tensor

Calculate forward propagation.

• Parameters:x (Tensor) – Input noise signal (B, 1, T).
• Returns: List of list of each discriminator outputs, which consists of each : layer output tensors.
• Return type: List