mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki 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

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

@ -110,7 +110,8 @@ class StackedResidualBlock(TimestepBlock):
self.add_module(
f'conv{i + 1}',
nn.Conv1d(channels + i * gc, out_channels, 3, 1, 1))
self.add_module(f'gn{i+1}', nn.GroupNorm(num_groups=8, num_channels=out_channels))
if i != 4:
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)
zero_module(self.conv5)
self.drop = nn.Dropout(p=dropout)