From 3c284b4ab4a2cff12256dc62e595f742a89be406 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Anna=20B=C3=B8gevang=20Ekner?= <s193396@dtu.dk> Date: Tue, 4 Feb 2025 14:36:23 +0100 Subject: [PATCH] updated segmentation pipeline notebook --- docs/notebooks/segmentation_pipeline.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/notebooks/segmentation_pipeline.ipynb b/docs/notebooks/segmentation_pipeline.ipynb index 326ab9d7..13c986e9 100644 --- a/docs/notebooks/segmentation_pipeline.ipynb +++ b/docs/notebooks/segmentation_pipeline.ipynb @@ -528,7 +528,7 @@ "outputs": [], "source": [ "# defining model\n", - "model = qim3d.ml.models.UNet(size = 'small', dropout = 0.25)" + "model = qim3d.ml.models.UNet2D(size = 'small', dropout = 0.25)" ] }, { -- GitLab