Fix bug: default value of weights_only changed in PyTorch 2.6

2025-07-26 16:22:01 +02:00
parent 8612e012ab
commit d76b24569e
1 changed files with 1 additions and 1 deletions
@@ -61,7 +61,7 @@ class fsModel(BaseModel):

        # Id network
        netArc_checkpoint = opt.Arc_path
-        netArc_checkpoint = torch.load(netArc_checkpoint, map_location=torch.device("cpu"))
+        netArc_checkpoint = torch.load(netArc_checkpoint, map_location=torch.device("cpu"), weights_only=False)
        self.netArc = netArc_checkpoint
        self.netArc = self.netArc.to(device)
        self.netArc.eval()