[doc] Fix regularization weights of pairwise registration tutorial

BioMedIA · Feb 1, 2024 · 28dfdd7 · 28dfdd7
1 parent 53b90ea
commit 28dfdd7
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/docs/tutorials/pairwise-registration-intro.ipynb b/docs/tutorials/pairwise-registration-intro.ipynb
@@ -753,7 +753,7 @@
     "    source=source_pyramid,\n",
     "    transform=(spatial.FreeFormDeformation, {\"stride\": 2}),\n",
     "    optimizer=(optim.Adam, {\"lr\": 1e-2}),\n",
-    "    loss_fn=loss_fn(w_bending=1e-1),\n",
+    "    loss_fn=loss_fn(w_bending=1e-3),\n",
     "    device=device,\n",
     ")\n",
     "transform = transform.cpu()\n",
@@ -858,7 +858,7 @@
     "    source=source_pyramid,\n",
     "    transform=(spatial.StationaryVelocityFreeFormDeformation, {\"stride\": 2}),\n",
     "    optimizer=(optim.Adam, {\"lr\": 1e-2}),\n",
-    "    loss_fn=loss_fn(w_bending=1e-1),\n",
+    "    loss_fn=loss_fn(w_bending=1e-3),\n",
     "    device=device,\n",
     ")"
    ]
@@ -915,7 +915,7 @@
     "circle_to_c_transform = multi_resolution_registration(\n",
     "    target=target_cshape[0],\n",
     "    source=source_circle[0],\n",
-    "    loss_fn=loss_fn(w_diffusion=100),\n",
+    "    loss_fn=loss_fn(w_diffusion=1e-3),\n",
     "    transform=spatial.StationaryVelocityFreeFormDeformation,\n",
     "    optimizer=(optim.Adam, {\"lr\": 1e-2}),\n",
     "    iterations=[200, 200, 300, 100],\n",
@@ -957,7 +957,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.11"
+   "version": "3.10.6"
   },
   "orig_nbformat": 4
  },