pytorch_pfn_extras.dataset.SharedDataset#

class pytorch_pfn_extras.dataset.SharedDataset(sm_size, cache_type=<class 'pytorch_pfn_extras.dataset.shared_dataset.InfiniteCache'>)#

Bases: Dataset

Dataset that caches the load samples in shared memory

Args

Methods

__init__(sm_size[, cache_type])

cache_item(idx, x)

is_cached(idx)

__init__(sm_size, cache_type=<class 'pytorch_pfn_extras.dataset.shared_dataset.InfiniteCache'>)#
cache_item(idx, x)#
is_cached(idx)#