Update restoration.ipynb

laura-gibbs · Jan 13, 2019 · 74cb8d2 · 74cb8d2
1 parent 0284191
commit 74cb8d2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/restoration.ipynb b/restoration.ipynb
@@ -25,7 +25,7 @@
     "%matplotlib inline\n",
     "\n",
     "import os\n",
-    "os.environ['CUDA_VISIBLE_DEVICES'] = '1'\n",
+    "#os.environ['CUDA_VISIBLE_DEVICES'] = '1'\n",
     "\n",
     "import numpy as np\n",
     "from models.resnet import ResNet\n",