mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

qcodes

Browse Source
This commit is contained in:
James Betker 2022-06-10 16:23:08 -06:00
parent 38a00f29c0
commit aca9024d9b
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_diffusion8.py
View File

@ -237,7 +237,7 @@ class TransformerDiffusionWithQuantizer(nn.Module):
def get_debug_values(self, step, __):
if self.quantizer.total_codes > 0:
return {'histogram_codes': self.quantizer.codes[:self.quantizer.total_codes],
return {'histogram_quant_codes': self.quantizer.codes[:self.quantizer.total_codes],
'gumbel_temperature': self.quantizer.quantizer.temperature}
else:
return {}