mod: descs

BKL255 · Feb 20, 2019 · 22ac14b · 22ac14b
1 parent 55370a9
commit 22ac14b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/code/deep/DaNN/DaNN.py b/code/deep/DaNN/DaNN.py
@@ -16,5 +16,5 @@ def forward(self, src, tar):
         x_tar = self.dropout(x_tar)
         x_src_mmd = self.relu(x_src)
         x_tar_mmd = self.relu(x_tar)
-        x_src = self.layer_hidden(x_src_mmd)
-        return x_src, x_src_mmd, x_tar_mmd
+        y_src = self.layer_hidden(x_src_mmd)
+        return y_src, x_src_mmd, x_tar_mmd