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 26c3288 commit 4a1b3b7Copy full SHA for 4a1b3b7
tests/test_gradients.py
@@ -37,7 +37,7 @@ def test_no_gradients(optimizer_name):
37
sphere_loss(p1 + p3).backward(create_graph=True)
38
39
optimizer.step(lambda: 0.1) # for AliG optimizer
40
- if optimizer_name not in {'lookahead', 'trac'}:
+ if optimizer_name not in {'lookahead', 'trac', 'orthograd'}:
41
optimizer.zero_grad(set_to_none=True)
42
43
0 commit comments