update: test_mars_c_t_norm

kozistr · Dec 21, 2024 · c14aec8 · c14aec8
1 parent 2023956
commit c14aec8
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/test_optimizers.py b/tests/test_optimizers.py
@@ -812,3 +812,11 @@ def test_muon_rank(rank):
     model[2].weight.grad = torch.randn(1, 1, 1)
 
     optimizer.step()
+
+
+def test_mars_c_t_norm():
+    param = simple_parameter(True)
+    param.grad[0] = 100.0
+
+    optimizer = load_optimizer('mars')([param], optimize_1d=True)
+    optimizer.step()