Shortcuts

Struct ChunkDatasetOptions

Page Contents

Struct Documentation

struct torch::data::datasets::ChunkDatasetOptions

Options to configure a ChunkDataset.

Public Functions

ChunkDatasetOptions() = delete
ChunkDatasetOptions(size_t preloader_count, size_t batch_size, size_t cache_size = 2048, size_t cross_chunk_shuffle_count = 1)
auto preloader_count(const size_t &new_preloader_count) -> decltype(*this)

The number of worker thread to preload chunk data.

auto preloader_count(size_t &&new_preloader_count) -> decltype(*this)
const size_t &preloader_count() const noexcept
size_t &preloader_count() noexcept
auto batch_size(const size_t &new_batch_size) -> decltype(*this)

The size of each batch.

auto batch_size(size_t &&new_batch_size) -> decltype(*this)
const size_t &batch_size() const noexcept
size_t &batch_size() noexcept
auto cache_size(const size_t &new_cache_size) -> decltype(*this)

The capacity of the queue for batch caching.

auto cache_size(size_t &&new_cache_size) -> decltype(*this)
const size_t &cache_size() const noexcept
size_t &cache_size() noexcept
auto cross_chunk_shuffle_count(const size_t &new_cross_chunk_shuffle_count) -> decltype(*this)
auto cross_chunk_shuffle_count(size_t &&new_cross_chunk_shuffle_count) -> decltype(*this)
const size_t &cross_chunk_shuffle_count() const noexcept
size_t &cross_chunk_shuffle_count() noexcept

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources