From 6ccff3f49fc757ad06c6478f6b8b99b18043560c Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 7 Dec 2021 09:22:45 -0700 Subject: [PATCH] Record codes more often --- codes/models/gpt_voice/lucidrains_dvae.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/models/gpt_voice/lucidrains_dvae.py b/codes/models/gpt_voice/lucidrains_dvae.py index e60271b3..ab9085fa 100644 --- a/codes/models/gpt_voice/lucidrains_dvae.py +++ b/codes/models/gpt_voice/lucidrains_dvae.py @@ -249,7 +249,7 @@ class DiscreteVAE(nn.Module): def log_codes(self, codes): # 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() l = codes.shape[0] i = self.code_ind if (self.codes.shape[0] - self.code_ind) > l else self.codes.shape[0] - l