oa::DataLoader

DataLoader class in the OA Data public surface.

ClC++ public headers

Public Types

oa::I64 oa::DataLoader::currentBatch()

oa::Optional<Batch> oa::DataLoader::nextBatch()

oa::I64 oa::DataLoader::numBatches()

void oa::DataLoader::reset()

oa::DataLoader::DataLoader(
	Dataset & inDataset,
	DataLoaderConfig inConfig = {}
)

No public source comment is attached to this declaration.

inDataset

Dataset &

—

inConfig

DataLoaderConfig

Default: {}

oa::I64 oa::DataLoader::currentBatch()

current batch index

oa::I64

The declared return value.

oa::Optional<Batch> oa::DataLoader::nextBatch()

get next batch as multi-tensor. Returns nullopt when epoch ends.

oa::Optional<Batch>

The declared return value.

oa::I64 oa::DataLoader::numBatches()

Number of batches per epoch

oa::I64

The declared return value.

void oa::DataLoader::reset()

reset to beginning of epoch (re-shuffles if enabled)

void

The declared return value.