vrr/Ruview
2
0
Fork 0
mirror of https://github.com/ruvnet/RuView.git synced 2026-04-28 05:59:32 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 10

Merge pull request #133 from ruvnet/fix/pickle-deserialization-safety

Browse source 
fix: safe PyTorch model loading (weights_only=True)
This commit is contained in:
rUv 2026-03-03 18:11:29 -05:00 • committed by GitHub
commit 26655d397e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
references/wifi_densepose_pytorch.py
View file

@ -441,7 +441,7 @@ class WiFiDensePoseTrainer:
}, path)
def load_model(self, path):
checkpoint = torch.load(path)
checkpoint = torch.load(path, map_location=self.device, weights_only=True)
self.model.load_state_dict(checkpoint['model_state_dict'])
self.optimizer.load_state_dict(checkpoint['optimizer_state_dict'])