Struct DataLoaderConfig 

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

Configuration for the DataLoader.

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 parallel workers for sample fetching (0 = sequential).

seed: Option<u64>

Optional random seed for reproducible shuffling.

Implementations

impl DataLoaderConfig

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 seed(self, s: u64) -> Self

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.