deepr.prepros.Prefetch
- class deepr.prepros.Prefetch(buffer_size)[source]
Creates a dataset that prefetch element on CPU / GPU.
- buffer_size
Number of element to prefetch. High values may lead to high memory consumption, it is recommended to use a buffer_size of 1.
- Type:
Methods
__init__(buffer_size)apply(dataset[, mode])Pre-process a dataset