pytorch_pfn_extras.dataset.SharedDataset¶

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

Dataset that caches the load samples in shared memory

Args

__init__(sm_size, cache_type=<class 'pytorch_pfn_extras.dataset.shared_dataset.InfiniteCache'>)¶

Methods

`__init__`(sm_size[, cache_type])
`cache_item`(idx, x)
`is_cached`(idx)
`register_datapipe_as_function`(function_name, …)
`register_function`(function_name, function)

Attributes

functions