Skip to content

Commit c14aec8

Browse files
committed
update: test_mars_c_t_norm
1 parent 2023956 commit c14aec8

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tests/test_optimizers.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -812,3 +812,11 @@ def test_muon_rank(rank):
812812
model[2].weight.grad = torch.randn(1, 1, 1)
813813

814814
optimizer.step()
815+
816+
817+
def test_mars_c_t_norm():
818+
param = simple_parameter(True)
819+
param.grad[0] = 100.0
820+
821+
optimizer = load_optimizer('mars')([param], optimize_1d=True)
822+
optimizer.step()

0 commit comments

Comments
 (0)