Merge branch 'master' of github.com:kaanaksit/odak

kaanaksit · Nov 14, 2024 · 70c4136 · 70c4136
2 parents 7c512b8 + 3dcd315
commit 70c4136
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/odak/learn/wave/models.py b/odak/learn/wave/models.py
@@ -259,7 +259,7 @@ def load_weights(self, weight_filename, key_mapping_filename):
                                Path to the JSON file containing the key mappings.
         """
         # Load old model weights
-        old_model_weights = torch.load(weight_filename, map_location = self.device)
+        old_model_weights = torch.load(weight_filename, map_location = self.device,weights_only=True)
 
         # Load key mappings from JSON file
         with open(key_mapping_filename, 'r') as json_file: