From 1bbaa2f1193867412210b2fe9902dcf51229437c Mon Sep 17 00:00:00 2001 From: runame Date: Sat, 7 Dec 2024 19:19:51 +0000 Subject: [PATCH] Fix docstring --- distributed_shampoo/distributed_shampoo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/distributed_shampoo/distributed_shampoo.py b/distributed_shampoo/distributed_shampoo.py index 376daad..c31043c 100644 --- a/distributed_shampoo/distributed_shampoo.py +++ b/distributed_shampoo/distributed_shampoo.py @@ -272,7 +272,7 @@ class DistributedShampoo(torch.optim.Optimizer): preconditioner_config (PreconditionerConfig): Configuration for preconditioner computation. If this field is an instance ShampooPreconditionerConfig, Shampoo uses the root inverse of the preconditioner. If this field is an instance EigenvalueCorrectedShampooPreconditionerConfig Shampoo uses corrected the eigenvalues/running Adam in the eigenbasis of preconditioner. - (Default: DefaultEigenConfig) + (Default: DefaultShampooConfig) """