From 101a01f744a5fd81d6a3ee336e17319460a4ed69 Mon Sep 17 00:00:00 2001 From: James Betker Date: Thu, 2 Dec 2021 23:28:36 -0700 Subject: [PATCH] Fix dvae codes issue --- 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 1295a50b..888753c0 100644 --- a/codes/models/gpt_voice/lucidrains_dvae.py +++ b/codes/models/gpt_voice/lucidrains_dvae.py @@ -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: