Skip to content

Commit e650bae

Browse files
committed
update: use_svd to False
1 parent 24282a9 commit e650bae

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

pytorch_optimizer/optimizer/shampoo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -194,7 +194,7 @@ def __init__(
194194
nesterov: bool = True,
195195
diagonal_eps: float = 1e-10,
196196
matrix_eps: float = 1e-6,
197-
use_svd: bool = True,
197+
use_svd: bool = False,
198198
):
199199
self.lr = lr
200200
self.betas = betas

pytorch_optimizer/optimizer/shampoo_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -216,7 +216,7 @@ def __init__(
216216
shape_interpretation: bool,
217217
matrix_eps: float = 1e-6,
218218
pre_conditioner_type: int = PreConditionerType.ALL,
219-
use_svd: bool = True,
219+
use_svd: bool = False,
220220
):
221221
self.beta2 = beta2
222222
self.inverse_exponent_override = inverse_exponent_override

0 commit comments

Comments
 (0)