pytorch model/dataset

microsoft · Sep 26, 2023 · ce790cc · ce790cc
1 parent 47dd3a4
commit ce790cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...responsibleaidashboard/vision/responsibleaidashboard-standard-pytorch-model-dataset.ipynb b/...responsibleaidashboard/vision/responsibleaidashboard-standard-pytorch-model-dataset.ipynb
@@ -91,7 +91,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "device = 'cpu'\n",
+    "device = 'cuda'\n",
     "\n",
     "class CNN(nn.Module):\n",
     "    def __init__(self):\n",