pytorch dataset wrappers for in-memory caching
Project description
KappaBenchmark
Utilities for benchmarking pytorch applications.
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
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
kappabenchmark-0.0.6.tar.gz
(4.3 kB
view hashes)
Built Distribution
Close
Hashes for kappabenchmark-0.0.6-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 0573481c0caacc0c567ef9597609beec46595c130e35d252afafa0690bbe7ac6 |
|
MD5 | b5641b9edd22fe3ad64f83e95b1c13e7 |
|
BLAKE2b-256 | 1a27f2d59b840e6872fefcf5df3dac3c35cdb136c35059ce39cd30fa244c447c |