Struct DataLoaderBuilder 

pub struct DataLoaderBuilder<B: Backend, I, O> { /* private fields */ }

Available on crate features std and dataset only.

A builder for data loaders.

Implementations

impl<B, I, O> DataLoaderBuilder<B, I, O>

where B: Backend, I: Send + Sync + Clone + Debug + 'static, O: Send + Clone + Debug + 'static,

pub fn new<Bt>(batcher: Bt) -> Self

where Bt: Batcher<B, I, O> + 'static,

Creates a new data loader builder.

Arguments

• batcher - The batcher.

Returns

The data loader builder.

pub fn batch_size(self, batch_size: usize) -> Self

Sets the batch size to a fix number.

The fix batch strategy will be used.

Arguments

• batch_size - The batch size.

Returns

The data loader builder.

pub fn shuffle(self, seed: u64) -> Self

Sets the seed for shuffling.

Each time the dataloader starts a new iteration, the dataset will be shuffled.

Arguments

• seed - The seed.

Returns

The data loader builder.

pub fn num_workers(self, num_workers: usize) -> Self

Sets the number of workers.

• Some(0) or None: the dataloader will run without work threads.
• Some(n); n > 0: the dataloader will run with n background threads.

A 1-worker threaded dataloader will run loads in a background thread, while a 0-worker threaded dataloader will run loads in the main thread.

Arguments

• num_workers - The number of workers.

Returns

The data loader builder.

pub fn set_device(self, device: B::Device) -> Self

Sets the data loader device.

Arguments

• device - The device to use when loading a batch.

Returns

The data loader builder.

pub fn build<D>(self, dataset: D) -> Arc<dyn DataLoader<B, O>>

where D: Dataset<I> + 'static,

Builds the data loader.

Arguments

• dataset - The dataset.

Returns

The data loader.