update: test cases

kozistr · Dec 21, 2024 · 42b1d76 · 42b1d76
1 parent b4a32b0
commit 42b1d76
Showing 1 changed file with 2 additions and 0 deletions.
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),
 ]