handle negative eigenvalues

Kapoorlabs-CAPED · Dec 4, 2023 · 746d26f · 746d26f
1 parent 55b913c
commit 746d26f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/napatrackmater/Trackmate.py b/src/napatrackmater/Trackmate.py
@@ -14,7 +14,7 @@
 from pathlib import Path
 import concurrent
 from .clustering import Clustering
-
+from lightning import Trainer
 
 class TrackMate:
     def __init__(
@@ -59,6 +59,7 @@ def __init__(
         self.center = center
         self.compute_with_autoencoder = compute_with_autoencoder
         self.latent_features = latent_features
+        self.pretrainer = Trainer(accelerator=self.accelerator, devices=self.devices)
         if image is not None:
             self.image = image.astype(np.uint8)
         else:

diff --git a/src/napatrackmater/clustering.py b/src/napatrackmater/clustering.py
@@ -174,7 +174,6 @@ def _latent_computer(self, i, dim):
     def _create_cluster_labels(self):
 
         ndim = len(self.label_image.shape)
-        self.pretrainer = Trainer(accelerator=self.accelerator, devices=self.devices)
         if ndim == 2:
 
             labels, centroids, clouds, marching_cube_points = _label_cluster(