Merge pull request yandexdataschool#64 from NickVeld/patch-18

justheuristic · web-flow · commit 930ee6a1c49d · 2019-11-27T02:44:35.000+03:00
week9:1d_gan_pytorch scalar tensor is not "[ x ]" now
diff --git a/week09_adversarial/simple_1d_gan_pytorch.ipynb b/week09_adversarial/simple_1d_gan_pytorch.ipynb
@@ -81,7 +81,7 @@
     "    disc_opt.zero_grad()\n",
     "    disc_loss.backward()\n",
     "    disc_opt.step()\n",
-    "    return disc_loss.data.numpy()[0]"
+    "    return disc_loss.data.numpy()"
    ]
   },
   {
@@ -106,7 +106,7 @@
     "    gen_opt.zero_grad()\n",
     "    gen_loss.backward()\n",
     "    gen_opt.step()\n",
-    "    return gen_loss.data.numpy()[0]"
+    "    return gen_loss.data.numpy()"
    ]
   },
   {
