From de4b068205a510c055306bb921740dcdb484cf70 Mon Sep 17 00:00:00 2001 From: kapoorlab Date: Sun, 26 Nov 2023 14:46:55 +0100 Subject: [PATCH] predict mitosis --- src/napatrackmater/Trackvector.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/napatrackmater/Trackvector.py b/src/napatrackmater/Trackvector.py index 4d266399..a50855bc 100644 --- a/src/napatrackmater/Trackvector.py +++ b/src/napatrackmater/Trackvector.py @@ -1495,7 +1495,7 @@ def predict_with_model(saved_model_path, saved_model_json, features_array): num_classes_class1=num_classes_class1, num_classes_class2=num_classes_class2, ) - model.load_state_dict(torch.load(saved_model_path), map_location=torch.device(device)) + model.load_state_dict(torch.load(saved_model_path, map_location=torch.device(device))) model.to(device) model.eval()