def test_lars_lr_greater_than_zero(tmpdir): reset_seed() model = LitMNIST() with pytest.raises(ValueError, match='Invalid learning rate.*'): opt = LARS(model.parameters(), lr=-0.5) opt = LARS(model.parameters(), lr=0.003)
def test_lars_weight_decay_greater_than_zero(tmpdir): reset_seed() model = LitMNIST() with pytest.raises(ValueError, match='Invalid weight_decay.*'): opt = LARS(model.parameters(), lr=0.003, weight_decay=-0.01) opt = LARS(model.parameters(), lr=0.003, weight_decay=0.01)
def test_lars_eta_greater_than_zero(tmpdir): reset_seed() model = LitMNIST() with pytest.raises(ValueError, match='Invalid LARS coefficient.*'): opt = LARS(model.parameters(), lr=0.003, eta=-0.01) opt = LARS(model.parameters(), lr=0.003, eta=0.01)
def test_lars_momentum_greater_than_zero(tmpdir): reset_seed() model = LitMNIST() with pytest.raises(ValueError, match='Invalid momentum.*'): opt = LARS(model.parameters(), lr=0.003, momentum=-0.01) opt = LARS(model.parameters(), lr=0.003, momentum=0.01)