o ´‹ j± ã@sZddlZddlmZddlmZddlmZmZGdd„dejjƒZGdd„dejƒZ dS) éN)Únn)ÚConv1d)ÚDiscriminatorPÚMultiPeriodDiscriminatorcs*eZdZdZd‡fdd„ Zdd„Z‡ZS)ÚDiscriminatorSz¾HiFiGAN Scale Discriminator. Channel sizes are different from the original HiFiGAN. Args: use_spectral_norm (bool): if `True` swith to spectral norm instead of weight norm. FcsÈtƒ ¡|r tjjntjjj}t |tddddddƒ|tdddddd d ƒ|tdd dddd d ƒ|td d dddd d ƒ|td d ddd d d ƒ|td d d dddƒg¡|_ |td dddddƒ|_ dS)Néééé)Úpaddingé@é)éé)Úgroupsr éiééé) ÚsuperÚ__init__rÚutilsÚ spectral_normÚparametrizationsÚ weight_normÚ ModuleListrÚconvsÚ conv_post)ÚselfÚuse_spectral_normÚnorm_f©Ú __class__©úT/home/kuhnn/.local/lib/python3.10/site-packages/TTS/tts/layers/vits/discriminator.pyrs úÿ zDiscriminatorS.__init__cCs\g}|jD]}||ƒ}tjj |d¡}| |¡q| |¡}| |¡t |dd¡}||fS)zÄ Args: x (Tensor): input waveform. Returns: Tensor: discriminator scores. List[Tensor]: list of features from the convolutiona layers. gš™™™™™¹?réÿÿÿÿ)rÚtorchrÚ functionalÚ leaky_reluÚappendrÚflatten)rÚxÚfeatÚlr#r#r$Úforwards    zDiscriminatorS.forward)F©Ú__name__Ú __module__Ú __qualname__Ú__doc__rr.Ú __classcell__r#r#r!r$rsrcs,eZdZdZd ‡fdd„ Zd dd„Z‡ZS) ÚVitsDiscriminatorasVITS discriminator wrapping one Scale Discriminator and a stack of Period Discriminator. :: waveform -> ScaleDiscriminator() -> scores_sd, feats_sd --> append() -> scores, feats |--> MultiPeriodDiscriminator() -> scores_mpd, feats_mpd ^ Args: use_spectral_norm (bool): if `True` swith to spectral norm instead of weight norm. ©rrrr é FcsDtƒ ¡t ¡|_|j tˆd¡|j ‡fdd„|Dƒ¡dS)N©rcsg|]}t|ˆd‘qS)r8)r)Ú.0Úir8r#r$Ú Asz.VitsDiscriminator.__init__..)rrrrÚnetsr)rÚextend)rÚperiodsrr!r8r$r=s  zVitsDiscriminator.__init__Nc Csˆg}|durgnd}g}|durgnd}|jD]&}||ƒ\}} | |¡| | ¡|dur=||ƒ\} } | | ¡| | ¡q||||fS)a Args: x (Tensor): ground truth waveform. x_hat (Tensor): predicted waveform. Returns: List[Tensor]: discriminator scores. List[List[Tensor]]: list of list of features from each layers of each discriminator. N)r<r)) rr+Úx_hatÚx_scoresÚ x_hat_scoresÚx_featsÚ x_hat_featsÚnetÚx_scoreÚx_featÚ x_hat_scoreÚ x_hat_featr#r#r$r.Cs        € zVitsDiscriminator.forward)r6F)Nr/r#r#r!r$r52s r5) r&rÚtorch.nn.modules.convrÚ(TTS.vocoder.models.hifigan_discriminatorrrÚModulerr5r#r#r#r$Ús   *