predict mitosis

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