Merge pull request yandexdataschool#61 from NickVeld/patch-16

justheuristic · web-flow · commit fb2ba5c6e01b · 2019-11-27T02:44:07.000+03:00
week9:1d_gan_pytorch fix real and fake mixture
diff --git a/week09_adversarial/simple_1d_gan_pytorch.ipynb b/week09_adversarial/simple_1d_gan_pytorch.ipynb
@@ -46,7 +46,7 @@
     "gen = nn.Sequential(nn.Linear(1, 16), nn.ELU(), nn.Linear(16, 1))\n",
     "gen_opt = torch.optim.SGD(gen.parameters(), lr=1e-3)\n",
     "\n",
-    "# Discriminator converts 1d data into two logits (0th for real, 1st for fake). \n",
+    "# Discriminator converts 1d data into two logits (0th for fake, 1st for real). \n",
     "# It is deliberately made stronger than generator to make sure disc is slightly \"ahead in the game\".\n",
     "disc = nn.Sequential(nn.Linear(1, 64), nn.ELU(), nn.Linear(64, 2))\n",
     "disc_opt = torch.optim.SGD(disc.parameters(), lr=1e-2)"
