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)#