Struct DataLoaderConfig 

pub struct DataLoaderConfig {
    pub batch_size: usize,
    pub shuffle: bool,
    pub num_workers: usize,
    pub pin_memory: bool,
    pub drop_last: bool,
    pub timeout: Option<Duration>,
    pub generator: Option<u64>,
    pub prefetch_buffer_size: usize,
    pub persistent_workers: bool,
    pub pinning_config: Option<PinningConfig>,
}

Configuration for DataLoader creation with all options

This struct provides a comprehensive configuration interface that combines options from all DataLoader components.

Fields

batch_size: usize

Batch size for data loading

shuffle: bool

Whether to shuffle the data

num_workers: usize

Number of worker threads

pin_memory: bool

Whether to pin memory for GPU transfers

drop_last: bool

Whether to drop the last incomplete batch

timeout: Option<Duration>

Timeout for batch collection

generator: Option<u64>

Random seed for reproducible shuffling

prefetch_buffer_size: usize

Prefetch buffer size (0 to disable prefetching)

persistent_workers: bool

Whether to use persistent workers

pinning_config: Option<PinningConfig>

Memory pinning configuration

Implementations

impl DataLoaderConfig

pub fn new() -> Self

Create a new DataLoader configuration

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

Set batch size

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

Set shuffle

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

Set number of workers

pub fn pin_memory(self, pin_memory: bool) -> Self

Set pin memory

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

Set drop last

pub fn timeout(self, timeout: Duration) -> Self

Set timeout

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

Set generator seed

pub fn prefetch_buffer_size(self, size: usize) -> Self

Set prefetch buffer size

pub fn persistent_workers(self, persistent: bool) -> Self

Enable persistent workers

pub fn pinning_config(self, config: PinningConfig) -> Self

Set memory pinning configuration

pub fn for_training() -> Self

Create a configuration optimized for training

pub fn for_inference() -> Self

Create a configuration optimized for inference

pub fn for_evaluation() -> Self

Create a configuration optimized for evaluation

Trait Implementations

impl Clone for DataLoaderConfig

fn clone(&self) -> DataLoaderConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DataLoaderConfig

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

Formats the value using the given formatter.

impl Default for DataLoaderConfig

fn default() -> Self

Returns the “default value” for a type.