torch::data::datasets::ChunkDatasetOptions Struct Reference

Options to configure a ChunkDataset. More...

#include <chunk.h>

Public Member Functions
	ChunkDatasetOptions (size_t preloader_count, size_t batch_size, size_t cache_size=2048)
	TORCH_ARG (size_t, preloader_count)
	The number of worker thread to preload chunk data.
	TORCH_ARG (size_t, batch_size)
	The size of each batch.
	TORCH_ARG (size_t, cache_size)

Detailed Description

Options to configure a ChunkDataset.

Definition at line 237 of file chunk.h.

---

The documentation for this struct was generated from the following file:

• torch/csrc/api/include/torch/data/datasets/chunk.h