[TTS]Fix diffusion wavenet denoiser final conv init param #2868
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
PR types
Bug fixes
PR changes
Models
Describe
When I used WaveNetDenoiser to train diffusion-related tasks, I found that the loss did not decrease. After compared with diffsinger again, I modified the initialization of the last convolution. The meaning of stack and
dilation_by_cycle_length
in diffsinger is different.stack
is actually equal tothe number of layers
divided bydilation_by_cycle_length
.我自己使用WaveNetDenoiser训练diffusion相关任务时发现损失不下降,与diffsinger再次对齐后修改了最后一个卷积的初始化,并且stack与diffsinger中的dilation_by_cycle_length含义不同,stack实际上等于layer数除以dilation_by_cycle_length。