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

U
<00><>d<>
<00>@s`dd
lZdd
l
Z
ddlmZ
ddlmZ
ddlmZ
ddlm	Z	
ddd	<09>i
Z
Gd
d<0B>de<07>Zd
S)<0C>N<>
)
<01>GaussianDiffusion<6F>
<01>DiffNet)
<01>FastSpeech2Task)
<01>hparams<6D>wavenetc

Cs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>eZ
dZd
d<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_binsrrr<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
dkr<>|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>outputr2rrr<00>	run_models<












<02>








zDiffFsTask.run_modelcCsR|<00>|j
|
<01>}td
d<02>|<04><03>D<00>
<01>
}|
d<00><04>d|d<|j<05><06>d|d<||fS)Nc
Ss"g|]}
t|
t
j<02>r|
jr|
<01>qSr)<04>
isinstance<EFBFBD>torch<63>Tensor<6F>
requires_grad)<02>.0<EFBFBD>
vrrr<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>
_Zlog_outputs<74>
total_lossrrr<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_outrOrrr<00>validation_step?s










zDiffFsTask.validation_stepcCstj
jj|
td
dd<03>S)N<>decay_stepsg<00>?)
<01>gamma)rA<00>optim<69>lr_scheduler<65>StepLRr)r<00>	optimizerrrr<00>build_schedulerKs
zDiffFsTask.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_idxrrr<00>optimizer_stepNs






zDiffFsTask.optimizer_stepN)FF)	<09>__name__<5F>
__module__<EFBFBD>__qualname__rr?rQrZrarfrrrrrs


r)rArUZdiff.diffusionr<00>diff.netr<00>
tasks.tts.fs2r<00>
utils.hparamsrrrrrrr<00><module>
s



<06>