Skip to content

Commit 2b05036

Browse files
committed
update some magical number
Signed-off-by: Hao Wu <skyw@nvidia.com>
1 parent 41b170c commit 2b05036

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tests/soap_mnist_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ def main() -> None:
102102
# Initialize optimizers
103103
optimizer_soap = SOAP(
104104
model_soap.parameters(),
105-
lr=2.1 * config["lr"],
105+
lr=2.05 * config["lr"],
106106
weight_decay=config["weight_decay"],
107107
betas=(config["adam_beta1"], config["adam_beta2"]),
108108
eps=config["eps"],

0 commit comments

Comments
 (0)