Skip to content

Commit 7c2a95f

Browse files
replace torch.symeig with torch.linalg.eigh
1 parent e621b86 commit 7c2a95f

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

torchstain/torch/normalizers/multitarget.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,8 @@ def __find_concentration(self, OD, HE):
4949
def __compute_matrices_single(self, I, Io, alpha, beta):
5050
OD, ODhat = self.__convert_rgb2od(I, Io=Io, beta=beta)
5151

52-
_, eigvecs = torch.symeig(cov(ODhat.T), eigenvectors=True)
52+
# _, eigvecs = torch.symeig(cov(ODhat.T), eigenvectors=True)
53+
_, eigvecs = torch.linalg.eigh(cov(ODhat.T), UPLO='U')
5354
eigvecs = eigvecs[:, [1, 2]]
5455

5556
HE = self.__find_HE(ODhat, eigvecs, alpha)

0 commit comments

Comments
 (0)