o iN*@sdZddlZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl mZddlmZddZd d Zed krMeejd ddSdS) z'Voice conversion model training script.N) __version__) TTSInterface) strtobool)BATCH_COUNT_CHOICEScCstjdtjtjd}|jdddd|jdddd|jd dd d|jd d td d|jddtddgdd|jdtddd|jddtdd|jddtdd|jdddtdd d!|jd"d#td$d%d&|jd'd(d)td*d|jd+d tdd,d!|jd-d.td/d|jd0dtd1d|jd2d3td4d|jd5td6d7|jd8td9d7|jd:tdd;d|jdtd?d@d&|jdAd)tddBd!|jdCdDtgdEddFdG|jdHdItdJdK|jdLdMdNd)tdOd|jdPd)tdQd|jdRd)tdSd|jdTd)tdUd|jdVd)tdWd|jdXdYd.tdZd[d\|jd]d^d_tdZd`d\|jdad)tdbd|jdctd ddd&|jdedft dgd|jdhdft did|jdjdktgdldmd|jdndtdod|jdpdqt drd|jdsdtt dud|jdvdtt dwd|jdxdydztd{d|jd|d}tdd~d!|jddtddd!|jddt dd|jddtdd|jddft dd|jdd tdd|jdddddd|jdd tdd|jdddddd|jdd dddd|S)z!Get parser of training arguments.zHTrain a new voice conversion (VC) model on one CPU, one or multiple GPUs) descriptionconfig_file_parser_classformatter_classz--configTzconfig file path)is_config_filehelpz --config2zCsecond config file path that overwrites the settings in `--config`.z --config3zRthird config file path that overwrites the settings in `--config` and `--config2`.z--ngpuNz5Number of GPUs. If not given, use all visible devices)defaulttyper z --backendpytorchchainerzBackend library)r r choicesr z--outdirzOutput directory)r requiredr z --debugmode Debugmodez--seedz Random seedz--resumez-r?z!Resume the training from snapshot)r r nargsr z --minibatchesz-Nz-1z&Process only N minibatches (for debug))r r r z --verbosez-VrzVerbose optionz--tensorboard-dirzTensorboard log directory pathz--eval-interval-epochsdzEvaluation interval epochsz--save-interval-epochszSave interval epochsz--report-interval-iters zReport interval iterationsz--srcspkzSource speaker)r r z--trgspkzTarget speakerz --train-jsonzFilename of training jsonz --valid-jsonzFilename of validation jsonz--model-modulez7espnet.nets.pytorch_backend.e2e_tts_tacotron2:Tacotron2zmodel defined modulez --sortagradzFHow many epochs to use sortagrad for. 0 = deactivated, -1 = all epochsz--batch-sort-keyshuffle)routputinputz@Batch sorting key. "shuffle" only work with --batch-count "seq".)r r rrr z --batch-countautozKHow to count batch_size. The default (auto) will find how to count by args.)r rr z --batch-sizez --batch-seqsz-bz*Maximum seqs in a minibatch (0 to disable)z --batch-binsz*Maximum bins in a minibatch (0 to disable)z--batch-frames-inz2Maximum input frames in a minibatch (0 to disable)z--batch-frames-outz3Maximum output frames in a minibatch (0 to disable)z--batch-frames-inoutz9Maximum input+output frames in a minibatch (0 to disable)z --maxlen-inz--batch-seq-maxlen-inMLzPWhen --batch-count=seq, batch size is reduced if the input sequence length > ML.)r r metavarr z --maxlen-outz--batch-seq-maxlen-outzPWhen --batch-count=seq, batch size is reduced if the output sequence length > MLz--num-iter-processeszNumber of processes of iteratorz--preprocess-confz-The configuration file for the pre-processingz--use-speaker-embeddingFz Whether to use speaker embeddingz--use-second-targetzWhether to use second targetz--optadam)rnoamlamb Optimizerz --accum-gradzNumber of gradient accumurationz--lrgMbP?zLearning rate for optimizerz--epsgư>zEpsilon for optimizerz--weight-decayz&Weight decay coefficient for optimizerz--epochsz-ezNumber of maximum epochsz--early-stop-criterionzvalidation/main/lossz=Value to monitor to trigger an early stopping of the trainingz --patiencezINumber of epochs to wait without improvement before stopping the trainingz --grad-clipzGradient norm threshold to clipz--num-save-attentionz*Number of samples of attention to be savedz--keep-all-data-on-memz"Whether to keep all data on memoryz --enc-initz-Pre-trained model path to initialize encoder.z--enc-init-modszenc.cfdddDS)Ncg|] }dkrt|qSrstr.0modsG/home/ubuntu/.local/lib/python3.10/site-packages/espnet/bin/vc_train.py 0get_parser....,splitr.r0r.r1zget_parser..z%(asctime)s (%(module)s:%(lineno)d) %(levelname)s: %(message)s)levelformatzSkip DEBUG/INFO messagesNCUDA_VISIBLE_DEVICESr5z CUDA_VISIBLE_DEVICES is not set.z nvidia-smiz-L)stdoutstderr rzngpu: zrandom seed = %dr )trainzOnly pytorch is supported.)&rCparse_known_argsespnet.utils.dynamic_importrD model_module issubclassr add_arguments parse_argsrversionverboselogging basicConfigINFOWARNwarningngpuosenvirongetlenr7 subprocessrunPIPECalledProcessErrorFileNotFoundErrorrIdecodeinfoseedrandomnpbackendespnet.vc.pytorch_backend.vcrKNotImplementedError) cmd_argsrBargs_rD model_classcvdrYprKr0r0r1main0sP              rq__main__r)__doc__rTrZrfr^sysr:numpyrgespnetrespnet.nets.tts_interfacerespnet.utils.cli_utilsrespnet.utils.training.batchfyrrCrq__name__argvr0r0r0r1s&    ?