Use weights_only for load

spot_rl_experiments/spot_rl/real_policy.py

@@ -87,7 +87,7 @@ def __init__(
             )
         else:
             if isinstance(checkpoint_path, str):
-                checkpoint = torch.load(checkpoint_path, map_location="cpu")
+                checkpoint = torch.load(checkpoint_path, map_location="cpu", weights_only=True)
             else:
                 checkpoint = checkpoint_path
 
@@ -461,7 +461,7 @@ def __init__(
                 ),
             }
         )
-        checkpoint = torch.load(mixer_checkpoint_path, map_location="cpu")
+        checkpoint = torch.load(mixer_checkpoint_path, map_location="cpu", weights_only=True)
         checkpoint["config"].RL.POLICY["nav_checkpoint_path"] = nav_checkpoint_path
         checkpoint["config"].RL.POLICY["gaze_checkpoint_path"] = gaze_checkpoint_path
         checkpoint["config"].RL.POLICY["place_checkpoint_path"] = place_checkpoint_path

spot_rl_experiments/utils/pytorch_to_torchscript.py

@@ -121,7 +121,7 @@ def __init__(
         self.device = torch.device(device)
 
         # Load the checkpoint
-        checkpoint = torch.load(checkpoint_path, map_location="cpu")
+        checkpoint = torch.load(checkpoint_path, map_location="cpu", weights_only=True)
 
         # Load the config
         config = checkpoint["config"]