Skip to content

Commit 8721654

Browse files
committed
ci(test): test_sam_no_gradient case
1 parent f43a8c3 commit 8721654

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

tests/test_optimizers.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -375,6 +375,15 @@ def test_no_closure():
375375
optimizer.step()
376376

377377

378+
def test_sam_no_gradient():
379+
param = torch.randn(1, 1).requires_grad_(False)
380+
param.grad = torch.randn(1, 1)
381+
382+
optimizer = SAM([param], AdamP)
383+
optimizer.zero_grad()
384+
optimizer.step(closure=dummy_closure)
385+
386+
378387
@pytest.mark.parametrize('optimizer_config', OPTIMIZERS, ids=ids)
379388
def test_reset(optimizer_config):
380389
_, model, _ = build_environment()

0 commit comments

Comments
 (0)