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

Fix dvae bug

Browse Source
This commit is contained in:
James Betker 2021-08-06 14:17:01 -06:00
parent 0799d95af5
commit 0237e96b34
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
codes/models/gpt_voice/lucidrains_dvae.py
View File

@ -125,8 +125,11 @@ class DiscreteVAE(nn.Module):
return images
def get_debug_values(self, step, __):
# Report annealing schedule
return {'histogram_codes': self.codes}
if self.record_codes:
# Report annealing schedule
return {'histogram_codes': self.codes}
else:
return {}
@torch.no_grad()
@eval_decorator