predict mitosis

Kapoorlabs-CAPED · Nov 26, 2023 · de4b068 · de4b068
1 parent e7f5beb
commit de4b068
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()