update: test cases

kozistr · kozistr · commit 42b1d7631c79 · 2024-12-21T16:04:57.000+09:00
diff --git a/tests/constants.py b/tests/constants.py
@@ -586,4 +586,6 @@
     (AdamP, {'lr': 1e0, 'cautious': True}, 2),
     (ADOPT, {'lr': 1e1, 'cautious': True}, 3),
     (AdaShift, {'lr': 1e1, 'keep_num': 1, 'cautious': True}, 3),
+    (MARS, {'lr': 5e-1, 'lr_1d': 5e-1, 'weight_decay': 1e-3, 'cautious': True}, 3),
+    (MARS, {'lr': 5e-1, 'lr_1d': 5e-1, 'weight_decay': 1e-3, 'optimize_1d': True, 'cautious': True}, 3),
 ]

Original file line number	Diff line number	Diff line change
`@@ -586,4 +586,6 @@`
`586`	`586`	`(AdamP, {'lr': 1e0, 'cautious': True}, 2),`
`587`	`587`	`(ADOPT, {'lr': 1e1, 'cautious': True}, 3),`
`588`	`588`	`(AdaShift, {'lr': 1e1, 'keep_num': 1, 'cautious': True}, 3),`
	`589`	`+ (MARS, {'lr': 5e-1, 'lr_1d': 5e-1, 'weight_decay': 1e-3, 'cautious': True}, 3),`
	`590`	`+ (MARS, {'lr': 5e-1, 'lr_1d': 5e-1, 'weight_decay': 1e-3, 'optimize_1d': True, 'cautious': True}, 3),`
`589`	`591`	`]`