From a66b87774e10f7b96ef8f5159850704511355c5c Mon Sep 17 00:00:00 2001 From: kozistr Date: Sat, 7 Dec 2024 01:12:11 +0900 Subject: [PATCH] update: test case --- tests/constants.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/constants.py b/tests/constants.py index ec397736..de3c1d44 100644 --- a/tests/constants.py +++ b/tests/constants.py @@ -489,7 +489,7 @@ (GrokFastAdamW, {'lr': 5e0, 'weight_decay': 1e-3, 'grokfast_after_step': 1}, 5), (Kate, {'lr': 5e-2}, 10), (StableAdamW, {'lr': 1e0}, 5), - (AdamG, {'lr': 1e0}, 20), + (AdamG, {'lr': 1e0, 'p': 0.5}, 60), (AdEMAMix, {'lr': 1e0}, 3), (AdEMAMix, {'lr': 1e0, 't_alpha_beta3': 5}, 3), (