Try iterative lambda aggregate

pomonam · Mar 12, 2024 · add12fe · add12fe
1 parent 6b9b736
commit add12fe
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/gpu_tests/prepare_tests.py b/tests/gpu_tests/prepare_tests.py
@@ -98,6 +98,7 @@ def run_analysis() -> None:
         activation_covariance_dtype=torch.float64,
         gradient_covariance_dtype=torch.float64,
         lambda_dtype=torch.float64,
+        lambda_iterative_aggregate=True,
     )
     analyzer.fit_all_factors(
         factors_name="single_gpu",