Function test_optimizer_error

tests/python/relax/test_training_optimizer.py:29–55

()

def test_optimizer_error():
    x1 = relax.Var("x1", R.Tensor((3, 3), "float32"))
    x2 = relax.Var("x2", R.Tensor((3, 3), "float64"))
    x3 = relax.Var("x3", R.Tuple([R.Tensor((3, 3), "float32")]))
    x4 = relax.Var("x4", R.Tensor((3, 3), "int64"))
    x5 = relax.Tuple([x1])

    # fine cases
    SGD(0.01).init(x1)
    SGD(0.01).init([x1])
    assert SGD(0.01).init([x2]).dtype == "float64"

    with pytest.raises(ValueError):
        SGD(0.01).init([x1, x1])
    with pytest.raises(ValueError):
        SGD(0.01).init([x1, x2])
    with pytest.raises(ValueError):
        SGD(0.01).init(x3)
    with pytest.raises(ValueError):
        SGD(0.01).init(x4)
    with pytest.raises(ValueError):
        SGD(0.01).init(x5)
    with pytest.raises(
        RuntimeError,
        match="Please call init\\(\\) for the optimizer before calling get_function\\(\\)",
    ):
        SGD(0.01).get_function()


def test_sgd_simple():