From d6b1e38b7a6e05e755b9d5bf36070cc4a9ea701f Mon Sep 17 00:00:00 2001 From: kapoorlab Date: Sun, 26 Nov 2023 14:51:03 +0100 Subject: [PATCH] predict mitosis --- src/napatrackmater/Trackvector.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/napatrackmater/Trackvector.py b/src/napatrackmater/Trackvector.py index a50855bc..c6cdeaae 100644 --- a/src/napatrackmater/Trackvector.py +++ b/src/napatrackmater/Trackvector.py @@ -1482,7 +1482,6 @@ def plot_metrics_from_npz(npz_file): def predict_with_model(saved_model_path, saved_model_json, features_array): device = torch.device("cuda" if torch.cuda.is_available() else "cpu") - print(device) with open(saved_model_json) as json_file: model_info = json.load(json_file)