Revert "Fix default initialization"

This reverts commit 554674909a.
This commit is contained in:
rockerBOO
2025-05-07 23:33:54 -04:00
parent 554674909a
commit 19b67643e4
7 changed files with 8 additions and 469 deletions

View File

@@ -11,7 +11,7 @@ class InitializeParams:
"""Parameters for initialization methods (PiSSA, URAE)"""
use_ipca: bool = False
use_lowrank: bool = False
use_lowrank: bool = True
lowrank_q: Optional[int] = None
lowrank_niter: int = 4
lowrank_seed: Optional[int] = None
@@ -187,7 +187,7 @@ def initialize_pissa(
device: Optional[torch.device] = None,
dtype: Optional[torch.dtype] = None,
use_ipca: bool = False,
use_lowrank: bool = False,
use_lowrank: bool = True,
lowrank_q: Optional[int] = None,
lowrank_niter: int = 4,
lowrank_seed: Optional[int] = None,