From 5bf2c86dddc00d75c959d9d4afabb91689fdf79f Mon Sep 17 00:00:00 2001 From: Eric Zelikman Date: Sat, 31 Oct 2020 15:50:58 -0400 Subject: [PATCH] changed adversarial loss --- C3W3_MUNIT_(Optional).ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/C3W3_MUNIT_(Optional).ipynb b/C3W3_MUNIT_(Optional).ipynb index 0274ba4..aa4686d 100644 --- a/C3W3_MUNIT_(Optional).ipynb +++ b/C3W3_MUNIT_(Optional).ipynb @@ -976,8 +976,8 @@ " c_a_loss, s_b_loss, x_ab = self.loss.latent_recon_loss(c_a, s_b, self.gen_b)\n", "\n", " # Compute adversarial losses\n", - " gen_a_adv_loss = self.loss.adversarial_loss(x_ba, self.dis_a, False)\n", - " gen_b_adv_loss = self.loss.adversarial_loss(x_ab, self.dis_b, False)\n", + " gen_a_adv_loss = self.loss.adversarial_loss(x_ba, self.dis_a, True)\n", + " gen_b_adv_loss = self.loss.adversarial_loss(x_ab, self.dis_b, True)\n", "\n", " # Sum up losses for gen\n", " gen_loss = (\n",