From edaea9390d4d6a85365eb51d91bc0d15e53abfc9 Mon Sep 17 00:00:00 2001 From: kozistr Date: Sun, 29 Sep 2024 16:12:47 +0900 Subject: [PATCH] update: SOAP recipe --- tests/constants.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/constants.py b/tests/constants.py index 1a2d3280..15cd66ab 100644 --- a/tests/constants.py +++ b/tests/constants.py @@ -480,7 +480,7 @@ ( SOAP, {'lr': 1e0, 'shampoo_beta': 0.95, 'precondition_frequency': 1, 'merge_dims': False, 'precondition_1d': True}, - 5, + 3, ), ] ADANORM_SUPPORTED_OPTIMIZERS: List[Tuple[Any, Dict[str, Union[float, bool, int]], int]] = [