a-One-Fan/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Pull Requests Activity

don't populate gn5

Browse Source
This commit is contained in:
James Betker 2022-06-19 21:07:13 -06:00
parent a5d2123daa
commit b9f53a3ff9
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
codes/models/audio/music/unet_diffusion_waveform_gen3.py
View File

@ -110,6 +110,7 @@ class StackedResidualBlock(TimestepBlock):
self.add_module( self.add_module(
f'conv{i + 1}', f'conv{i + 1}',
nn.Conv1d(channels + i * gc, out_channels, 3, 1, 1)) nn.Conv1d(channels + i * gc, out_channels, 3, 1, 1))
if i != 4:
self.add_module(f'gn{i+1}', nn.GroupNorm(num_groups=8, num_channels=out_channels)) self.add_module(f'gn{i+1}', nn.GroupNorm(num_groups=8, num_channels=out_channels))
self.lrelu = nn.LeakyReLU(negative_slope=0.2, inplace=True) self.lrelu = nn.LeakyReLU(negative_slope=0.2, inplace=True)
zero_module(self.conv5) zero_module(self.conv5)