o ¡¿¯i ã@sZdZddlmZzddlZWn eydZYn wGdd„dejƒZGdd„deƒZdS)zTTS Interface realted modules.é©Ú torch_loadNc@seZdZdZdd„ZdS)ÚReporterzReporter module.cCs|D] }tj ||¡qdS)z Report values from a given dict.N)ÚchainerÚreporterÚreport)ÚselfÚdictsÚd©r úM/home/ubuntu/.local/lib/python3.10/site-packages/espnet/nets/tts_interface.pyrsÿzReporter.reportN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rr r r r rs rc@s\eZdZdZedd„ƒZdd„Zdd„Zdd „Zd d „Z d d „Z e dd„ƒZ e dd„ƒZ dS)Ú TTSInterfacez.TTS Interface for ESPnet model implementation.cCs|S)z&Add model specific argments to parser.r )Úparserr r r Ú add_argumentsszTTSInterface.add_argumentscCs tƒ|_dS)zInitilize TTS module.N)rr©rr r r Ú__init__!s zTTSInterface.__init__cOótdƒ‚)z^Calculate TTS forward propagation. Returns: Tensor: Loss value. z!forward method is not implemented©ÚNotImplementedError©rÚargsÚkwargsr r r Úforward%ózTTSInterface.forwardcOr)aGenerate the sequence of features given the sequences of characters. Returns: Tensor: The sequence of generated features (L, odim). Tensor: The sequence of stop probabilities (L,). Tensor: The sequence of attention weights (L, T). z#inference method is not implementedrrr r r Ú inference.s zTTSInterface.inferencecOr)zyCalculate TTS attention weights. Args: Tensor: Batch of attention weights (B, Lmax, Tmax). z2calculate_all_attentions method is not implementedrrr r r Úcalculate_all_attentions9rz%TTSInterface.calculate_all_attentionscCst||ƒdS)z!Load pretrained model parameters.Nr)rÚ model_pathr r r Úload_pretrained_modelBsz"TTSInterface.load_pretrained_modelcCsddlm}|S)zPlot attention weights.r)ÚPlotAttentionReport)Úespnet.asr.asr_utilsr")rr"r r r Úattention_plot_classFs z!TTSInterface.attention_plot_classcCsdgS)a®Return base key names to plot during training. The keys should match what `chainer.reporter` reports. if you add the key `loss`, the reporter will report `main/loss` and `validation/main/loss` values. also `loss.png` will be created as a figure visulizing `main/loss` and `validation/main/loss` values. Returns: list[str]: Base keys to plot during training. Úlossr rr r r Úbase_plot_keysMszTTSInterface.base_plot_keysN)r rrrÚ staticmethodrrrrrr!Úpropertyr$r&r r r r rs     r) rr#rrÚ ImportErrorrÚChainÚobjectrr r r r Ús   ÿ