kappabenchmark 0.0.8

pytorch dataset wrappers for in-memory caching

KappaBenchmark

Utilities for benchmarking pytorch applications.

• Dataloading

Dataloading

import kappabenchmark as kbm
dataloader = ...
result = kbm.benchmark_dataloading(
    dataloader=dataloader,
    num_epochs=...,
)

predefined benchmarks examples

• python main_benchmark_grid.py --benchmark imagefolder --root ROOT --num_epochs 5 --batch_size 256 --num_workers 8,16 --num_fetch_workers 0,2,4