set device to auto in notebook

angelolab · Mar 5, 2024 · 0f929b7 · 0f929b7
1 parent 2a8cb83
commit 0f929b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/templates/1_Nimbus_Predict.ipynb b/templates/1_Nimbus_Predict.ipynb
@@ -198,7 +198,7 @@
     "    test_time_aug=True,\n",
     "    input_shape=[1024,1024],\n",
     "    suffix=\".tiff\",\n",
-    "    device=\"cpu\",\n",
+    "    device=\"auto\",\n",
     ")\n",
     "\n",
     "# check if all inputs are valid\n",