Esempi in Python per make_backpack_extension_runner

Esempio n. 1

File: my_pytest_benchmark.py Progetto: toiaydcdyywlhzvlob/backpack

def test_cifar10_backpack_extension(N, ext, benchmark):
    benchmark(
        utils.make_backpack_extension_runner(make_loader(N), model, lossfunc,
                                             ext))

Esempio n. 2

File: additional_benchmark_cifar10_sigmoid.py Progetto: toiaydcdyywlhzvlob/backpack

def test_backpack_diagGGN(N, benchmark):
    benchmark(
        utils.make_backpack_extension_runner(
            make_loader(N), model, lossfunc, backpack_all_extensions["DiagGGNExact"]
        )
    )