fix bug related to device id in conditional_mnist.ipynb (#92)

atong01 · Jan 4, 2024 · 01a088a · 01a088a
1 parent 6b3adb4
commit 01a088a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/images/conditional_mnist.ipynb b/examples/images/conditional_mnist.ipynb
@@ -34,7 +34,7 @@
    "outputs": [],
    "source": [
     "use_cuda = torch.cuda.is_available()\n",
-    "device = torch.device(\"cuda:3\" if use_cuda else \"cpu\")\n",
+    "device = torch.device(\"cuda\" if use_cuda else \"cpu\")\n",
     "batch_size = 128\n",
     "n_epochs = 10\n",
     "\n",