Skip to content

Commit a9ea862

Browse files
committed
update: test for Ranger21
1 parent 7410f83 commit a9ea862

File tree

1 file changed

+0
-11
lines changed

1 file changed

+0
-11
lines changed

tests/test_optimizer_parameters.py

Lines changed: 0 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -221,17 +221,6 @@ def test_ranger21_size_of_parameter():
221221
Ranger21(model.parameters(), 100).step()
222222

223223

224-
def test_ranger21_variance_normalized():
225-
model: nn.Module = Example()
226-
optimizer = Ranger21(model.parameters(), num_iterations=100, betas=(0.9, 1e-9))
227-
228-
y_pred = model(torch.ones((1, 1)))
229-
loss = F.binary_cross_entropy_with_logits(y_pred, torch.zeros(1, 1))
230-
231-
with pytest.raises(RuntimeError):
232-
optimizer.step()
233-
234-
235224
def test_ranger21_closure():
236225
model: nn.Module = Example()
237226
optimizer = Ranger21(model.parameters(), num_iterations=100, betas=(0.9, 1e-9))

0 commit comments

Comments
 (0)