fix torch.load deprecated error

Signed-off-by: YunLiu <[email protected]>

Author: KumoLiu

2d_classification/mednist_tutorial.ipynb

@@ -575,7 +575,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "model.load_state_dict(torch.load(os.path.join(root_dir, \"best_metric_model.pth\")))\n",
+    "model.load_state_dict(torch.load(os.path.join(root_dir, \"best_metric_model.pth\"), weights_only=True))\n",
     "model.eval()\n",
     "y_true = []\n",
     "y_pred = []\n",

2d_segmentation/torch/unet_evaluation_array.py

@@ -58,7 +58,7 @@ def main(tempdir):
         num_res_units=2,
     ).to(device)
 
-    model.load_state_dict(torch.load("best_metric_model_segmentation2d_array.pth"))
+    model.load_state_dict(torch.load("best_metric_model_segmentation2d_array.pth", weights_only=True))
     model.eval()
     with torch.no_grad():
         for val_data in val_loader:

2d_segmentation/torch/unet_evaluation_dict.py

@@ -72,7 +72,7 @@ def main(tempdir):
         num_res_units=2,
     ).to(device)
 
-    model.load_state_dict(torch.load("best_metric_model_segmentation2d_dict.pth"))
+    model.load_state_dict(torch.load("best_metric_model_segmentation2d_dict.pth", weights_only=True))
 
     model.eval()
     with torch.no_grad():

3d_classification/torch/densenet_evaluation_array.py

@@ -57,7 +57,7 @@ def main():
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     model = monai.networks.nets.DenseNet121(spatial_dims=3, in_channels=1, out_channels=2).to(device)
 
-    model.load_state_dict(torch.load("best_metric_model_classification3d_array.pth"))
+    model.load_state_dict(torch.load("best_metric_model_classification3d_array.pth", weights_only=True))
     model.eval()
     with torch.no_grad():
         num_correct = 0.0

3d_classification/torch/densenet_evaluation_dict.py

@@ -63,7 +63,7 @@ def main():
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     model = monai.networks.nets.DenseNet121(spatial_dims=3, in_channels=1, out_channels=2).to(device)
 
-    model.load_state_dict(torch.load("best_metric_model_classification3d_dict.pth"))
+    model.load_state_dict(torch.load("best_metric_model_classification3d_dict.pth", weights_only=True))
     model.eval()
     with torch.no_grad():
         num_correct = 0.0

3d_registration/learn2reg_nlst_paired_lung_ct.ipynb

@@ -1127,7 +1127,7 @@
     "    )\n",
     "    # load model weights\n",
     "    filename_best_model = glob.glob(os.path.join(dir_load, \"segresnet_kpt_loss_best_tre*\"))[0]\n",
-    "    model.load_state_dict(torch.load(filename_best_model))\n",
+    "    model.load_state_dict(torch.load(filename_best_model, weights_only=True))\n",
     "    # to GPU\n",
     "    model.to(device)\n",
     "\n",

3d_registration/learn2reg_oasis_unpaired_brain_mr.ipynb

@@ -840,7 +840,7 @@
     "    model = VoxelMorph()\n",
     "    # load model weights\n",
     "    filename_best_model = glob.glob(os.path.join(dir_load, \"voxelmorph_loss_best_dice_*\"))[0]\n",
-    "    model.load_state_dict(torch.load(filename_best_model))\n",
+    "    model.load_state_dict(torch.load(filename_best_model, weights_only=True))\n",
     "    # to GPU\n",
     "    model.to(device)\n",
     "\n",

3d_registration/paired_lung_ct.ipynb

@@ -860,7 +860,7 @@
     "resource = \"https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/pair_lung_ct.pth\"\n",
     "dst = f\"{root_dir}/pretrained_weight.pth\"\n",
     "download_url(resource, dst)\n",
-    "model.load_state_dict(torch.load(dst))"
+    "model.load_state_dict(torch.load(dst, weights_only=True))"
    ]
   },
   {

3d_segmentation/brats_segmentation_3d.ipynb

@@ -733,7 +733,7 @@
     }
    ],
    "source": [
-    "model.load_state_dict(torch.load(os.path.join(root_dir, \"best_metric_model.pth\")))\n",
+    "model.load_state_dict(torch.load(os.path.join(root_dir, \"best_metric_model.pth\"), weights_only=True))\n",
     "model.eval()\n",
     "with torch.no_grad():\n",
     "    # select one image to evaluate and visualize the model output\n",
@@ -835,7 +835,7 @@
     }
    ],
    "source": [
-    "model.load_state_dict(torch.load(os.path.join(root_dir, \"best_metric_model.pth\")))\n",
+    "model.load_state_dict(torch.load(os.path.join(root_dir, \"best_metric_model.pth\"), weights_only=True))\n",
     "model.eval()\n",
     "\n",
     "with torch.no_grad():\n",
@@ -977,7 +977,7 @@
    "source": [
     "onnx_model_path = os.path.join(root_dir, \"best_metric_model.onnx\")\n",
     "ort_session = onnxruntime.InferenceSession(onnx_model_path)\n",
-    "model.load_state_dict(torch.load(os.path.join(root_dir, \"best_metric_model.pth\")))\n",
+    "model.load_state_dict(torch.load(os.path.join(root_dir, \"best_metric_model.pth\"), weights_only=True))\n",
     "model.eval()\n",
     "\n",
     "with torch.no_grad():\n",

3d_segmentation/challenge_baseline/run_net.py

@@ -219,7 +219,7 @@ def infer(data_folder=".", model_folder="runs", prediction_folder="output"):
 
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     net = get_net().to(device)
-    net.load_state_dict(torch.load(ckpt, map_location=device))
+    net.load_state_dict(torch.load(ckpt, map_location=device, weights_only=True))
     net.eval()
 
     image_folder = os.path.abspath(data_folder)