o id @s6dZddlZddlmZddlmZGdddZdS)zMT Interface module.N get_parser)fill_missing_argsc@sbeZdZdZeddZededefddZdd Z dd d Z dd dZ ddZ e ddZd S) MTInterfacez-MT Interface for ESPnet model implementation.cCs|S)zAdd arguments to parser.parserrrL/home/ubuntu/.local/lib/python3.10/site-packages/espnet/nets/mt_interface.py add_arguments szMTInterface.add_argumentsidimodimcKs:dd}tjdi|}t||}t||j}||||S)aInitialize this class with python-level args. Args: idim (int): The number of an input feature dim. odim (int): The number of output vocab. Returns: ASRinterface: A new instance of ASRInterface. cSs t|ddS)NF)requiredrrrrr wraps zMTInterface.build..wrapNr)argparse Namespacerr )clsr r kwargsrargsrrr builds     zMTInterface.buildcCtd)aeCompute loss for training. :param xs: For pytorch, batch of padded source sequences torch.Tensor (B, Tmax, idim) For chainer, list of source sequences chainer.Variable :param ilens: batch of lengths of source sequences (B) For pytorch, torch.Tensor For chainer, list of int :param ys: For pytorch, batch of padded source sequences torch.Tensor (B, Lmax) For chainer, list of source sequences chainer.Variable :return: loss value :rtype: torch.Tensor for pytorch, chainer.Variable for chainer z!forward method is not implementedNotImplementedErrorselfxsilensysrrr forward%szMTInterface.forwardNcCr)aaTranslate x for evaluation. :param ndarray x: input acouctic feature (B, T, D) or (T, D) :param namespace trans_args: argment namespace contraining options :param list char_list: list of characters :param torch.nn.Module rnnlm: language model module :return: N-best decoding results :rtype: list z#translate method is not implementedrrx trans_args char_listrnnlmrrr translate6 zMTInterface.translatecCr)atBeam search implementation for batch. :param torch.Tensor x: encoder hidden state sequences (B, Tmax, Henc) :param namespace trans_args: argument namespace containing options :param list char_list: list of characters :param torch.nn.Module rnnlm: language model module :return: N-best decoding results :rtype: list z$Batch decoding is not supported yet.rrrrr translate_batchBr$zMTInterface.translate_batchcCr)a_Calculate attention. :param list xs: list of padded input sequences [(T1, idim), (T2, idim), ...] :param ndarray ilens: batch of lengths of input sequences (B) :param list ys: list of character id sequence tensor [(L1), (L2), (L3), ...] :return: attention weights (B, Lmax, Tmax) :rtype: float ndarray z2calculate_all_attentions method is not implementedrrrrr calculate_all_attentionsNs z$MTInterface.calculate_all_attentionscCsddlm}|S)zGet attention plot class.r)PlotAttentionReport)espnet.asr.asr_utilsr')rr'rrr attention_plot_classYs z MTInterface.attention_plot_class)NN)__name__ __module__ __qualname____doc__ staticmethodr classmethodintrrr#r%r&propertyr)rrrr rs     r)r-respnet.bin.asr_trainrespnet.utils.fill_missing_argsrrrrrr s