This commit is contained in:
wassname
2019-11-03 10:05:48 +08:00
parent 24485b65d0
commit be042a91bb
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -253,7 +253,7 @@ class Decoder(nn.Module):
x = self._target_transform(target_x)
# concatenate target_x and representation
if r is not None:
if self.use_deterministic_path:
z = torch.cat([r, z], dim=-1)
representation = torch.cat([z, x], dim=-1)