changed adversarial loss

This commit is contained in:
Eric Zelikman 2020-10-31 15:50:58 -04:00
parent b5f38cc639
commit 5bf2c86ddd
1 changed files with 2 additions and 2 deletions

View File

@ -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",