pub struct DataLoaderBuilder<B: Backend, I, O> { /* private fields */ }dataset and std only.Expand description
A builder for data loaders.
Implementations§
Source§impl<B, I, O> DataLoaderBuilder<B, I, O>
impl<B, I, O> DataLoaderBuilder<B, I, O>
Sourcepub fn batch_size(self, batch_size: usize) -> Self
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.
Sourcepub fn num_workers(self, num_workers: usize) -> Self
pub fn num_workers(self, num_workers: usize) -> Self
Sets the number of workers.
Some(0)orNone: the dataloader will run without work threads.Some(n); n > 0: the dataloader will run withnbackground 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.
Sourcepub fn reuse_workers(self, enabled: bool) -> Self
pub fn reuse_workers(self, enabled: bool) -> Self
Enables reuse of idle background workers across iterations (default: false).
Requires a positive Self::num_workers setting. Finishing or dropping an
iterator waits for its in-flight work; concurrent iterators use transient
workers if the reusable pool is busy. No effect for a single-threaded loader.
Sourcepub fn batch_reading(self, enabled: bool) -> Self
pub fn batch_reading(self, enabled: bool) -> Self
Enables batch dataset reads (default: false) for all workers.
Uses the strategy’s batch size, or one for an unknown or zero size. Reads may fetch later samples before the strategy emits a batch. Missing items or a mismatched result length panic without falling back to single reads.