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

Fix dvae codes issue

Browse Source
This commit is contained in:
James Betker 2021-12-02 23:28:36 -07:00
parent 31fc693a8a
commit 101a01f744
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/gpt_voice/lucidrains_dvae.py
View File

@ -176,7 +176,7 @@ class DiscreteVAE(nn.Module):
return images
def get_debug_values(self, step, __):
if self.record_codes:
if self.record_codes and self.total_codes > 0:
# Report annealing schedule
return {'histogram_codes': self.codes[:self.total_codes]}
else: