changed adversarial loss
This commit is contained in:
parent
b5f38cc639
commit
5bf2c86ddd
|
@ -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",
|
" c_a_loss, s_b_loss, x_ab = self.loss.latent_recon_loss(c_a, s_b, self.gen_b)\n",
|
||||||
"\n",
|
"\n",
|
||||||
" # Compute adversarial losses\n",
|
" # Compute adversarial losses\n",
|
||||||
" gen_a_adv_loss = self.loss.adversarial_loss(x_ba, self.dis_a, 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, False)\n",
|
" gen_b_adv_loss = self.loss.adversarial_loss(x_ab, self.dis_b, True)\n",
|
||||||
"\n",
|
"\n",
|
||||||
" # Sum up losses for gen\n",
|
" # Sum up losses for gen\n",
|
||||||
" gen_loss = (\n",
|
" gen_loss = (\n",
|
||||||
|
|
Loading…
Reference in New Issue