o …wÖiÇã@sLddlZddlmZmZddlmZmZmZddlm Z Gdd„deƒZ dS)éN)ÚLossÚ typecheck)ÚLossTypeÚNormalDistributionSamplesTypeÚVoidType)Ú NeuralTypec@s<eZdZdZedd„ƒZedd„ƒZeƒddœdd „ƒZd S)ÚWaveGlowLossz3 A Loss module that computes loss for WaveGlow cCs0tdtƒƒtdtƒƒgttƒdgtdddœS)N)ÚBÚ flowgroupÚT©Ú elements_typeT)Úoptional)ÚzÚ log_s_listÚlog_det_W_listÚsigma)rrr©Úself©rúe/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/nemo/collections/tts/losses/waveglowloss.pyÚinput_typess üzWaveGlowLoss.input_typescCsdttƒdiS)NÚlossr)rrrrrrÚoutput_types"sÿzWaveGlowLoss.output_typesgð?)rc CsŒt|ƒD]\}}|dkrt |¡}||}q|t |¡}|||7}qt ||¡d||||} | | d¡| d¡| d¡S)Nréé)Ú enumerateÚtorchÚsumÚsize) rrrrrÚiÚlog_sÚlog_s_totalÚlog_det_W_totalrrrrÚforward(s ""zWaveGlowLoss.forwardN) Ú__name__Ú __module__Ú__qualname__Ú__doc__Úpropertyrrrr$rrrrrs r)rÚnemo.core.classesrrÚnemo.core.neural_types.elementsrrrÚ"nemo.core.neural_types.neural_typerrrrrrÚs