Files
AudioGPT/NeuralSeq/usr/__pycache__/task.cpython-38.pyc

26 lines
3.4 KiB
Plaintext
Raw Normal View History

2023-03-20 15:43:44 +08:00
U
2023-03-24 17:19:37 +08:00
<00><>d<> <00>@s`ddlZddlZddlmZddlmZddlmZddlm Z ddd <09>iZ
2023-03-20 15:43:44 +08:00
Gd
d <0B>d e<07>Z dS) <0C>N<>)<01>GaussianDiffusion<6F><01>DiffNet)<01>FastSpeech2Task)<01>hparams<6D>wavenetcCs t|d<00>S)N<>audio_num_mel_binsr)<01>hp<68>r <00>H/mnt/sdc/hongzhiqing/github/AudioGPT/text_to_sing/DiffSinger/usr/task.py<70><lambda> <00>r c@s>eZdZdd<02>Zddd<05>Zdd<07>Zdd <09>Zd
d <0B>Zd d <0A>ZdS)<10>
DiffFsTaskc CsBtd}t|j|ttdt<00>tdtdtdtdd<07>|_dS)Nr <00>diff_decoder_type<70> timesteps<70>diff_loss_type<70>spec_min<69>spec_max)<07> phone_encoder<65>out_dims<6D>
denoise_fnr<00> loss_typerr)rrr<00> DIFF_DECODERS<52>model)<02>self<6C>mel_binsr r r <00>build_tts_models<06>zDiffFsTask.build_tts_modelFc
Cs|d}|d}|d}|d}|d} |d}
tdsB|<02>d<08>n|<02>d <09>} td
d kr<>|d } |d } |d}|<01>| | ||<07>|d<}|||| ||| |
|d<10>}i}d|kr<>|d|d<|j|d|||d<14>tdr<>|<00>|||<10>tdr<>|<00>|d|
|<10>|<03>s|S||fSdS)N<>
txt_tokens<EFBFBD>mels<6C>mel2ph<70>f0<66>uv<75>energy<67>
use_spk_id<EFBFBD> spk_embed<65>spk_ids<64>
pitch_type<EFBFBD>cwt<77>cwt_spec<65>f0_mean<61>f0_std<74>f0_cwt)r r%<00>ref_melsr!r"r#<00>infer<65> diff_loss<73>mel<65>dur)<01>losses<65>use_pitch_embed<65>use_energy_embed<65> energy_pred)r<00>get<65> cwt2f0_norm<72> add_dur_loss<73>add_pitch_loss<73>add_energy_loss)rr<00>sample<6C> return_outputr.r<00>targetr r!r"r#r%r)r*r+<00>outputr2r r r <00> run_models< <02> zDiffFsTask.run_modelcCsR|<00>|j|<01>}tdd<02>|<04><03>D<00><01>}|d<00><04>d|d<|j<05><06>d|d<||fS)NcSs"g|]}t|tj<02>r|jr|<01>qSr )<04>
isinstance<EFBFBD>torch<63>Tensor<6F> requires_grad)<02>.0<EFBFBD>vr r r <00>
<listcomp>:s z-DiffFsTask._training_step.<locals>.<listcomp>rr<00>
batch_size<EFBFBD>lr)r?r<00>sum<75>values<65>size<7A> scheduler<65>get_lr)rr;<00> batch_idx<64>_Z log_outputs<74>
total_lossr r r <00>_training_step8s
zDiffFsTask._training_stepcCs<>i}i|d<|j|j|ddd<04>\|d<}t|d<00><03><00>|d<|d|d<t<04>|<03>}|tdkr<>|j|j|ddd<04>\}}|<00>||d|d <00>|S)
Nr2TF)r<r.rP<00>nsamples<65>num_valid_plotsr<00>mel_out)r?rrIrJ<00>utils<6C>tensors_to_scalarsr<00>plot_mel)rr;rN<00>outputs<74> model_outrOr r r <00>validation_step?s 
2023-03-24 17:19:37 +08:00
 zDiffFsTask.validation_stepcCstjjj|tddd<03>S)N<> decay_stepsg<00>?)<01>gamma)rA<00>optim<69> lr_scheduler<65>StepLRr)r<00> optimizerr r r <00>build_schedulerKszDiffFsTask.build_schedulercCs@|dkr dS|<03><00>|<03><01>|jdk r<|j<02>|jtd<00>dS)NZaccumulate_grad_batches)<05>step<65> zero_gradrL<00> global_stepr)r<00>epochrNr`Z optimizer_idxr r r <00>optimizer_stepNs 
2023-03-20 15:43:44 +08:00
zDiffFsTask.optimizer_stepN)FF) <09>__name__<5F>
2023-03-24 17:19:37 +08:00
__module__<EFBFBD> __qualname__rr?rQrZrarfr r r r rs 
2023-03-20 15:43:44 +08:00
 r) rArUZdiff.diffusionr<00>diff.netr<00> tasks.tts.fs2r<00> utils.hparamsrrrr r r r <00><module>s    <06>