We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2f4564e commit 61bd683Copy full SHA for 61bd683
synapse_net/inference/util.py
@@ -219,7 +219,7 @@ def get_prediction_torch_em(
219
if os.path.isdir(model_path): # Load the model from a torch_em checkpoint.
220
model = torch_em.util.load_model(checkpoint=model_path, device=device)
221
else: # Load the model directly from a serialized pytorch model.
222
- model = torch.load(model_path)
+ model = torch.load(model_path, weights_only=False)
223
224
# Run prediction with the model.
225
with torch.no_grad():
0 commit comments