o j@s.ddlmZddlmZGdddejZdS))nnMelganDiscriminatorcs<eZdZ           d fd d Zd d ZZS)MelganMultiscaleDiscriminatorrr r r c sLttfddt|D|_tj|| | dd|_dS)Nc s"g|] }tdqS)) in_channels out_channels kernel_sizes base_channels max_channelsdownsample_factorsgroups_denominatorr).0_rrrrrrre/home/kuhnn/.local/lib/python3.10/site-packages/TTS/vocoder/models/melgan_multiscale_discriminator.py s z:MelganMultiscaleDiscriminator.__init__..F) kernel_sizestridepaddingcount_include_pad)super__init__r ModuleListrangediscriminators AvgPool1dpooling) selfrr num_scalesrrrrpooling_kernel_sizepooling_stridepooling_paddingr __class__rrr!s   z&MelganMultiscaleDiscriminator.__init__cCsFg}g}|jD]}||\}}||||||}q||fS)N)r$appendr&)r'xscoresfeatsdiscscorefeatrrrforward*s     z%MelganMultiscaleDiscriminator.forward) rrrrr r r r rrr )__name__ __module__ __qualname__r!r5 __classcell__rrr,rrs#rN)torchr'TTS.vocoder.models.melgan_discriminatorrModulerrrrrs