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

B
<00>Xd<>
<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)
<01>DiffNet)
<01>FastSpeech2Task)
<01>hparams<6D>wavenetc

Cst|d
<00>
S)N<>audio_num_mel_bins)
r)
<01>hp<68>r
<00>K/mnt/sdc/hongzhiqing/code/audio_chatgpt/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_binsr
r
r<00>build_tts_models






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)rr$<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>targetrr r!r"r$r(r)r*<00>outputr1r
r
r<00>	run_models2





















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>
vr
r
r<00>
<listcomp>:sz-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_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)
Nr1TF)r;r-rO<00>nsamples<65>num_valid_plotsr<00>mel_out)r>rrHrI<00>utils<6C>tensors_to_scalarsr<00>plot_mel)rr:rM<00>outputs<74>	model_outrNr
r
r<00>validation_step?s










zDiffFsTask.validation_stepcCstj
jj|
td
dd<03>S)NZdecay_stepsg<00>?)
<01>gamma)r@<00>optim<69>lr_scheduler<65>StepLRr)r<00>	optimizerr
r
r<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_gradrK<00>global_stepr)r<00>epochrMr^Z
optimizer_idxr
r
r<00>optimizer_stepNs






zDiffFsTask.optimizer_stepN)FF)	<09>__name__<5F>
__module__<EFBFBD>__qualname__rr>rPrYr_rdr
r
r
rrs


r)r@rTZdiff.diffusionr<00>diff.netr<00>
tasks.tts.fs2r<00>
utils.hparamsrrrr
r
r
r<00><module>
s