From 1287915f3c8f3bb47f3a101491d857a6f3a45d0d Mon Sep 17 00:00:00 2001 From: James Betker Date: Thu, 18 Nov 2021 00:58:36 -0700 Subject: [PATCH] Fix dvae test failure --- 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 37aa8f47..673bf9eb 100644 --- a/codes/models/gpt_voice/lucidrains_dvae.py +++ b/codes/models/gpt_voice/lucidrains_dvae.py @@ -228,7 +228,7 @@ class DiscreteVAE(nn.Module): out = d(out) else: # This is non-differentiable, but gives a better idea of how the network is actually performing. - out = self.decode(codes) + out, _ = self.decode(codes) # reconstruction loss recon_loss = self.loss_fn(img, out, reduction='none')