replace torch.symeig with torch.linalg.eigh

EIDOSLAB · Jan 13, 2025 · 7c2a95f · 7c2a95f
1 parent e621b86
commit 7c2a95f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/torchstain/torch/normalizers/multitarget.py b/torchstain/torch/normalizers/multitarget.py
@@ -49,7 +49,8 @@ def __find_concentration(self, OD, HE):
     def __compute_matrices_single(self, I, Io, alpha, beta):
         OD, ODhat = self.__convert_rgb2od(I, Io=Io, beta=beta)
 
-        _, eigvecs = torch.symeig(cov(ODhat.T), eigenvectors=True)
+        # _, eigvecs = torch.symeig(cov(ODhat.T), eigenvectors=True)
+        _, eigvecs = torch.linalg.eigh(cov(ODhat.T), UPLO='U')
         eigvecs = eigvecs[:, [1, 2]]
 
         HE = self.__find_HE(ODhat, eigvecs, alpha)