Struct AsyncDataLoaderConfig 

pub struct AsyncDataLoaderConfig {
    pub batch_size: usize,
    pub shuffle: bool,
    pub drop_last: bool,
    pub dtype: DType,
    pub num_workers: usize,
    pub prefetch_factor: usize,
    pub seed: Option<u64>,
}

Configuration for the async prefetching data loader.

Fields

batch_size: usize

Number of samples per batch.

shuffle: bool

Whether to shuffle indices each epoch.

drop_last: bool

Whether to drop the last incomplete batch.

dtype: DType

DType for the created tensors.

num_workers: usize

Number of background workers (threads) for loading + transforming. 0 = no background threads (falls back to sync loading, with prefetch still happening on a single background thread).

prefetch_factor: usize

How many batches to pre-load ahead of the consumer. Total buffered batches = prefetch_factor * max(num_workers, 1).

seed: Option<u64>

Optional random seed for reproducible shuffling.

Implementations

impl AsyncDataLoaderConfig

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

pub fn shuffle(self, s: bool) -> Self

pub fn drop_last(self, d: bool) -> Self

pub fn dtype(self, d: DType) -> Self

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

pub fn prefetch_factor(self, pf: usize) -> Self

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

Trait Implementations

impl Clone for AsyncDataLoaderConfig

fn clone(&self) -> AsyncDataLoaderConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for AsyncDataLoaderConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for AsyncDataLoaderConfig

fn default() -> Self

Returns the “default value” for a type.