update: test case

kozistr · kozistr · commit a5e0894cd473 · 2024-12-21T19:35:49.000+09:00
diff --git a/tests/constants.py b/tests/constants.py
@@ -540,6 +540,7 @@
     (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': 1e0}, 15),
+    (SGDSaI, {'lr': 1e0, 'momentum': 0.0}, 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
`@@ -540,6 +540,7 @@`
`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`	`542`	`(SGDSaI, {'lr': 1e0}, 15),`
	`543`	`+ (SGDSaI, {'lr': 1e0, 'momentum': 0.0}, 15),`
`543`	`544`	`]`
`544`	`545`	`ADANORM_SUPPORTED_OPTIMIZERS: List[Tuple[Any, Dict[str, Union[float, bool, int]], int]] = [`
`545`	`546`	`(AdaBelief, {'lr': 5e-1, 'weight_decay': 1e-3, 'adanorm': True}, 10),`