Python bp Examples

Example #1

File: test_second_order_warnings.py Project: f-dangel/backpack

def test_sqrt_hessian_mse_should_pass(extension):
    loss = dummy_mse()

    with bp(extension()):
        loss.backward()

Example #2

File: test_second_order_warnings.py Project: f-dangel/backpack

def test_sqrt_hessian_modified_mse_should_fail(extension):
    loss = dummy_mse() * 2

    with pytest.warns(UserWarning):
        with bp(extension()):
            loss.backward()

Example #3

File: test_second_order_warnings.py Project: f-dangel/backpack

def test_sqrt_hessian_crossentropy_should_pass(extension):
    loss = dummy_cross_entropy()

    with bp(extension()):
        loss.backward()