eschmidbauer/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix2

Browse Source
This commit is contained in:
James Betker 2022-04-20 00:03:29 -06:00
parent 0af0051399
commit 24fdafd855
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/models/audio/music/unet_diffusion_waveform_gen2.py
View File

@ -395,8 +395,6 @@ class DiffusionTts(nn.Module):
def get_grad_norm_parameter_groups(self): def get_grad_norm_parameter_groups(self):
if self.freeze_main_net:
return {}
groups = { groups = {
'input_blocks': list(self.input_blocks.parameters()), 'input_blocks': list(self.input_blocks.parameters()),
'output_blocks': list(self.output_blocks.parameters()), 'output_blocks': list(self.output_blocks.parameters()),