update: test cases

kozistr · kozistr · commit bc9ab50afab2 · 2024-12-21T19:31:30.000+09:00
diff --git a/tests/constants.py b/tests/constants.py
@@ -539,7 +539,7 @@
     (MARS, {'lr': 1e-1, 'weight_decay': 1e-3, 'mars_type': 'lion', 'optimize_1d': True}, 5),
     (MARS, {'lr': 5e-1, 'lr_1d': 5e-1, 'weight_decay': 1e-3, 'mars_type': 'shampoo'}, 5),
     (MARS, {'lr': 5e-1, 'lr_1d': 5e-1, 'weight_decay': 1e-3, 'mars_type': 'adamw', 'ams_bound': True}, 5),
-    (SGDSaI, {'lr': 1e-4, 'weight_decay': 1e-4}, 15),
+    (SGDSaI, {'lr': 1e0}, 15),
 ]
 ADANORM_SUPPORTED_OPTIMIZERS: List[Tuple[Any, Dict[str, Union[float, bool, int]], int]] = [
     (AdaBelief, {'lr': 5e-1, 'weight_decay': 1e-3, 'adanorm': True}, 10),

Original file line number	Diff line number	Diff line change
`@@ -539,7 +539,7 @@`
`539`	`539`	`(MARS, {'lr': 1e-1, 'weight_decay': 1e-3, 'mars_type': 'lion', 'optimize_1d': True}, 5),`
`540`	`540`	`(MARS, {'lr': 5e-1, 'lr_1d': 5e-1, 'weight_decay': 1e-3, 'mars_type': 'shampoo'}, 5),`
`541`	`541`	`(MARS, {'lr': 5e-1, 'lr_1d': 5e-1, 'weight_decay': 1e-3, 'mars_type': 'adamw', 'ams_bound': True}, 5),`
`542`		`- (SGDSaI, {'lr': 1e-4, 'weight_decay': 1e-4}, 15),`
	`542`	`+ (SGDSaI, {'lr': 1e0}, 15),`
`543`	`543`	`]`
`544`	`544`	`ADANORM_SUPPORTED_OPTIMIZERS: List[Tuple[Any, Dict[str, Union[float, bool, int]], int]] = [`
`545`	`545`	`(AdaBelief, {'lr': 5e-1, 'weight_decay': 1e-3, 'adanorm': True}, 10),`