From 0fd967cfd685d0dd735dfc51a23136f5041bf46e Mon Sep 17 00:00:00 2001
From: YunLiu <55491388+KumoLiu@users.noreply.github.com>
Date: Wed, 19 Mar 2025 15:12:22 +0800
Subject: [PATCH] fix #1901

Signed-off-by: YunLiu <55491388+KumoLiu@users.noreply.github.com>
---
 model_zoo/TCIA_PROSTATEx_Prostate_MRI_Anatomy_Model.ipynb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/model_zoo/TCIA_PROSTATEx_Prostate_MRI_Anatomy_Model.ipynb b/model_zoo/TCIA_PROSTATEx_Prostate_MRI_Anatomy_Model.ipynb
index f1dc6f0e5f..a9fc2da09c 100644
--- a/model_zoo/TCIA_PROSTATEx_Prostate_MRI_Anatomy_Model.ipynb
+++ b/model_zoo/TCIA_PROSTATEx_Prostate_MRI_Anatomy_Model.ipynb
@@ -116,7 +116,7 @@
     "!python -c \"import cv2\" || pip install -q opencv-python-headless\n",
     "\n",
     "# These are the libraries used to read DICOM Seg objects.\n",
-    "!python -m pip install -q pydicom pydicom-seg\n",
+    "!python -m pip install -q pydicom==2.4.4 pydicom-seg\n",
     "\n",
     "# Install tcia_utils to download the datasets.\n",
     "!python -m pip install --upgrade -q --no-deps tcia_utils\n",
@@ -500,7 +500,7 @@
    },
    "outputs": [],
    "source": [
-    "model.load_state_dict(torch.load(checkpoint, map_location=device))\n",
+    "model.load_state_dict(torch.load(checkpoint, map_location=device, weights_only=True))\n",
     "model.eval()\n",
     "results = []\n",
     "with torch.no_grad():\n",