pulsatrix
Loading...
Searching...
No Matches
pulsatrix::DataLoaderOptions Struct Reference

Configuration for a DataLoader. More...

#include <data_loader.hpp>

Public Attributes

int64_t batch_size = 1
 
bool shuffle = false
 
std::optional< unsigned > shuffle_seed
 Shuffle seed. Unset draws one from the global seed stream (next_seed(), FND-7) when the loader is built, so loaders don't all share one shuffle order.
 
int num_workers = 0
 0 = fully synchronous, no threads spawned (this phase's only exercised path).
 
int64_t prefetch_batches = 2
 
bool drop_last = false
 
CollateFn collate_fn = DefaultCollate
 

Detailed Description

Configuration for a DataLoader.

Member Data Documentation

◆ batch_size

int64_t pulsatrix::DataLoaderOptions::batch_size = 1

◆ collate_fn

CollateFn pulsatrix::DataLoaderOptions::collate_fn = DefaultCollate

◆ drop_last

bool pulsatrix::DataLoaderOptions::drop_last = false

◆ num_workers

int pulsatrix::DataLoaderOptions::num_workers = 0

0 = fully synchronous, no threads spawned (this phase's only exercised path).

◆ prefetch_batches

int64_t pulsatrix::DataLoaderOptions::prefetch_batches = 2

◆ shuffle

bool pulsatrix::DataLoaderOptions::shuffle = false

◆ shuffle_seed

std::optional<unsigned> pulsatrix::DataLoaderOptions::shuffle_seed

Shuffle seed. Unset draws one from the global seed stream (next_seed(), FND-7) when the loader is built, so loaders don't all share one shuffle order.


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