update: SOAP recipe

kozistr · Sep 29, 2024 · edaea93 · edaea93
1 parent 851eb0d
commit edaea93
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]] = [