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

whoops

Browse Source
This commit is contained in:
James Betker 2022-06-19 23:22:30 -06:00
parent a659cd865c
commit 56c4a00e71
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -233,7 +233,7 @@ class TransformerDiffusionWithConditioningEncoder(nn.Module):
def get_grad_norm_parameter_groups(self): def get_grad_norm_parameter_groups(self):
groups = self.diff.get_grad_norm_parameter_groups() groups = self.diff.get_grad_norm_parameter_groups()
groups['conditioning_encoder'] = list(self.conditioning_encoder.parameters()) groups['conditioning_encoder'] = list(self.conditioning_encoder.parameters())
return return groups
def before_step(self, step): def before_step(self, step):
scaled_grad_parameters = list(itertools.chain.from_iterable([lyr.out.parameters() for lyr in self.diff.layers])) + \ scaled_grad_parameters = list(itertools.chain.from_iterable([lyr.out.parameters() for lyr in self.diff.layers])) + \