We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 799d09a commit 7dd7911Copy full SHA for 7dd7911
tests/test_gradients.py
@@ -30,6 +30,8 @@ def test_no_gradients(optimizer_name):
30
optimizer = TRAC(load_optimizer('adamw')(params))
31
elif optimizer_name == 'orthograd':
32
optimizer = OrthoGrad(load_optimizer('adamw')(params))
33
+ elif optimizer_name == 'alice':
34
+ optimizer = load_optimizer('alice')(params, rank=2, leading_basis=1)
35
else:
36
optimizer = load_optimizer(optimizer_name)(params)
37
0 commit comments