Adding back in explained variance and explained covariance as well as…

… clarification of canonical loadings and loadings.
jameschapman19 · Aug 30, 2023 · ed5fb15 · ed5fb15
1 parent a49dc6d
commit ed5fb15
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cca_zoo/linear/_gradient/_base.py b/cca_zoo/linear/_gradient/_base.py
@@ -20,7 +20,7 @@
 
 DEFAULT_LOADER_KWARGS = dict(pin_memory=False, drop_last=False, shuffle=True)
 
-DEFAULT_OPTIMIZER_KWARGS = dict(optimizer="SGD", momentum=0.9, nesterov=True)
+DEFAULT_OPTIMIZER_KWARGS = dict(optimizer="Adam")
 
 
 class BaseGradientModel(BaseModel, pl.LightningModule):