Fixed noise() cuda return

diegoalejogm · web-flow · commit 154027137ff0 · 2018-07-01T16:50:32.000+02:00
diff --git a/1. Vanilla GAN PyTorch.ipynb b/1. Vanilla GAN PyTorch.ipynb
@@ -181,7 +181,7 @@
     "# Noise\n",
     "def noise(size):\n",
     "    n = Variable(torch.randn(size, 100))\n",
-    "    if torch.cuda.is_available(): return n.cuda \n",
+    "    if torch.cuda.is_available(): return n.cuda() \n",
     "    return n"
    ]
   },

Original file line number	Diff line number	Diff line change
`@@ -181,7 +181,7 @@`
`181`	`181`	`"# Noise\n",`
`182`	`182`	`"def noise(size):\n",`
`183`	`183`	`" n = Variable(torch.randn(size, 100))\n",`
`184`		`- " if torch.cuda.is_available(): return n.cuda \n",`
	`184`	`+ " if torch.cuda.is_available(): return n.cuda() \n",`
`185`	`185`	`" return n"`
`186`	`186`	`]`
`187`	`187`	`},`