gpu

olive004 · Oct 30, 2024 · fc4d439 · fc4d439
1 parent ab766d8
commit fc4d439
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/notebooks/23_Monte_Carlo_adaptability_2.ipynb b/notebooks/23_Monte_Carlo_adaptability_2.ipynb
@@ -24,7 +24,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": null,
    "metadata": {},
    "outputs": [
     {
@@ -52,7 +52,7 @@
     "import os\n",
     "import sys\n",
     "\n",
-    "jax.config.update('jax_platform_name', 'cpu')\n",
+    "jax.config.update('jax_platform_name', 'gpu')\n",
     "\n",
     "if __package__ is None:\n",
     "\n",
@@ -89,7 +89,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -117,7 +117,7 @@
     "t0 = 0\n",
     "t1 = 300\n",
     "ts = np.linspace(t0, t1, 500)\n",
-    "tmax = 700\n",
+    "tmax = 1000\n",
     "dt0 = 0.0005555558569638981\n",
     "dt1_factor = 5\n",
     "dt1 = dt0 * dt1_factor\n",

diff --git a/notebooks/24_autodiff.ipynb b/notebooks/24_autodiff.ipynb