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

Fix dvae test failure

Browse Source
This commit is contained in:
James Betker 2021-11-18 00:58:36 -07:00
parent 019acfa4c5
commit 1287915f3c
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

@ -228,7 +228,7 @@ class DiscreteVAE(nn.Module):
out = d(out) out = d(out)
else: else:
# This is non-differentiable, but gives a better idea of how the network is actually performing. # 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 # reconstruction loss
recon_loss = self.loss_fn(img, out, reduction='none') recon_loss = self.loss_fn(img, out, reduction='none')