fix weights_only

juliusge · juliusge · commit 6870ee3af843 · 2025-01-31T14:21:20.000+01:00
diff --git a/fundus_image_toolbox/registration/inference.py b/fundus_image_toolbox/registration/inference.py
@@ -210,7 +210,7 @@ def load_model(config: dict = None):
 
     model = SuperRetina().to(device)
     model.load_state_dict(
-        torch.load(config["model_save_path"], map_location=device)["net"], weights_only=False
+        torch.load(config["model_save_path"], map_location=device, weights_only = False)["net"]
     )
     model.eval()
 

Original file line number	Diff line number	Diff line change
`@@ -210,7 +210,7 @@ def load_model(config: dict = None):`
`210`	`210`
`211`	`211`	`model = SuperRetina().to(device)`
`212`	`212`	`model.load_state_dict(`
`213`		`- torch.load(config["model_save_path"], map_location=device)["net"], weights_only=False`
	`213`	`+ torch.load(config["model_save_path"], map_location=device, weights_only = False)["net"]`
`214`	`214`	`)`
`215`	`215`	`model.eval()`
`216`	`216`