Skip to content

Commit 7dd7911

Browse files
committed
update: recipe
1 parent 799d09a commit 7dd7911

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/test_gradients.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,8 @@ def test_no_gradients(optimizer_name):
3030
optimizer = TRAC(load_optimizer('adamw')(params))
3131
elif optimizer_name == 'orthograd':
3232
optimizer = OrthoGrad(load_optimizer('adamw')(params))
33+
elif optimizer_name == 'alice':
34+
optimizer = load_optimizer('alice')(params, rank=2, leading_basis=1)
3335
else:
3436
optimizer = load_optimizer(optimizer_name)(params)
3537

0 commit comments

Comments
 (0)