Skip to content

Commit 4a1b3b7

Browse files
committed
update: test_no_gradients
1 parent 26c3288 commit 4a1b3b7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_gradients.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def test_no_gradients(optimizer_name):
3737
sphere_loss(p1 + p3).backward(create_graph=True)
3838

3939
optimizer.step(lambda: 0.1) # for AliG optimizer
40-
if optimizer_name not in {'lookahead', 'trac'}:
40+
if optimizer_name not in {'lookahead', 'trac', 'orthograd'}:
4141
optimizer.zero_grad(set_to_none=True)
4242

4343

0 commit comments

Comments
 (0)