From e31c25135ed60aa11d8026653d7aec8d7d104255 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Sun, 9 Oct 2022 11:24:06 +0200 Subject: [PATCH] Fixed Bug with Givens Rotation (Eigen) --- metastable_baselines/distributions/distributions.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/metastable_baselines/distributions/distributions.py b/metastable_baselines/distributions/distributions.py index 282d69c..7667cab 100644 --- a/metastable_baselines/distributions/distributions.py +++ b/metastable_baselines/distributions/distributions.py @@ -594,7 +594,7 @@ class CholNet(nn.Module): eigenv = th.cumsum(eigenv, -1) # reverse order, oh well... - self._givens_rot.theta = theta + self._givens_rotator.theta = theta Q = self._givens_rotator(self._givens_ident) Qinv = Q.transpose(dim0=-2, dim1=-1)