o iD@sfdZddlZddlZddlZddlZddlZddlZddZddZ e dkr1e ej dddSdS) z1Neural machine translation model decoding script.NcCstjdtjtjd}|jdddd|jdddd|jd dd d|jd td d d|jddddd|jdtdddgdd|jdtddd|jdtddd|jddtdd d|jd!tdd"d|jd#td$d%d|jd&d'd'd(gd)d*|jd+td,d-|jd.tdd/d0|jd1tdd2d0|jd3td$d4d|jd5tdd6d|jd7tdd8d|jd9td:d;d|jdd|jd?td@dAd|jdBdCtdDdE|S)FzGet default arguments.zMTranslate text from speech using a speech translation model on one CPU or GPU) descriptionconfig_file_parser_classformatter_classz--configTzConfig file path)is_config_filehelpz --config2zBSecond config file path that overwrites the settings in `--config`z --config3zQThird config file path that overwrites the settings in `--config` and `--config2`z--ngpurzNumber of GPUs)typedefaultrz--dtype)float16float32float64r z,Float precision (only available in --api v2))choicesrrz --backendchainerpytorchzBackend library)rrr rz --debugmode Debugmodez--seedz Random seedz --verbosez-VzVerbose optionz --batchsizez9Batch size for beam search (0: means no batch processing)z--preprocess-confNz-The configuration file for the pre-processingz--apiv1v2zBeam search APIs v1: Default API. It only supports the ASRInterface.recognize method and DefaultRNNLM. v2: Experimental API. It supports any models that implements ScorerInterface.)rr rz --trans-jsonz#Filename of translation data (json))rrz--result-labelz$Filename of result label data (json))rrequiredrz--modelzModel file parameters to readz --model-confzModel config filez--nbestzOutput N-best hypothesesz --beam-sizez Beam sizez --penaltyg?zIncertion penaltyz --maxlenratiog@zInput length ratio to obtain max output length. If maxlenratio=0.0 (default), it uses a end-detect function to automatically find maximum hypothesis lengthsz --minlenratiogz.Input length ratio to obtain min output lengthz --tgt-langFz4target language ID (e.g., , , and etc.))rrr) configargparseArgumentParserYAMLConfigFileParserArgumentDefaultsHelpFormatteradd add_argumentintstrfloat)parserrG/home/ubuntu/.local/lib/python3.10/site-packages/espnet/bin/mt_trans.py get_parsers r cCsjt}||}|jdkrtjtjddn|jdkr$tjtjddn tjtjddtd|j dkrit j d}|durFtd n|j t |d krZtd td|j dkritd tdtd t j ddt|jtj|jtd|jtd|j|jdkrddlm}|jdkrtd|jd||dStd)zRun the main decoding function.rz>%(asctime)s (%(module)s:%(lineno)d) %(levelname)s: %(message)s)levelformatzSkip DEBUG/INFO messagesrCUDA_VISIBLE_DEVICESNz CUDA_VISIBLE_DEVICES is not set.,z/#gpus is not matched with CUDA_VISIBLE_DEVICES.z!The program only supports ngpu=1.zpython path = PYTHONPATHz(None)zset random seed = %dz backend = r)transr z `--dtype z#` is only available with `--api v2`zOnly pytorch are supported.)r parse_argsverboselogging basicConfigINFODEBUGWARNwarningngpuosenvirongetlenspliterrorsysexitinforandomseednpbackendespnet.mt.pytorch_backend.mtr'dtypeNotImplementedError ValueError)argsrcvdr'rrrmain}sP                  rD__main__r) __doc__r*r1r:r7rnumpyr<r rD__name__argvrrrrsj<