From 01a088a812da700b19ab214d54e572dcebd6e731 Mon Sep 17 00:00:00 2001
From: Imahn Shekhzadeh <120030096+ImahnShekhzadeh@users.noreply.github.com>
Date: Thu, 4 Jan 2024 16:31:14 +0100
Subject: [PATCH] fix bug related to device id in `conditional_mnist.ipynb`
 (#92)

---
 examples/images/conditional_mnist.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/images/conditional_mnist.ipynb b/examples/images/conditional_mnist.ipynb
index 875321c..c5b19b8 100644
--- 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",