diff --git a/docs/tutorials/pairwise-registration-intro.ipynb b/docs/tutorials/pairwise-registration-intro.ipynb index a57a010..a0fd2ea 100644 --- 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 },