Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
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: