Skip to content

Commit 87e1a60

Browse files
committed
update: test_spam_optimizer
1 parent 8c5e9c2 commit 87e1a60

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/test_optimizers.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -847,6 +847,9 @@ def test_mars_c_t_norm():
847847
optimizer.step()
848848

849849

850-
def test_spam_non_zero_count():
850+
def test_spam_optimizer():
851851
optimizer = load_optimizer('spam')(Example().parameters(), density=0.0)
852852
optimizer.step()
853+
854+
optimizer = load_optimizer('spam')([simple_parameter(True)], grad_accu_steps=0, update_proj_gap=1)
855+
optimizer.step()

0 commit comments

Comments
 (0)