Fix TF backend string

alan-turing-institute · Jun 6, 2023 · 135c4ae · 135c4ae
1 parent b34e5a4
commit 135c4ae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/notebooks/around_deepsensor_in_80_lines.ipynb b/notebooks/around_deepsensor_in_80_lines.ipynb
@@ -106,7 +106,7 @@
     "    if torch.cuda.is_available():\n",
     "        torch.set_default_device(\"cuda\")\n",
     "        B.set_global_device(\"cuda:0\")\n",
-    "elif deepsensor.backend.str == \"tensorflow\":\n",
+    "elif deepsensor.backend.str == \"tf\":\n",
     "    import tensorflow as tf\n",
     "    if tf.test.is_gpu_available():\n",
     "        tf.config.set_visible_devices(tf.config.list_physical_devices('GPU')[0], 'GPU')\n",