Update galaxies.ipynb

s-thiele · Nov 29, 2022 · 7805b32 · 7805b32
1 parent fc9c1a1
commit 7805b32
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/galaxies.ipynb b/galaxies.ipynb
@@ -85,7 +85,7 @@
         "\n",
         "\n",
         "# Make sure tf does not allocate gpu memory.\n",
-        "tf.config.experimental.set_visible_devices([], \"GPU\")\n",
+        "# tf.config.experimental.set_visible_devices([], \"GPU\")\n",
         "\n",
         "# init rng seed\n",
         "rng = jax.random.PRNGKey(0)"
@@ -236,4 +236,4 @@
       }
     }
   ]
-}
+}