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

Record codes more often

Browse Source
This commit is contained in:
James Betker 2021-12-07 09:22:45 -07:00
parent d0b2f931bf
commit 6ccff3f49f
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

@ -249,7 +249,7 @@ class DiscreteVAE(nn.Module):
def log_codes(self, codes): def log_codes(self, codes):
# This is so we can debug the distribution of codes being learned. # This is so we can debug the distribution of codes being learned.
if self.record_codes and self.internal_step % 50 == 0: if self.record_codes and self.internal_step % 10 == 0:
codes = codes.flatten() codes = codes.flatten()
l = codes.shape[0] l = codes.shape[0]
i = self.code_ind if (self.codes.shape[0] - self.code_ind) > l else self.codes.shape[0] - l i = self.code_ind if (self.codes.shape[0] - self.code_ind) > l else self.codes.shape[0] - l