pytorch-pfn-extras
Dataset that caches the load samples in shared memory
Args
Methods
__init__(sm_size[, cache_type])
__init__
cache_item(idx, x)
cache_item
is_cached(idx)
is_cached