Set weights_only explicitly

Author: vl-dud

deepxde/model.py

@@ -1137,9 +1137,9 @@ def restore(self, save_path, device=None, verbose=0):
             self.net.load_weights(save_path)
         elif backend_name == "pytorch":
             if device is not None:
-                checkpoint = torch.load(save_path, map_location=torch.device(device))
+                checkpoint = torch.load(save_path, map_location=torch.device(device), weights_only=True)
             else:
-                checkpoint = torch.load(save_path)
+                checkpoint = torch.load(save_path, weights_only=True)
             self.net.load_state_dict(checkpoint["model_state_dict"])
             self.opt.load_state_dict(checkpoint["optimizer_state_dict"])
         elif backend_name == "paddle":